fix

PaddlePaddle · Jan 16, 2024 · 7616fcb · 7616fcb
1 parent 72c58f8
commit 7616fcb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/test/legacy_test/test_arange.py b/test/legacy_test/test_arange.py
@@ -132,6 +132,7 @@ def init_config(self):
 
 
 class TestArangeOpError(unittest.TestCase):
+    @test_with_pir_api
     def test_static_errors(self):
         with program_guard(Program(), Program()):
             paddle.enable_static()

diff --git a/test/legacy_test/test_compare_op.py b/test/legacy_test/test_compare_op.py
@@ -524,6 +524,7 @@ def test_check_output(self):
 
 
 class TestCompareOpError(unittest.TestCase):
+    @test_with_pir_api
     def test_int16_support(self):
         paddle.enable_static()
         with paddle.static.program_guard(