From d585bac7ff6e5138d3de8e66925905e1f5b0ef94 Mon Sep 17 00:00:00 2001 From: QQSong Date: Mon, 24 Jun 2024 14:47:53 -0700 Subject: [PATCH 1/2] Bug fix for TORCH_VERSION_AFTER_* importation Bug fix for TORCH_VERSION_AFTER_* importation --- benchmarks/intmm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/intmm.py b/benchmarks/intmm.py index edadf2a7cf..950f100f20 100644 --- 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 From 7d546824f5dabbd1e4f62d277a00b9b6738b3207 Mon Sep 17 00:00:00 2001 From: QQSong Date: Mon, 24 Jun 2024 14:48:28 -0700 Subject: [PATCH 2/2] Update benchmark_aq.py --- benchmarks/benchmark_aq.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/benchmark_aq.py b/benchmarks/benchmark_aq.py index 09fe1bc452..fc19d1ab8a 100644 --- 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 (