fix xpu case (#66227)

wanghuancoder · web-flow · commit c49014e9b87c · 2024-07-19T12:39:21.000+08:00
diff --git a/test/xpu/test_pad_op_xpu.py b/test/xpu/test_pad_op_xpu.py
@@ -140,7 +140,8 @@ def test_static(self):
 
                     sgd = paddle.optimizer.SGD()
                     sgd.minimize(paddle.mean(out))
-                    self.assertTrue(self.var_prefix() in str(main_prog))
+                    if not paddle.framework.use_pir_api():
+                        self.assertTrue(self.var_prefix() in str(main_prog))
                     exe = paddle.static.Executor(paddle.XPUPlace(0))
                     exe.run(startup_prog)
                     res = exe.run(fetch_list=[feat, out])
diff --git a/test/xpu/test_prelu_op_xpu.py b/test/xpu/test_prelu_op_xpu.py
@@ -24,7 +24,6 @@
 
 import paddle
 from paddle import base
-from paddle.base import Program
 
 paddle.enable_static()
 
@@ -175,26 +174,28 @@ def setUp(self):
         self.x_np = np.ones([1, 2, 3, 4]).astype('float32')
 
     def test_mode_error(self):
-        main_program = Program()
-        with base.program_guard(main_program, Program()):
-            x = paddle.static.data(name='x', shape=[2, 3, 4, 5])
-            try:
-                y = prelu_t(x, 'any')
-            except Exception as e:
-                assert e.args[0].find('InvalidArgument') != -1
+        with paddle.pir_utils.OldIrGuard():
+            main_program = paddle.base.Program()
+            with base.program_guard(main_program, paddle.base.Program()):
+                x = paddle.static.data(name='x', shape=[2, 3, 4, 5])
+                try:
+                    y = prelu_t(x, 'any')
+                except Exception as e:
+                    assert e.args[0].find('InvalidArgument') != -1
 
     def test_data_format_error1(self):
-        main_program = Program()
-        with base.program_guard(main_program, Program()):
-            x = paddle.static.data(name='x', shape=[2, 3, 4, 5])
-            try:
-                y = prelu_t(x, 'channel', data_format='N')
-            except Exception as e:
-                assert e.args[0].find('InvalidArgument') != -1
+        with paddle.pir_utils.OldIrGuard():
+            main_program = paddle.base.Program()
+            with base.program_guard(main_program, paddle.base.Program()):
+                x = paddle.static.data(name='x', shape=[2, 3, 4, 5])
+                try:
+                    y = prelu_t(x, 'channel', data_format='N')
+                except Exception as e:
+                    assert e.args[0].find('InvalidArgument') != -1
 
     def test_data_format_error2(self):
-        main_program = Program()
-        with base.program_guard(main_program, Program()):
+        main_program = paddle.base.Program()
+        with base.program_guard(main_program, paddle.base.Program()):
             x = paddle.static.data(name='x', shape=[2, 3, 4, 5])
             try:
                 y = paddle.static.nn.prelu(x, 'channel', data_format='N')
