File tree Expand file tree Collapse file tree 1 file changed +1
-5
lines changed Expand file tree Collapse file tree 1 file changed +1
-5
lines changed Original file line number Diff line number Diff line change 14
14
print ("CUDA is not available. Exiting the script." )
15
15
sys .exit (0 )
16
16
17
- if not TORCH_VERSION_AFTER_2_2 :
18
- print ("torch version must be 2.2 or higher" )
19
- sys .exit (0 )
20
-
21
17
import torch .nn .functional as F
22
18
import torch .utils .benchmark as benchmark
23
19
from torchao .kernel .intmm import int_matmul , int_scaled_matmul
@@ -94,7 +90,7 @@ def run_benchmarks(shapes):
94
90
95
91
if __name__ == "__main__" :
96
92
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" )
98
94
args = parser .parse_args ()
99
95
# Access the file path provided as an argument
100
96
file_path = args .file_path
You can’t perform that action at this time.
0 commit comments