fix test_quant_linear_op in pir (#67487)

wanghuancoder · web-flow · commit 93b8aa6ea9c8 · 2024-08-19T09:44:37.000+08:00
diff --git a/test/legacy_test/test_quant_linear_op.py b/test/legacy_test/test_quant_linear_op.py
@@ -544,13 +544,13 @@ def config(self):
 class TestQuantLinearOp_NumFlattenDims_NegOne(unittest.TestCase):
     def test_api(self):
         def run_program(num_flatten_dims):
-            paddle.seed(SEED)
-            np.random.seed(SEED)
-            startup_program = Program()
-            main_program = Program()
-
             with paddle.pir_utils.OldIrGuard():
-                with program_guard(main_program, startup_program):
+                paddle.seed(SEED)
+                np.random.seed(SEED)
+                startup_program = paddle.base.Program()
+                main_program = paddle.base.Program()
+
+                with paddle.base.program_guard(main_program, startup_program):
                     quant_round_type = 0
                     quant_max_bound = 127.0
                     quant_min_bound = -127.0
