Skip to content

Commit

Permalink
fix cli
Browse files Browse the repository at this point in the history
  • Loading branch information
msaroufim committed Apr 16, 2024
1 parent 5b1abbb commit 29c0631
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions benchmarks/intmm.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,10 +14,6 @@
print("CUDA is not available. Exiting the script.")
sys.exit(0)

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 import int_matmul, int_scaled_matmul
Expand Down Expand Up @@ -94,7 +90,7 @@ def run_benchmarks(shapes):

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

0 comments on commit 29c0631

Please sign in to comment.