github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-12 20:54:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

rename

Browse Source
This commit is contained in:
oahzxl
2023-01-09 17:42:25 +08:00
parent a4ed5b0d0d
commit 865f2e0196
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/autochunk/trace_indice.py
View File

@@ -525,7 +525,7 @@ class TraceIndice(object):
merge_to if i == merge_from else i for i in trace["indice"]
]
def trace_index(self):
def trace_indice(self):
for idx, node in enumerate(self.node_list):
if node.op == "placeholder":
self._assign_all_indice(node, idx)