github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-07 03:52:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

rename function from index to indice

Browse Source
This commit is contained in:
oahzxl
2023-01-09 17:34:30 +08:00
parent 0ea903b94e
commit cb9817f75d
4 changed files with 91 additions and 91 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/autochunk/search_chunk.py
View File

@@ -205,7 +205,7 @@ class SearchChunk(object):
possible_chunk_region (List)
"""
possible_chunk_region = []
output_trace = copy.deepcopy(self.trace_indice.idx_trace_list)
output_trace = copy.deepcopy(self.trace_indice.indice_trace_list)
input_trace = [] # trace of a node's input nodes
for _, n in enumerate(self.trace_indice.node_list):
cur_trace = {}