Update benchmark_aq.py

qingquansong · web-flow · commit 7d546824f5da · 2024-06-24T14:48:28.000-07:00
diff --git a/benchmarks/benchmark_aq.py b/benchmarks/benchmark_aq.py
@@ -5,7 +5,7 @@
     Int8WeightOnlyQuantizedLinearWeight,
     Int4WeightOnlyQuantizedLinearWeight,
 )
-from torchao.quantization.utils import (
+from torchao.utils import (
     TORCH_VERSION_AFTER_2_4,
 )
 from torchao.quantization.quant_api import (

Original file line number	Diff line number	Diff line change
`@@ -5,7 +5,7 @@`
`5`	`5`	`Int8WeightOnlyQuantizedLinearWeight,`
`6`	`6`	`Int4WeightOnlyQuantizedLinearWeight,`
`7`	`7`	`)`
`8`		`-from torchao.quantization.utils import (`
	`8`	`+from torchao.utils import (`
`9`	`9`	`TORCH_VERSION_AFTER_2_4,`
`10`	`10`	`)`
`11`	`11`	`from torchao.quantization.quant_api import (`