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

89e60d15 · Frank Lee · GitHub · c7221cb2 · 89e60d15
Unverified Commit 89e60d15 authored Jul 30, 2022 by Frank Lee Committed by GitHub Jul 30, 2022
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

colossalai/fx/codegen/activation_checkpoint_codegen.py colossalai/fx/codegen/activation_checkpoint_codegen.py +3 -3

No files found.
--- a/colossalai/fx/codegen/activation_checkpoint_codegen.py
+++ b/colossalai/fx/codegen/activation_checkpoint_codegen.py
@@ -373,10 +373,10 @@ if codegen_available:
            code = ''.join(body)
            code = '\n'.join('    ' + line for line in code.split('\n'))
            fn_code = f"""
-    {wrap_stmts}
+{wrap_stmts}

-    {prologue}
-    {code}"""
+{prologue}
+{code}"""
            return PythonCode(fn_code, globals_)

 else: