Update test_llm_int8_linear.py

PaddlePaddle · Nov 15, 2023 · 22f5a85 · 22f5a85
1 parent e125c34
commit 22f5a85
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/quantization/test_llm_int8_linear.py b/test/quantization/test_llm_int8_linear.py
@@ -90,9 +90,9 @@ def get_llm_int8_linear_out(self):
     @test_with_pir_api
     def get_llm_int8_linear_out_static(self):
         paddle.enable_static()
-        main = base.Program()
-        start = base.Program()
-        with base.program_guard(main, start):
+        main = base.static.Program()
+        start = base.static.Program()
+        with base.static.program_guard(main, start):
             x = paddle.static.data("x", self.x.shape, dtype=self.x.dtype)
 
             weight = paddle.static.data(