Skip to content

Commit 29c0631

Browse files
committed
fix cli
1 parent 5b1abbb commit 29c0631

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

benchmarks/intmm.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -14,10 +14,6 @@
1414
print("CUDA is not available. Exiting the script.")
1515
sys.exit(0)
1616

17-
if not TORCH_VERSION_AFTER_2_2:
18-
print("torch version must be 2.2 or higher")
19-
sys.exit(0)
20-
2117
import torch.nn.functional as F
2218
import torch.utils.benchmark as benchmark
2319
from torchao.kernel.intmm import int_matmul, int_scaled_matmul
@@ -94,7 +90,7 @@ def run_benchmarks(shapes):
9490

9591
if __name__ == "__main__":
9692
parser = argparse.ArgumentParser(description="integer matmul benchmarks")
97-
parser.add_argument("file_path", type=str, help="Path to csv file with shapes")
93+
parser.add_argument("--file_path", type=str, required=True, help="Path to csv file with shapes")
9894
args = parser.parse_args()
9995
# Access the file path provided as an argument
10096
file_path = args.file_path

0 commit comments

Comments
 (0)