[SOT] Support .tolist() in SOT Static Mode (#71201)

Author: DrRyanHuang

python/paddle/jit/sot/utils/paddle_api_config.py

@@ -123,6 +123,7 @@ def _get_tensor_methods():
     'register_hook',
     'numpy',
     'clear_gradient',
+    'tolist',
     # TODO: Browse all possible functions and make prior judgments.
 }
 

python/paddle/pir/math_op_patch.py

@@ -1000,6 +1000,7 @@ def numpy(self):
             ndarray: dtype is same as current Variable
         Examples:
             .. code-block:: python
+
                 >>> import paddle
                 >>> import paddle.base as base
                 >>> from paddle.nn import Linear
@@ -1013,6 +1014,32 @@ def numpy(self):
         """
         pass
 
+    @fake_interface_only
+    def tolist(self):
+        """
+        **Notes**:
+            **This API is ONLY available in Dygraph mode**
+        Returns a Python list that contains the elements of current :ref:`api_guide_Variable_en`
+
+        Returns:
+            list: The Python list containing the elements of current Variable.
+
+        Returns type:
+            list: Elements have the same dtype as current Variable
+
+        Examples:
+            .. code-block:: python
+
+                >>> import paddle
+                >>> import paddle.base as base
+                >>> import numpy as np
+                >>> data = np.random.uniform(-1, 1, [2, 3]).astype('float32')
+                >>> with base.dygraph.guard():
+                ...     x = paddle.to_tensor(data)
+                ...     print(x.tolist())  # Convert tensor to Python list
+        """
+        pass
+
     @fake_interface_only
     def register_hook(self, hook):
         """
@@ -1049,6 +1076,7 @@ def register_hook(self, hook):
         ('values', values),
         ("_to", _to),
         ("to", to),
+        ("tolist", tolist),
         ("numpy", numpy),
         ("register_hook", register_hook),
         # For basic operators

test/dygraph_to_static/test_tensor_attr_consistency.py

@@ -75,7 +75,6 @@
         'strides',
         'to_sparse_coo',
         'to_sparse_csr',
-        'tolist',
         'value',
         'zero_',
         "__cuda_array_interface__",

test/sot/test_18_tensor_method.py

@@ -17,13 +17,16 @@
 from test_case_base import TestCaseBase, test_with_faster_guard
 
 import paddle
+from paddle.jit.sot.psdb import check_no_breakgraph
 
 
+@check_no_breakgraph
 def tensor_method_call_1(x: paddle.Tensor):
     y = x + 1
     return y.mean()
 
 
+@check_no_breakgraph
 def tensor_method_call_2(a: paddle.Tensor, b: paddle.Tensor):
     c = a.add(b)
     d = c.multiply(a)
@@ -39,15 +42,15 @@ def tensor_method_passed_by_user(a: paddle.Tensor, func: paddle.Tensor):
     return func(a)
 
 
-def tensor_method_property(a: paddle.Tensor, b: paddle.Tensor):
+@check_no_breakgraph
+def tensor_method_property_without_breakgraph(
+    a: paddle.Tensor, b: paddle.Tensor
+):
     return (
         a.name,
-        str(a.place),
         a.persistable,
         a.dtype,
-        a.type,
         a.is_tensor(),
-        a.clear_gradient(),
         a @ b.T.astype(a.dtype)
         + len(a.shape)
         + b.size
@@ -57,10 +60,22 @@ def tensor_method_property(a: paddle.Tensor, b: paddle.Tensor):
     )
 
 
+def tensor_method_property_with_breakgraph(a: paddle.Tensor, b: paddle.Tensor):
+    return (
+        a.type,
+        a.numpy(),
+        a.tolist(),
+        str(a.place),
+        a.clear_gradient(),
+    )
+
+
+@check_no_breakgraph
 def tensor_method_property_mT(a: paddle.Tensor):
     return a.mT
 
 
+@check_no_breakgraph
 def middle_tensor_name(a: paddle.Tensor, b: paddle.Tensor):
     c = a + b
     return c.name
@@ -87,7 +102,8 @@ def test_tensor_method_passed_by_user(self):
     def test_tensor_method_property(self):
         x = paddle.rand([42, 24], dtype='float64')
         y = paddle.rand([42, 24], dtype='float32')
-        self.assert_results(tensor_method_property, x, y)
+        self.assert_results(tensor_method_property_without_breakgraph, x, y)
+        self.assert_results(tensor_method_property_with_breakgraph, x, y)
 
     @unittest.skip("TODO: dynamic tensor name is different")
     def test_middle_tensor_name(self):