rename

colossalai/autochunk/search_chunk.py

@@ -48,7 +48,7 @@ class SearchChunk(object):
         self.gm = gm
         self.print_mem = print_mem
         self.trace_indice = TraceIndice(list(gm.graph.nodes))
-        self.trace_indice.trace_index()
+        self.trace_indice.trace_indice()
         self.trace_flow = TraceFlow(self.trace_indice)
         self.reorder_graph = ReorderGraph(self.trace_indice)
         self.estimate_memory = EstimateMemory()

colossalai/autochunk/trace_indice.py

@@ -525,7 +525,7 @@ class TraceIndice(object):
                         merge_to if i == merge_from else i for i in trace["indice"]
                     ]
 
-    def trace_index(self):
+    def trace_indice(self):
         for idx, node in enumerate(self.node_list):
             if node.op == "placeholder":
                 self._assign_all_indice(node, idx)