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