[Dy2St]Fix func.__self__ problem in FunctionSpec (#50404)

PaddlePaddle · Feb 10, 2023 · 3374600 · 3374600
1 parent 86cc694
commit 3374600
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_list.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_list.py
@@ -338,8 +338,10 @@ class ListWithCondNet(paddle.nn.Layer):
     def __init__(self):
         super().__init__()
 
+    # Add *args to test function.__self__ in FunctionSpec.
+    # DO NOT remove *args.
     @paddle.jit.to_static
-    def forward(self, x, index):
+    def forward(self, x, index, *args):
         y = paddle.nn.functional.relu(x)
         a = []
 

diff --git a/python/paddle/jit/dy2static/function_spec.py b/python/paddle/jit/dy2static/function_spec.py
@@ -54,7 +54,7 @@ def __init__(self, function, input_spec=None):
         # parse *args
         self.varargs_name = parse_varargs_name(function)
         if self.varargs_name is not None and isinstance(
-            function.__self__, TranslatedLayer
+            getattr(function, '__self__', None), TranslatedLayer
         ):
             self._arg_names += function.__self__._input_args_names