[benchmarks] change tritonbench path

Differential Revision: D84927071

Pull Request resolved: #966

Author: xuzhao9

benchmarks/run.py

@@ -1056,7 +1056,10 @@ def accuracy_fail_hook(
     try:
         from tritonbench.run import run as tritonbench_run
     except ImportError:
-        from pytorch.tritonbench.run import run as tritonbench_run
+        try:
+            from tritonbench.utils.run_utils import tritonbench_run
+        except ImportError:
+            from pytorch.tritonbench.run import run as tritonbench_run
 
     with tempfile.NamedTemporaryFile(mode="w+t", suffix=".csv") as tmp:
         tritonbench_args.extend(["--output", tmp.name])
@@ -1093,7 +1096,7 @@ def process_result(
     metrics = collections.defaultdict(list)
     for row in lines[1:]:
         row_data = row.strip().split(";")
-        if row_data[0] == "average":
+        if row_data[0] == "average" or len(row_data) == 1:
             continue
         for idx, (name, item) in enumerate(zip(names, row_data, strict=True)):
             if idx == 0: