[PIR]Migrate increment into pir (#58245)

0x45f · web-flow · commit 0b7d1d26d51a · 2023-10-23T14:29:51.000+08:00
diff --git a/python/paddle/tensor/math.py b/python/paddle/tensor/math.py
@@ -4681,7 +4681,7 @@ def increment(x, value=1.0, name=None):
             [1.])
 
     """
-    if in_dynamic_mode():
+    if in_dynamic_or_pir_mode():
         return _C_ops.increment_(x, value)
     else:
         check_variable_and_dtype(
diff --git a/test/legacy_test/test_increment.py b/test/legacy_test/test_increment.py
@@ -18,9 +18,11 @@
 
 import paddle
 from paddle import base
+from paddle.pir_utils import test_with_pir_api
 
 
 class TestIncrement(unittest.TestCase):
+    @test_with_pir_api
     def test_api(self):
         with base.program_guard(base.Program(), base.Program()):
             input = paddle.tensor.fill_constant(
@@ -41,6 +43,7 @@ def test_api(self):
 
 
 class TestInplaceApiWithDataTransform(unittest.TestCase):
+    @test_with_pir_api
     def test_increment(self):
         if base.core.is_compiled_with_cuda():
             paddle.enable_static()
