github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-20 02:10:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

[fx] fixed indentation error in checkpointing codegen (#1385)

Browse Source
This commit is contained in:
Frank Lee 2022-07-30 00:27:12 +08:00 committed by GitHub
parent c7221cb2d4
commit 89e60d1505
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -373,10 +373,10 @@ if codegen_available:
code = ''.join(body) code = ''.join(body)
code = '\n'.join(' ' + line for line in code.split('\n')) code = '\n'.join(' ' + line for line in code.split('\n'))
fn_code = f""" fn_code = f"""
{wrap_stmts} {wrap_stmts}
{prologue} {prologue}
{code}""" {code}"""
return PythonCode(fn_code, globals_) return PythonCode(fn_code, globals_)
else: else: