diff --git a/benchmarks/run.py b/benchmarks/run.py index 639f4017d..86df60714 100644 --- a/benchmarks/run.py +++ b/benchmarks/run.py @@ -477,11 +477,11 @@ class RunResult: "helion_gather_gemv_tritonbench-accuracy": "helion_accuracy", }, "int4_gemm": { - "eager_int4_gemm": "baseline", - "triton_int4_gemm-speedup": "triton_speedup", - "triton_int4_gemm-accuracy": "triton_accuracy", - "torch_compile_int4_gemm-speedup": "torch_compile_speedup", - "torch_compile_int4_gemm-accuracy": "torch_compile_accuracy", + "preprocessed_eager_int4_gemm": "baseline", + "preprocessed_triton_int4_gemm-speedup": "triton_speedup", + "preprocessed_triton_int4_gemm-accuracy": "triton_accuracy", + "preprocessed_torch_compile_int4_gemm-speedup": "torch_compile_speedup", + "preprocessed_torch_compile_int4_gemm-accuracy": "torch_compile_accuracy", "helion_int4_gemm_tritonbench-speedup": "helion_speedup", "helion_int4_gemm_tritonbench-accuracy": "helion_accuracy", },