Skip to content

GitLab

Commit 6d99994a authored by oahzxl's avatar oahzxl
Browse files

rename index tracer

parent a9d64377
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
chunk_codegen.py
View file @ 6d99994a
...@@ -17,7 +17,7 @@ def _delete_free_var_from_last_use(user_to_last_uses): ...@@ -17,7 +17,7 @@ def _delete_free_var_from_last_use(user_to_last_uses):
user_to_last_uses[key].remove(n) user_to_last_uses[key].remove(n)
class NodeIndexTracer(object): class IndexTracer(object):
def __init__(self, gm) -> None: def __init__(self, gm) -> None:
self.gm = gm self.gm = gm
self.nodes_list = list(gm.graph.nodes) self.nodes_list = list(gm.graph.nodes)
...@@ -683,7 +683,7 @@ class ChunkRegionSearch(object): ...@@ -683,7 +683,7 @@ class ChunkRegionSearch(object):
self.gm = gm self.gm = gm
self.node_list = list(gm.graph.nodes) self.node_list = list(gm.graph.nodes)
self.memory_estimator = MemoryEstimator() self.memory_estimator = MemoryEstimator()
self.index_tracer = NodeIndexTracer(gm) self.index_tracer = IndexTracer(gm)
self.index_tracer.trace_node_idx() self.index_tracer.trace_node_idx()
def _find_peak_node(self, mem_peak): def _find_peak_node(self, mem_peak):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment