From 89e60d1505dff8e63195274f7a82171d8f7379e0 Mon Sep 17 00:00:00 2001
From: Frank Lee <somerlee.9@gmail.com>
Date: Sat, 30 Jul 2022 00:27:12 +0800
Subject: [PATCH] [fx] fixed indentation error in checkpointing codegen (#1385)

---
 colossalai/fx/codegen/activation_checkpoint_codegen.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/colossalai/fx/codegen/activation_checkpoint_codegen.py b/colossalai/fx/codegen/activation_checkpoint_codegen.py
index 6c1c2ae82..1bbb1418f 100644
--- 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: