[PIR]Open uts for 241-250 (#60817)

0x45f · web-flow · commit d38d7e0dd688 · 2024-01-18T14:16:00.000+08:00
diff --git a/test/legacy_test/test_unique_consecutive_op.py b/test/legacy_test/test_unique_consecutive_op.py
@@ -100,7 +100,7 @@ def setUp(self):
         }
 
     def test_check_output(self):
-        self.check_output()
+        self.check_output(check_pir=True)
 
 
 class TestUniqueConsecutiveOp2(TestUniqueConsecutiveOp):
@@ -130,7 +130,7 @@ def setUp(self):
             'return_inverse': self.return_inverse,
             'dtype': paddle.int32,
         }
-        self.python_out_sig = ["Out"]
+        self.python_out_sig = ["Out", "Index"]
         self.outputs = {'Out': result, 'Index': inverse}
 
 
@@ -161,7 +161,7 @@ def setUp(self):
             'return_counts': self.return_counts,
             'dtype': paddle.int32,
         }
-        self.python_out_sig = ["Out"]
+        self.python_out_sig = ["Out", "Counts"]
         self.outputs = {'Out': result, 'Counts': counts}
 
 
@@ -194,7 +194,7 @@ def setUp(self):
             'return_counts': self.return_counts,
             'dtype': paddle.int32,
         }
-        self.python_out_sig = ["Out"]
+        self.python_out_sig = ["Out", "Index", "Counts"]
         self.outputs = {'Out': result, 'Index': inverse, 'Counts': counts}
 
 
@@ -204,9 +204,10 @@ def setUp(self):
         if core.is_compiled_with_cuda():
             self.places.append(base.CUDAPlace(0))
 
-    @test_with_pir_api
     def check_static_result(self, place):
-        with base.program_guard(base.Program(), base.Program()):
+        with paddle.static.program_guard(
+            paddle.static.Program(), paddle.static.Program()
+        ):
             paddle.enable_static()
             input_x = paddle.static.data(
                 name="input_x",
@@ -223,6 +224,7 @@ def check_static_result(self, place):
                 fetch_list=[result],
             )
 
+    @test_with_pir_api
     def test_static(self):
         for place in self.places:
             self.check_static_result(place=place)
@@ -241,9 +243,10 @@ def setUp(self):
         if core.is_compiled_with_cuda():
             self.places.append(base.CUDAPlace(0))
 
-    @test_with_pir_api
     def check_static_result(self, place):
-        with base.program_guard(base.Program(), base.Program()):
+        with paddle.static.program_guard(
+            paddle.static.Program(), paddle.static.Program()
+        ):
             paddle.enable_static()
             input_x = paddle.static.data(
                 name="input_x",
@@ -262,6 +265,7 @@ def check_static_result(self, place):
                 fetch_list=[result],
             )
 
+    @test_with_pir_api
     def test_static(self):
         for place in self.places:
             self.check_static_result(place=place)
@@ -282,9 +286,10 @@ def setUp(self):
         if core.is_compiled_with_cuda():
             self.places.append(base.CUDAPlace(0))
 
-    @test_with_pir_api
     def check_static_result(self, place):
-        with base.program_guard(base.Program(), base.Program()):
+        with paddle.static.program_guard(
+            paddle.static.Program(), paddle.static.Program()
+        ):
             paddle.enable_static()
             input_x = paddle.static.data(
                 name="input_x",
@@ -303,6 +308,7 @@ def check_static_result(self, place):
                 fetch_list=[result],
             )
 
+    @test_with_pir_api
     def test_static(self):
         for place in self.places:
             self.check_static_result(place=place)
diff --git a/test/legacy_test/test_variable_length_memory_efficient_attention.py b/test/legacy_test/test_variable_length_memory_efficient_attention.py
@@ -25,7 +25,6 @@
     variable_length_memory_efficient_attention,
 )
 from paddle.pir_utils import test_with_pir_api
-from paddle.static import Program, program_guard
 
 paddle.seed(2023)
 
@@ -306,7 +305,9 @@ def setUp(self):
     @test_with_pir_api
     def test_all(self):
         paddle.enable_static()
-        with program_guard(Program(), Program()):
+        with paddle.static.program_guard(
+            paddle.static.Program(), paddle.static.Program()
+        ):
             q = paddle.static.data(
                 name="query", shape=self.shape, dtype=self.dtype
             )
diff --git a/test/legacy_test/test_viterbi_decode_op.py b/test/legacy_test/test_viterbi_decode_op.py
@@ -122,10 +122,11 @@ def setUp(self):
         decoder = Decoder(self.transitions, self.use_tag)
         self.scores, self.path = decoder(self.input, self.length)
 
-    @test_with_pir_api
     def check_static_result(self, place):
         bz, length, ntags = self.bz, self.len, self.ntags
-        with base.program_guard(base.Program(), base.Program()):
+        with paddle.static.program_guard(
+            paddle.static.Program(), paddle.static.Program()
+        ):
             Input = paddle.static.data(
                 name="Input", shape=[bz, length, ntags], dtype="float32"
             )
@@ -147,6 +148,7 @@ def check_static_result(self, place):
             np.testing.assert_allclose(fetches[0], self.scores, rtol=1e-5)
             np.testing.assert_allclose(fetches[1], self.path)
 
+    @test_with_pir_api
     def test_static_net(self):
         for place in self.places:
             self.check_static_result(place)
diff --git a/test/legacy_test/test_warpctc_op.py b/test/legacy_test/test_warpctc_op.py
@@ -22,6 +22,7 @@
 import paddle
 import paddle.nn.functional as F
 from paddle.base import core
+from paddle.pir_utils import test_with_pir_api
 
 CUDA_BLOCK_SIZE = 32
 
@@ -526,6 +527,7 @@ def test_check_grad(self):
 
 
 class TestWarpCTCOpError(unittest.TestCase):
+    @test_with_pir_api
     def test_errors(self):
         paddle.enable_static()
         main_program = paddle.static.Program()
diff --git a/test/legacy_test/test_warprnnt_op.py b/test/legacy_test/test_warprnnt_op.py
@@ -19,7 +19,8 @@
 
 import paddle
 from paddle import _C_ops
-from paddle.base import Program, core, program_guard
+from paddle.base import core
+from paddle.pir_utils import test_with_pir_api
 
 paddle.enable_static()
 
@@ -260,12 +261,11 @@ def test_check_grad(self):
 
 
 class TestWarpRNNTOpError(unittest.TestCase):
-    def test_errors(self):
-        print("test_errors")
-        with program_guard(Program(), Program()):
-            logits = paddle.static.data(
-                name='input', shape=[5, 16, 6], dtype='float32'
-            )
+    @test_with_pir_api
+    def test_errors1(self):
+        with paddle.static.program_guard(
+            paddle.static.Program(), paddle.static.Program()
+        ):
             logits_length = paddle.static.data(
                 name='logit_lengths', shape=[None], dtype='int32'
             )
@@ -289,6 +289,23 @@ def test_logits_Variable():
 
             self.assertRaises(TypeError, test_logits_Variable)
 
+    def test_errors2(self):
+        with paddle.static.program_guard(
+            paddle.static.Program(), paddle.static.Program()
+        ):
+            logits = paddle.static.data(
+                name='input', shape=[5, 16, 6], dtype='float32'
+            )
+            logits_length = paddle.static.data(
+                name='logit_lengths', shape=[None], dtype='int32'
+            )
+            label = paddle.static.data(
+                name='labels', shape=[16, 3], dtype='int32'
+            )
+            label_length = paddle.static.data(
+                name='label_lengths', shape=[None], dtype='int32'
+            )
+
             def test_label_Variable():
                 label_data = paddle.static.data(
                     name='label_data', shape=[16, 3], dtype='int64'
