 Aman Gupta Karmani
|
b4b6e90334
add benchmark for xformers fa2 wrapper (#492)
|
1 gadu atpakaļ |
 Tri Dao
|
60499abcfd
[Benchmark] Add script to benchmark FlashAttention
|
1 gadu atpakaļ |
|
Tri Dao
|
4f285b3547
FlashAttention-2 release
|
1 gadu atpakaļ |
|
Tri Dao
|
fb88e5e4b3
Move benchmark utils, support AMP
|
2 gadi atpakaļ |
|
Tri Dao
|
6c3a8c65af
Implement cross attention
|
2 gadi atpakaļ |
|
Tri Dao
|
5a61cb7729
Rename src -> flash_attn
|
2 gadi atpakaļ |
|
Tri Dao
|
67c3779598
Reorganize directories, add banner figure
|
2 gadi atpakaļ |
|
Tri Dao
|
9dbc491aa5
Rename, add benchmarking script
|
2 gadi atpakaļ |
