Bug fix for TORCH_VERSION_AFTER_* importation (#433)

qingquansong · web-flow · commit 96d49cd08826 · 2024-06-24T17:15:35.000-07:00
* Bug fix for TORCH_VERSION_AFTER_* importation

Bug fix for TORCH_VERSION_AFTER_* importation

* Update benchmark_aq.py
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

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 (`