[NewIR] No.19 Migrate paddle.reduce_max into pir (#57123)

PommesPeter · web-flow · commit a8c4195d65b8 · 2023-09-14T18:30:11.000+08:00
* refactor: added new_ir mode in reduce_max

* fix: updated code style

* fix: updated reduce_max test func 'check_output'

* refactor: rename new ir mode

* refactor: rename new ir mode
diff --git a/python/paddle/tensor/math.py b/python/paddle/tensor/math.py
@@ -2807,7 +2807,7 @@ def max(x, axis=None, keepdim=False, name=None):
               [1., 1.]]])
     """
 
-    if in_dynamic_mode():
+    if in_dynamic_or_pir_mode():
         return _C_ops.max(x, axis, keepdim)
     else:
         reduce_all, axis = _get_reduce_axis_with_tensor(axis, x)
diff --git a/test/legacy_test/test_reduce_op.py b/test/legacy_test/test_reduce_op.py
@@ -278,7 +278,7 @@ def setUp(self):
         }
 
     def test_check_output(self):
-        self.check_output()
+        self.check_output(check_new_ir=True)
 
     def test_check_grad(self):
         # only composite op support gradient check of reduce_max
@@ -312,7 +312,7 @@ def init_inputs_and_outputs(self):
         }
 
     def test_check_output(self):
-        self.check_output()
+        self.check_output(check_new_ir=True)
 
     def test_check_grad(self):
         # only composite op support gradient check of reduce_max
@@ -365,7 +365,7 @@ def if_enable_cinn(self):
         pass
 
     def test_check_output(self):
-        self.check_output()
+        self.check_output(check_new_ir=True)
 
     def test_check_grad(self):
         # only composite op support gradient check of reduce_max
@@ -399,7 +399,7 @@ def if_enable_cinn(self):
         self.enable_cinn = False
 
     def test_check_output(self):
-        self.check_output_with_place(core.CUDAPlace(0))
+        self.check_output_with_place(core.CUDAPlace(0), check_new_ir=True)
 
     def test_check_grad(self):
         # only composite op support gradient check of reduce_max
@@ -1298,7 +1298,7 @@ def setUp(self):
         }
 
     def test_check_output(self):
-        self.check_output()
+        self.check_output(check_new_ir=True)
 
     def test_check_grad(self):
         # only composite op support gradient check of reduce_max
