update

PaddlePaddle · Sep 28, 2023 · f8383c0 · f8383c0
1 parent 5fc1ee6
commit f8383c0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sot/opcode_translator/skip_files.py b/sot/opcode_translator/skip_files.py
@@ -119,7 +119,7 @@ def _module_dir(m: types.ModuleType):
 
 customed_skip_code = set()
 
-no_skip_code = {paddle.nn.Sequential.forward.__code__}
+no_skip_code = {paddle.nn.Sequential.forward.__code__, paddle.nn.Layer.__call__}
 
 
 def need_skip_path(filepath: str) -> bool:

diff --git a/sot/opcode_translator/transform.py b/sot/opcode_translator/transform.py
@@ -47,7 +47,7 @@ def eval_frame_callback(frame, **kwargs) -> CustomCode:
             return CustomCode(None, True)
 
         if need_skip(frame):
-            return CustomCode(None, False)
+            return CustomCode(None, True)
 
         log(
             2,