[PIR]Migrate remainder into pir (#58139)

0x45f · web-flow · commit 6d4744b63474 · 2023-10-17T19:12:09.000+08:00
diff --git a/python/paddle/tensor/math.py b/python/paddle/tensor/math.py
@@ -998,7 +998,7 @@ def remainder(x, y, name=None):
             [0, 3, 2, 1])
 
     """
-    if in_dynamic_mode():
+    if in_dynamic_or_pir_mode():
         return _C_ops.remainder(x, y)
     else:
         return _elementwise_op(LayerHelper('elementwise_mod', **locals()))
diff --git a/test/legacy_test/test_elementwise_mod_op.py b/test/legacy_test/test_elementwise_mod_op.py
@@ -45,9 +45,9 @@ def setUp(self):
 
     def test_check_output(self):
         if self.attrs['axis'] == -1:
-            self.check_output()
+            self.check_output(check_pir=True)
         else:
-            self.check_output()
+            self.check_output(check_pir=True)
 
     def init_input_output(self):
         self.x = np.random.uniform(0, 10000, [10, 10]).astype(self.dtype)
@@ -102,9 +102,9 @@ def init_input_output(self):
 
     def test_check_output(self):
         if self.attrs['axis'] == -1:
-            self.check_output()
+            self.check_output(check_pir=True)
         else:
-            self.check_output()
+            self.check_output(check_pir=True)
 
 
 @unittest.skipIf(
@@ -121,9 +121,9 @@ def init_input_output(self):
 
     def test_check_output(self):
         if self.attrs['axis'] == -1:
-            self.check_output()
+            self.check_output(check_pir=True)
         else:
-            self.check_output()
+            self.check_output(check_pir=True)
 
 
 class TestElementwiseModFP16Op_ZeroDim1(TestElementwiseModFP16Op):
@@ -181,7 +181,7 @@ def setUp(self):
 
     def test_check_output(self):
         place = core.CUDAPlace(0)
-        self.check_output_with_place(place)
+        self.check_output_with_place(place, check_pir=True)
 
     def init_dtype(self):
         self.dtype = np.uint16
