From 22f5a85133016ae69123d3c14654892949d0731a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BC=A0=E6=98=A5=E4=B9=94?= <83450930+Liyulingyue@users.noreply.github.com> Date: Wed, 15 Nov 2023 08:31:14 +0800 Subject: [PATCH] Update test_llm_int8_linear.py --- test/quantization/test_llm_int8_linear.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/quantization/test_llm_int8_linear.py b/test/quantization/test_llm_int8_linear.py index 4817f1384ed7c..5a35b0d512461 100644 --- 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(