Kernels
drbh HF Staff commited on
Commit
97e63f9
·
verified ·
1 Parent(s): 98af189

Benchmarks uploaded using `kernels`.

Browse files
Files changed (1) hide show
  1. benchmarks/benchmark.py +17 -0
benchmarks/benchmark.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from kernels.benchmarks import (
2
+ FlashAttentionBenchmark,
3
+ FlashAttentionCausalBenchmark,
4
+ FlashAttentionVarlenBenchmark,
5
+ )
6
+
7
+
8
+ class FlashAttn(FlashAttentionBenchmark):
9
+ pass
10
+
11
+
12
+ class FlashAttnCausal(FlashAttentionCausalBenchmark):
13
+ pass
14
+
15
+
16
+ class FlashAttnVarlen(FlashAttentionVarlenBenchmark):
17
+ pass