[PIR] add all paddle.dot pir test (#58081)

MarioLulab · web-flow · commit ff4d772ed3cd · 2023-10-17T11:03:35.000+08:00
* add dot new ir test

* supply full dot op test

* replace check_new_ir
diff --git a/test/legacy_test/test_dot_op.py b/test/legacy_test/test_dot_op.py
@@ -230,28 +230,30 @@ def test_check_output(self):
         if core.is_compiled_with_cuda():
             place = core.CUDAPlace(0)
             if core.is_float16_supported(place):
-                self.check_output_with_place(place, atol=0.125)
+                self.check_output_with_place(place, atol=0.125, check_pir=True)
 
     def test_check_grad_normal(self):
         if core.is_compiled_with_cuda():
             place = core.CUDAPlace(0)
             if core.is_float16_supported(place):
-                self.check_grad_with_place(place, ['X', 'Y'], 'Out')
+                self.check_grad_with_place(
+                    place, ['X', 'Y'], 'Out', check_pir=True
+                )
 
     def test_check_grad_ingore_x(self):
         if core.is_compiled_with_cuda():
             place = core.CUDAPlace(0)
             if core.is_float16_supported(place):
                 self.check_grad_with_place(
-                    place, ['Y'], 'Out', no_grad_set=set("X")
+                    place, ['Y'], 'Out', no_grad_set=set("X"), check_pir=True
                 )
 
     def test_check_grad_ingore_y(self):
         if core.is_compiled_with_cuda():
             place = core.CUDAPlace(0)
             if core.is_float16_supported(place):
                 self.check_grad_with_place(
-                    place, ['X'], 'Out', no_grad_set=set("Y")
+                    place, ['X'], 'Out', no_grad_set=set("Y"), check_pir=True
                 )
 
     def init_input_output(self):
@@ -302,7 +304,7 @@ def test_check_output(self):
         if core.is_compiled_with_cuda():
             place = core.CUDAPlace(0)
             if core.is_bfloat16_supported(place):
-                self.check_output_with_place(place, atol=0.5)
+                self.check_output_with_place(place, atol=0.5, check_pir=True)
 
     def test_check_grad_normal(self):
         if core.is_compiled_with_cuda():
@@ -313,6 +315,7 @@ def test_check_grad_normal(self):
                     ['X', 'Y'],
                     'Out',
                     user_defined_grads=[self.inputs['Y'], self.inputs['X']],
+                    check_pir=True,
                 )
 
     def test_check_grad_ingore_x(self):
@@ -325,6 +328,7 @@ def test_check_grad_ingore_x(self):
                     'Out',
                     no_grad_set=set("X"),
                     user_defined_grads=[self.inputs['X']],
+                    check_pir=True,
                 )
 
     def test_check_grad_ingore_y(self):
@@ -337,6 +341,7 @@ def test_check_grad_ingore_y(self):
                     'Out',
                     no_grad_set=set("Y"),
                     user_defined_grads=[self.inputs['Y']],
+                    check_pir=True,
                 )
 
     def init_input_output(self):
@@ -374,6 +379,7 @@ def test_check_grad_normal(self):
                         self.y / self.y.shape[0],
                         self.x / self.x.shape[0],
                     ],
+                    check_pir=True,
                 )
 
     def test_check_grad_ingore_x(self):
@@ -386,6 +392,7 @@ def test_check_grad_ingore_x(self):
                     'Out',
                     no_grad_set=set("X"),
                     user_defined_grads=[self.x / self.x.shape[0]],
+                    check_pir=True,
                 )
 
     def test_check_grad_ingore_y(self):
@@ -398,6 +405,7 @@ def test_check_grad_ingore_y(self):
                     'Out',
                     no_grad_set=set("Y"),
                     user_defined_grads=[self.y / self.y.shape[0]],
+                    check_pir=True,
                 )
 
 
