Skip to content

Commit 6d0d720

Browse files
committed
[Benchmark] Add cross_entropy to tritonbench integration
stack-info: PR: #321, branch: yf225/stack/29
1 parent 3663c9c commit 6d0d720

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
@@ -65,6 +65,11 @@
6565
"examples.attention",
6666
"attention",
6767
),
68+
"cross_entropy": (
69+
"tritonbench.operators.cross_entropy.operator",
70+
"examples.cross_entropy",
71+
"cross_entropy",
72+
),
6873
}
6974

7075

0 commit comments

Comments
 (0)