Skip to content

Commit 4750946

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

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
@@ -73,6 +73,11 @@
7373
"examples.fp8_attention",
7474
"fp8_attention_tritonbench",
7575
),
76+
"cross_entropy": (
77+
"tritonbench.operators.cross_entropy.operator",
78+
"examples.cross_entropy",
79+
"cross_entropy",
80+
),
7681
}
7782

7883

0 commit comments

Comments
 (0)