Bugfix

pytorch-labs · Dec 12, 2024 · d98dbec · d98dbec
1 parent 1241c1d
commit d98dbec
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/run.py b/run.py
@@ -8,6 +8,7 @@
 import argparse
 import copy
 import subprocess
+import sys
 from typing import List
 
 from tritonbench.operator_loader import load_opbench_by_name_from_loader