Skip to content

Commit

Permalink
only get run_individual_node on local
Browse files Browse the repository at this point in the history
  • Loading branch information
YJ Shi committed May 13, 2022
1 parent dd4574f commit 14edabd
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion python/tvm/contrib/debugger/debug_executor.py
Original file line number Diff line number Diff line change
Expand Up @@ -113,7 +113,8 @@ def __init__(self, module, device, graph_json_str, dump_root):
self._dump_root = dump_root
self._dump_path = None
self._run_individual = module["run_individual"]
self._run_individual_node = module["run_individual_node"]
if module.type_key != "rpc":
self._run_individual_node = module["run_individual_node"]
self._debug_get_output = module["debug_get_output"]
self._execute_node = module["execute_node"]
self._get_node_output = module["get_node_output"]
Expand Down

0 comments on commit 14edabd

Please sign in to comment.