【PIR API adaptor No.59, 61, 63】Migrate some ops into pir (#58697)

* migrate dist, eigh and eigvalsh into pir

* fix bug

* fix bug

Author: longranger2

python/paddle/tensor/linalg.py

@@ -717,7 +717,7 @@ def dist(x, y, p=2, name=None):
             Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True,
             0.)
     """
-    if in_dynamic_mode():
+    if in_dynamic_or_pir_mode():
         return _C_ops.dist(x, y, p)
 
     check_variable_and_dtype(
@@ -2818,7 +2818,7 @@ def eigh(x, UPLO='L', name=None):
              [ 0.3826833963394165j    , -0.9238795042037964j    ]])
 
     """
-    if in_dynamic_mode():
+    if in_dynamic_or_pir_mode():
         return _C_ops.eigh(x, UPLO)
     else:
 
@@ -3324,7 +3324,7 @@ def eigvalsh(x, UPLO='L', name=None):
             Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True,
             [0.17157286, 5.82842731])
     """
-    if in_dynamic_mode():
+    if in_dynamic_or_pir_mode():
         values, _ = _C_ops.eigvalsh(x, UPLO, x.stop_gradient)
         return values
     else:

test/legacy_test/test_dist_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
 
 paddle.enable_static()
 
@@ -113,13 +114,11 @@ def get_reduce_dims(x, y):
         return x_grad, y_grad
 
     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",
-            user_defined_grads=self.gradient,
+            ["X", "Y"], "Out", user_defined_grads=self.gradient, check_pir=True
         )
 
 
@@ -244,10 +243,11 @@ def init_data_type(self):
             'float32' if core.is_compiled_with_rocm() else 'float64'
         )
 
+    @test_with_pir_api
     def test_api(self):
         self.init_data_type()
-        main_program = base.Program()
-        startup_program = base.Program()
+        main_program = paddle.static.Program()
+        startup_program = paddle.static.Program()
         with base.program_guard(main_program, startup_program):
             x = paddle.static.data(
                 name='x', shape=[2, 3, 4, 5], dtype=self.data_type
@@ -266,7 +266,7 @@ def test_api(self):
             )
             exe = base.Executor(place)
             out = exe.run(
-                base.default_main_program(),
+                main_program,
                 feed={'x': x_i, 'y': y_i},
                 fetch_list=[result],
             )

test/legacy_test/test_eigh_op.py

@@ -18,6 +18,7 @@
 from op_test import OpTest
 
 import paddle
+from paddle.pir_utils import test_with_pir_api
 
 
 def valid_eigh_result(A, eigh_value, eigh_vector, uplo):
@@ -92,7 +93,7 @@ def init_input(self):
     #     self.check_output(no_check_set=['Eigenvectors'])
 
     def test_grad(self):
-        self.check_grad(["X"], ["Eigenvalues"])
+        self.check_grad(["X"], ["Eigenvalues"], check_pir=True)
 
 
 class TestEighUPLOCase(TestEighOp):
@@ -183,6 +184,7 @@ def check_static_complex_result(self):
             )
             valid_eigh_result(self.complex_symm, actual_w, actual_v, self.UPLO)
 
+    @test_with_pir_api
     def test_in_static_mode(self):
         paddle.enable_static()
         self.check_static_float_result()

test/legacy_test/test_eigvalsh_op.py

@@ -18,6 +18,7 @@
 from op_test import OpTest
 
 import paddle
+from paddle.pir_utils import test_with_pir_api
 
 
 def compare_result(actual, expected):
@@ -72,10 +73,10 @@ def init_input(self):
 
     def test_check_output(self):
         # Vectors in posetive or negative is equivalent
-        self.check_output(no_check_set=['Eigenvectors'])
+        self.check_output(no_check_set=['Eigenvectors'], check_pir=True)
 
     def test_grad(self):
-        self.check_grad(["X"], ["Eigenvalues"])
+        self.check_grad(["X"], ["Eigenvalues"], check_pir=True)
 
 
 class TestEigvalshUPLOCase(TestEigvalshOp):
@@ -166,6 +167,7 @@ def check_static_complex_result(self):
             expected_w = np.linalg.eigvalsh(self.complex_symm)
             compare_result(actual_w[0], expected_w)
 
+    @test_with_pir_api
     def test_in_static_mode(self):
         paddle.enable_static()
         self.check_static_float_result()