Bug fix for TORCH_VERSION_AFTER_* importation (#433)

* Bug fix for TORCH_VERSION_AFTER_* importation Bug fix for TORCH_VERSION_AFTER_* importation * Update benchmark_aq.py
pytorch · Jun 25, 2024 · 96d49cd · 96d49cd
1 parent c2cf973
commit 96d49cd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
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 (

diff --git a/benchmarks/intmm.py b/benchmarks/intmm.py
@@ -6,7 +6,7 @@
 import pathlib
 
 import torch
-from torchao.quantization.utils import TORCH_VERSION_AFTER_2_4, TORCH_VERSION_AFTER_2_2
+from torchao.utils import TORCH_VERSION_AFTER_2_4, TORCH_VERSION_AFTER_2_2
 
 
 # Check if CUDA is available, if not, exit the script