Skip to content

Commit d884774

Browse files
authored
[Benchmark] Add cross_entropy to tritonbench integration (#321)
1 parent deb7c8a commit d884774

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)