fix broken API

benchmarks/intmm.py

@@ -6,21 +6,21 @@
 import pathlib
 
 import torch
-from torchao.quantization.utils import TORCH_VERSION_AFTER_2_4
+from torchao.quantization.utils import TORCH_VERSION_AFTER_2_4, TORCH_VERSION_AFTER_2_2
 
 
 # Check if CUDA is available, if not, exit the script
 if not torch.cuda.is_available():
     print("CUDA is not available. Exiting the script.")
     sys.exit(0)
 
-if not TORCH_VERSION_AFTER_2_4:
-    print("torch version must be 2.4 or higher")
+if not TORCH_VERSION_AFTER_2_2:
+    print("torch version must be 2.2 or higher")
     sys.exit(0)
 
 import torch.nn.functional as F
 import torch.utils.benchmark as benchmark
-from torchao.kernel.intmm_triton import int_matmul, int_scaled_matmul
+from torchao.kernel.intmm import int_matmul, int_scaled_matmul
 
 torch._dynamo.config.cache_size_limit = 128
 torch._dynamo.config.accumulated_cache_size_limit = 128