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

DrRyanHuang · web-flow · commit 1bb12d042592 · 2025-02-20T20:27:09.000+08:00
diff --git a/python/paddle/jit/sot/utils/paddle_api_config.py b/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.
 }
 
diff --git a/python/paddle/pir/math_op_patch.py b/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
diff --git a/test/dygraph_to_static/test_tensor_attr_consistency.py b/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__",
diff --git a/test/sot/test_18_tensor_method.py b/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):

Original file line number	Diff line number	Diff line change
`@@ -123,6 +123,7 @@ def _get_tensor_methods():`
`123`	`123`	`'register_hook',`
`124`	`124`	`'numpy',`
`125`	`125`	`'clear_gradient',`
	`126`	`+ 'tolist',`
`126`	`127`	`# TODO: Browse all possible functions and make prior judgments.`
`127`	`128`	`}`
`128`	`129`