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

fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808)

Browse Source
This commit is contained in:
digger yu
2023-05-24 09:01:50 +08:00
committed by GitHub
parent 725365f297
commit 7f8203af69
19 changed files with 31 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/fx/codegen/activation_checkpoint_codegen.py
View File

@@ -523,7 +523,7 @@ def emit_code_with_activation_checkpoint(body, ckpt_func, nodes, emit_node_func,
# append code text to body
for idx, node in enumerate(node_list):
# if this is the first node of the ckpt region
# append the ckpt function defition
# append the ckpt function definition
if idx in start_idx:
label = start_idx.index(idx)
ckpt_fn_def = _gen_ckpt_fn_def(label, input_vars[label])