github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-19 20:23:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove useless function

Browse Source
This commit is contained in:
oahzxl 2023-01-09 17:46:43 +08:00
parent d914a21d64
commit 0b6af554df
1 changed files with 0 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
colossalai/autochunk/trace_indice.py
View File

@ -10,7 +10,6 @@ class TraceIndice(object):
def __init__(self, node_list) -> None: def __init__(self, node_list) -> None:
self.node_list = node_list self.node_list = node_list
self.indice_trace_list = self._init_indice_trace_list() self.indice_trace_list = self._init_indice_trace_list()
self.indice_trace_equal = []
self.indice_view_list = {} self.indice_view_list = {}
self.indice_count = -1 self.indice_count = -1
@ -486,18 +485,6 @@ class TraceIndice(object):
} }
self.indice_view_list[node] = view_dict 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): def trace_indice(self):
for idx, node in enumerate(self.node_list): for idx, node in enumerate(self.node_list):
if node.op == "placeholder": if node.op == "placeholder":