[NPU][MLU] Fix test_cumxxx_op on NPU (#1608)

Author: LittleHeroZZZX

backends/npu/tests/unittests/test_cumprod_op_npu.py

@@ -67,7 +67,7 @@ def run_static(self, use_custom_device=False):
             exe.run(base.default_startup_program())
             out = exe.run(
                 feed={"X": data_np},
-                fetch_list=[y.name, y2.name, y3.name, y4.name, y5.name, y6.name],
+                fetch_list=[y, y2, y3, y4, y5, y6],
             )
 
             z = np.cumprod(data_np, axis=0)
@@ -87,10 +87,11 @@ def test_npu(self):
         self.run_static(use_custom_device=True)
 
     def test_name(self):
-        with base.program_guard(base.Program()):
-            x = paddle.static.data("x", [3, 4])
-            y = paddle.cumprod(x, dim=0, name="out")
-            self.assertTrue("out" in y.name)
+        with paddle.pir_utils.OldIrGuard():
+            with base.program_guard(base.Program()):
+                x = paddle.static.data("x", [3, 4])
+                y = paddle.cumprod(x, dim=0, name="out")
+                self.assertTrue("out" in y.name)
 
 
 if __name__ == "__main__":

backends/npu/tests/unittests/test_cumsum_op_npu.py

@@ -68,7 +68,7 @@ def run_static(self, use_custom_device=False):
             exe.run(base.default_startup_program())
             out = exe.run(
                 feed={"X": data_np},
-                fetch_list=[y.name, y2.name, y3.name, y4.name, y5.name, y6.name],
+                fetch_list=[y, y2, y3, y4, y5, y6],
             )
 
             z = np.cumsum(data_np)
@@ -88,10 +88,11 @@ def test_npu(self):
         self.run_static(use_custom_device=True)
 
     def test_name(self):
-        with base.program_guard(base.Program()):
-            x = paddle.static.data("x", [3, 4])
-            y = paddle.cumsum(x, name="out")
-            self.assertTrue("out" in y.name)
+        with paddle.pir_utils.OldIrGuard():
+            with base.program_guard(base.Program()):
+                x = paddle.static.data("x", [3, 4])
+                y = paddle.cumsum(x, name="out")
+                self.assertTrue("out" in y.name)
 
 
 class TestNPUCumSumOp1(OpTest):