Update benchmark_aq.py

pytorch · Jun 24, 2024 · 7d54682 · 7d54682
1 parent d585bac
commit 7d54682
Showing 1 changed file with 1 addition and 1 deletion.
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 (