[NewIR] No.19 Migrate paddle.reduce_max into pir

python/paddle/tensor/math.py

@@ -2810,7 +2810,7 @@ def max(x, axis=None, keepdim=False, name=None):
               [1., 1.]]])
     """
 
-    if in_dynamic_mode():
+    if in_dynamic_or_new_ir_mode():
         return _C_ops.max(x, axis, keepdim)
     else:
         reduce_all, axis = _get_reduce_axis_with_tensor(axis, x)

test/legacy_test/test_reduce_op.py

@@ -56,6 +56,9 @@ def calc_output(self):
     def test_check_output(self):
         self.check_output()
 
+    def test_check_output_new_ir(self):
+        self.check_output(check_new_ir=True)
+
     def test_check_grad(self):
         self.check_grad(['X'], 'Out', check_prim=True)