Skip to content

Commit cc95ff1

Browse files
committed
[Benchmark] Add softmax tritonbench integration
stack-info: PR: #286, branch: yf225/stack/17
1 parent 38753a6 commit cc95ff1

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

benchmarks/run.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,11 @@
4545
"rms_norm_tritonbench",
4646
),
4747
"sum": ("tritonbench.operators.sum.operator", "examples.sum", "sum_tritonbench"),
48+
"softmax": (
49+
"tritonbench.operators.softmax.operator",
50+
"examples.softmax",
51+
"softmax",
52+
),
4853
"jagged_mean": (
4954
"tritonbench.operators.jagged_mean.operator",
5055
"examples.jagged_mean",

0 commit comments

Comments
 (0)