mirror of https://github.com/hpcaitech/ColossalAI
remove useless function
parent
d914a21d64
commit
0b6af554df
|
@ -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…
Reference in New Issue