[Dy2St] Enable dy2st ut test_assert in PIR mode (#60216)

PaddlePaddle · Dec 21, 2023 · 582da31 · 582da31
1 parent a75c5eb
commit 582da31
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test/dygraph_to_static/test_assert.py b/test/dygraph_to_static/test_assert.py
@@ -18,6 +18,7 @@
 from dygraph_to_static_utils import (
     Dy2StTestBase,
     enable_to_static_guard,
+    test_legacy_and_pt_and_pir,
 )
 
 import paddle
@@ -48,6 +49,7 @@ def _run_dy_static(self, func, x, with_exception):
         self._run(func, x, with_exception, True)
         self._run(func, x, with_exception, False)
 
+    @test_legacy_and_pt_and_pir
     def test_non_variable(self):
         self._run_dy_static(
             paddle.jit.to_static(dyfunc_assert_non_variable),
@@ -60,6 +62,7 @@ def test_non_variable(self):
             with_exception=False,
         )
 
+    @test_legacy_and_pt_and_pir
     def test_bool_variable(self):
         self._run_dy_static(
             paddle.jit.to_static(dyfunc_assert_variable),
@@ -72,6 +75,7 @@ def test_bool_variable(self):
             with_exception=False,
         )
 
+    @test_legacy_and_pt_and_pir
     def test_int_variable(self):
         self._run_dy_static(
             paddle.jit.to_static(dyfunc_assert_variable),