Browse Source

remove useless function

pull/2364/head
oahzxl 2 years ago
parent
commit
0b6af554df
  1. 13
      colossalai/autochunk/trace_indice.py

13
colossalai/autochunk/trace_indice.py

@ -10,7 +10,6 @@ class TraceIndice(object):
def __init__(self, node_list) -> None:
self.node_list = node_list
self.indice_trace_list = self._init_indice_trace_list()
self.indice_trace_equal = []
self.indice_view_list = {}
self.indice_count = -1
@ -486,18 +485,6 @@ class TraceIndice(object):
}
self.indice_view_list[node] = view_dict
def _merge_equal_idx(self):
idx_equal = copy.deepcopy(self.indice_trace_equal)
idx_equal.reverse()
for idx in idx_equal:
merge_to = min(idx)
merge_from = max(idx)
for trace in self.indice_trace_list:
if merge_from in trace["indice"]:
trace["indice"] = [
merge_to if i == merge_from else i for i in trace["indice"]
]
def trace_indice(self):
for idx, node in enumerate(self.node_list):
if node.op == "placeholder":

Loading…
Cancel
Save