[PIR] Open some cf ut for pir api (#61182)

zhangbo9674 · web-flow · commit f9b8c6c9bbe7 · 2024-01-29T11:23:43.000+08:00
* fix

* fix
diff --git a/test/dygraph_to_static/test_break_continue.py b/test/dygraph_to_static/test_break_continue.py
@@ -274,7 +274,7 @@ class TestBreakContinueInFor(TestContinueBase):
     def init_dygraph_func(self):
         self.dygraph_func = test_break_continue_in_for
 
-    @test_legacy_and_pt
+    @test_legacy_and_pt_and_pir
     def test_transformed_static_result(self):
         self.init_dygraph_func()
         dygraph_res = self.run_dygraph_mode()
@@ -296,8 +296,7 @@ class TestContinueInWhile(TestContinueNotPirBase):
     def init_dygraph_func(self):
         self.dygraph_func = test_continue_in_while
 
-    # TODO(dev): Remove this after fix PT Rename issue
-    @test_legacy_and_pt
+    @test_legacy_and_pt_and_pir
     def test_transformed_static_result(self):
         self.init_dygraph_func()
         dygraph_res = self.run_dygraph_mode()
@@ -314,8 +313,7 @@ class TestBreakInWhile(TestContinueInWhile):
     def init_dygraph_func(self):
         self.dygraph_func = test_break_in_while
 
-    # TODO(dev): Remove this after fix PT Rename issue
-    @test_legacy_and_pt
+    @test_legacy_and_pt_and_pir
     def test_transformed_static_result(self):
         self.init_dygraph_func()
         dygraph_res = self.run_dygraph_mode()
@@ -332,7 +330,7 @@ class TestWhileLoopClassVar(TestContinueInWhile):
     def init_dygraph_func(self):
         self.dygraph_func = while_loop_class_var
 
-    @test_legacy_and_pt
+    @test_legacy_and_pt_and_pir
     def test_transformed_static_result(self):
         self.init_dygraph_func()
         dygraph_res = self.run_dygraph_mode()
@@ -356,7 +354,7 @@ class TestOptimBreakInWhile(TestContinueInWhile):
     def init_dygraph_func(self):
         self.dygraph_func = test_optim_break_in_while
 
-    # TODO(dev): Remove this after fix PT Rename issue
+    # TODO: Open PIR test when while_loop dy2st fixed
     @test_legacy_and_pt
     def test_transformed_static_result(self):
         self.init_dygraph_func()
diff --git a/test/dygraph_to_static/test_jit_setitem.py b/test/dygraph_to_static/test_jit_setitem.py
@@ -18,7 +18,6 @@
 import numpy as np
 from dygraph_to_static_utils import (
     Dy2StTestBase,
-    test_legacy_and_pt,
     test_legacy_and_pt_and_pir,
 )
 
@@ -203,8 +202,6 @@ def run_dygraph(self, func):
         y = func()
         return (y,)
 
-    # TODO: Open PIR test when while_loop dy2st fixed
-    @test_legacy_and_pt
     def test_case(self):
         func = self.init_func()
         dy_res = self.run_dygraph(func)
