Skip to content

Commit 22f5a85

Browse files
authored
Update test_llm_int8_linear.py
1 parent e125c34 commit 22f5a85

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/quantization/test_llm_int8_linear.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -90,9 +90,9 @@ def get_llm_int8_linear_out(self):
9090
@test_with_pir_api
9191
def get_llm_int8_linear_out_static(self):
9292
paddle.enable_static()
93-
main = base.Program()
94-
start = base.Program()
95-
with base.program_guard(main, start):
93+
main = base.static.Program()
94+
start = base.static.Program()
95+
with base.static.program_guard(main, start):
9696
x = paddle.static.data("x", self.x.shape, dtype=self.x.dtype)
9797

9898
weight = paddle.static.data(

0 commit comments

Comments
 (0)