【PIR API adaptor No.41、138、153】 Migrate paddle.complex, paddle.logspace, paddle.meshgrid into pir

python/paddle/tensor/creation.py

@@ -457,7 +457,7 @@ def logspace(start, stop, num, base=10.0, dtype=None, name=None):
     if not isinstance(base, Variable):
         with device_guard("cpu"):
             tensor_base = fill_constant([1], dtype, base)
-    if in_dynamic_mode():
+    if in_dynamic_or_pir_mode():
         return _C_ops.logspace(
             tensor_start,
             tensor_stop,
@@ -1643,7 +1643,7 @@ def meshgrid(*args, **kwargs):
 
     if len(args) == 1 and isinstance(args[0], (list, tuple)):
         args = args[0]
-    if in_dynamic_mode():
+    if in_dynamic_or_pir_mode():
         return _C_ops.meshgrid(list(args))
     else:
         name = kwargs.get("name", None)
@@ -2603,7 +2603,7 @@ def complex(real, imag, name=None):
             [[0j    , 1j    , 2j    ],
              [(1+0j), (1+1j), (1+2j)]])
     """
-    if in_dynamic_mode():
+    if in_dynamic_or_pir_mode():
         return _C_ops.complex(real, imag)
     else:
         check_variable_and_dtype(

test/legacy_test/test_complex_op.py

@@ -20,6 +20,7 @@
 import paddle
 from paddle import static
 from paddle.base import dygraph
+from paddle.pir_utils import test_with_pir_api
 
 paddle.enable_static()
 
@@ -45,26 +46,29 @@ def setUp(self):
         self.outputs = {'Out': out_ref}
 
     def test_check_output(self):
-        self.check_output()
+        self.check_output(check_pir=True)
 
     def test_check_grad(self):
         self.check_grad(
             ['X', 'Y'],
             'Out',
+            check_pir=True,
         )
 
     def test_check_grad_ignore_x(self):
         self.check_grad(
             ['Y'],
             'Out',
             no_grad_set=set('X'),
+            check_pir=True,
         )
 
     def test_check_grad_ignore_y(self):
         self.check_grad(
             ['X'],
             'Out',
             no_grad_set=set('Y'),
+            check_pir=True,
         )
 
 
@@ -102,6 +106,7 @@ def test_dygraph(self):
             out_np = paddle.complex(x, y).numpy()
         np.testing.assert_allclose(self.out, out_np, rtol=1e-05)
 
+    @test_with_pir_api
     def test_static(self):
         mp, sp = static.Program(), static.Program()
         with static.program_guard(mp, sp):

test/legacy_test/test_logspace.py

@@ -19,6 +19,7 @@
 
 import paddle
 from paddle.base import core
+from paddle.pir_utils import test_with_pir_api
 
 
 class TestLogspaceOpCommonCase(OpTest):
@@ -39,7 +40,7 @@ def init_data(self):
         self.outputs = {'Out': np.power(2, np.arange(0, 11)).astype(dtype)}
 
     def test_check_output(self):
-        self.check_output()
+        self.check_output(check_pir=True)
 
 
 class TestLogspaceFP16Op(TestLogspaceOpCommonCase):
@@ -87,7 +88,7 @@ def init_data(self):
         self.place = core.CUDAPlace(0)
 
     def test_check_output(self):
-        self.check_output_with_place(self.place)
+        self.check_output_with_place(self.place, check_pir=True)
 
 
 class TestLogspaceOpReverseCase(TestLogspaceOpCommonCase):
@@ -143,6 +144,7 @@ def init_data(self):
 
 
 class TestLogspaceAPI(unittest.TestCase):
+    @test_with_pir_api
     def test_variable_input1(self):
         paddle.enable_static()
         prog = paddle.static.Program()
@@ -170,6 +172,7 @@ def test_variable_input2(self):
         self.assertEqual((out.numpy() == np_res).all(), True)
         paddle.enable_static()
 
+    @test_with_pir_api
     def test_dtype(self):
         paddle.enable_static()
         prog = paddle.static.Program()

test/legacy_test/test_meshgrid_op.py

@@ -20,6 +20,7 @@
 import paddle
 from paddle import base
 from paddle.base import core
+from paddle.pir_utils import test_with_pir_api
 
 
 def meshgrid_wrapper(x):
@@ -41,10 +42,12 @@ def init_data_type(self):
         self.dtype = np.float64
 
     def test_check_output(self):
-        self.check_output(check_prim=True)
+        self.check_output(check_prim=True, check_pir=True)
 
     def test_check_grad(self):
-        self.check_grad(['x0'], ['out0', 'out1'], check_prim=True)
+        self.check_grad(
+            ['x0'], ['out0', 'out1'], check_prim=True, check_pir=True
+        )
 
     def init_inputs_and_outputs(self):
         self.shape = self.get_x_shape()
@@ -122,15 +125,20 @@ def if_enable_cinn(self):
         self.enable_cinn = False
 
     def test_check_output(self):
-        self.check_output_with_place(place=paddle.CUDAPlace(0))
+        self.check_output_with_place(place=paddle.CUDAPlace(0), check_pir=True)
 
     def test_check_grad(self):
         self.check_grad_with_place(
-            paddle.CUDAPlace(0), ['x0'], ['out0', 'out1'], check_prim=True
+            paddle.CUDAPlace(0),
+            ['x0'],
+            ['out0', 'out1'],
+            check_prim=True,
+            check_pir=True,
         )
 
 
 class TestMeshgridOp3(unittest.TestCase):
+    @test_with_pir_api
     def test_api(self):
         x = paddle.static.data(shape=[100], dtype='int32', name='x')
         y = paddle.static.data(shape=[200], dtype='int32', name='y')
@@ -167,6 +175,7 @@ def test_api(self):
 
 
 class TestMeshgridOp4(unittest.TestCase):
+    @test_with_pir_api
     def test_list_input(self):
         x = paddle.static.data(shape=[100], dtype='int32', name='x')
         y = paddle.static.data(shape=[200], dtype='int32', name='y')
@@ -204,6 +213,7 @@ def test_list_input(self):
 
 
 class TestMeshgridOp5(unittest.TestCase):
+    @test_with_pir_api
     def test_tuple_input(self):
         x = paddle.static.data(shape=[100], dtype='int32', name='x')
         y = paddle.static.data(shape=[200], dtype='int32', name='y')