Skip to content

Commit ad73fba

Browse files
committed
add test
1 parent 95b2255 commit ad73fba

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

benchmarks/run.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -534,6 +534,15 @@ class RunResult:
534534
"helion_addmm_tritonbench-speedup": "helion_speedup",
535535
"helion_addmm_tritonbench-accuracy": "helion_accuracy",
536536
},
537+
"addmm-bwd": {
538+
"aten_addmm": "baseline",
539+
"triton_addmm-speedup": "triton_speedup",
540+
"triton_addmm-accuracy": "triton_accuracy",
541+
"pt2_addmm_maxautotune-speedup": "torch_compile_speedup",
542+
"pt2_addmm_maxautotune-accuracy": "torch_compile_accuracy",
543+
"helion_addmm_tritonbench-speedup": "helion_speedup",
544+
"helion_addmm_tritonbench-accuracy": "helion_accuracy",
545+
},
537546
# "ragged_attention": {
538547
# "triton_ragged_attention-speedup": "triton_speedup",
539548
# "triton_ragged_attention-accuracy": "triton_accuracy",
@@ -603,6 +612,15 @@ class RunResult:
603612
"helion_matmul_tritonbench-speedup": "helion_speedup",
604613
"helion_matmul_tritonbench-accuracy": "helion_accuracy",
605614
},
615+
"gemm-bwd": {
616+
"aten_matmul": "baseline",
617+
"triton_tutorial_matmul-speedup": "triton_speedup",
618+
"triton_tutorial_matmul-accuracy": "triton_accuracy",
619+
"pt2_triton_matmul-speedup": "torch_compile_speedup",
620+
"pt2_triton_matmul-accuracy": "torch_compile_accuracy",
621+
"helion_matmul_tritonbench-speedup": "helion_speedup",
622+
"helion_matmul_tritonbench-accuracy": "helion_accuracy",
623+
},
606624
"fp8_gemm": {
607625
"torch_fp8_gemm": "baseline",
608626
f"{'blackwell_persistent_tma' if IS_B200 else 'triton_tma_persistent'}_fp8_gemm-speedup": "triton_speedup",

0 commit comments

Comments
 (0)