Skip to content

Commit 739a225

Browse files
committed
[Benchmark] Add softmax tritonbench integration
stack-info: PR: #286, branch: yf225/stack/17
1 parent 3cec8cd commit 739a225

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
@@ -69,6 +69,11 @@
6969
"examples.fp8_attention",
7070
"fp8_attention_tritonbench",
7171
),
72+
"softmax": (
73+
"tritonbench.operators.softmax.operator",
74+
"examples.softmax",
75+
"softmax",
76+
),
7277
}
7378

7479

0 commit comments

Comments
 (0)