Skip to content

Commit

Permalink
remove useless function
Browse files Browse the repository at this point in the history
  • Loading branch information
oahzxl committed Jan 9, 2023
1 parent d914a21 commit 0b6af55
Showing 1 changed file with 0 additions and 13 deletions.
13 changes: 0 additions & 13 deletions colossalai/autochunk/trace_indice.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down Expand Up @@ -486,18 +485,6 @@ def _assign_view_reshape_indice(self, node, node_idx):
}
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":
Expand Down

0 comments on commit 0b6af55

Please sign in to comment.