[ab test] fix triton repo location (#454)

davidberard98 · web-flow · commit f70c7c3d365c · 2025-09-18T22:28:11.000-04:00
diff --git a/tritonbench/utils/scuba_utils.py b/tritonbench/utils/scuba_utils.py
@@ -136,7 +136,7 @@ def decorate_benchmark_data(
 
     repo_locs = {
         "tritonbench": REPO_PATH,
-        "triton": os.environ.get("TRITONBENCH_TRITON_REPO", "unknown"),
+        "triton": os.environ.get("TRITONBENCH_TRITON_INSTALL_DIR", "unknown"),
         "pytorch": os.environ.get("TRITONBENCH_PYTORCH_REPO_PATH", "unknown"),
     }
     aggregated_obj = {

Original file line number	Diff line number	Diff line change
`@@ -136,7 +136,7 @@ def decorate_benchmark_data(`
`136`	`136`
`137`	`137`	`repo_locs = {`
`138`	`138`	`"tritonbench": REPO_PATH,`
`139`		`- "triton": os.environ.get("TRITONBENCH_TRITON_REPO", "unknown"),`
	`139`	`+ "triton": os.environ.get("TRITONBENCH_TRITON_INSTALL_DIR", "unknown"),`
`140`	`140`	`"pytorch": os.environ.get("TRITONBENCH_PYTORCH_REPO_PATH", "unknown"),`
`141`	`141`	`}`
`142`	`142`	`aggregated_obj = {`