diff --git a/benchmarks/run.py b/benchmarks/run.py index bcbe0461f..4610e38e7 100644 --- a/benchmarks/run.py +++ b/benchmarks/run.py @@ -536,8 +536,8 @@ class RunResult: "triton_gemm-accuracy": "triton_accuracy", "torch_compile_gemm-speedup": "torch_compile_speedup", "torch_compile_gemm-accuracy": "torch_compile_accuracy", - "helion_matmul_split_k_tritonbench-speedup": "helion_speedup", - "helion_matmul_split_k_tritonbench-accuracy": "helion_accuracy", + "helion_matmul_tritonbench-speedup": "helion_speedup", + "helion_matmul_tritonbench-accuracy": "helion_accuracy", }, "fp8_gemm": { "torch_fp8_gemm": "baseline",