squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
50ca23488d
flash-attention/benchmarks
History
Tri Dao 50ca23488d Add Triton implementation for benchmarking
2022-10-23 17:25:56 -07:00
..
benchmark_causal.py Add Triton implementation for benchmarking 2022-10-23 17:25:56 -07:00
benchmark_flash_attention.py Move benchmark utils, support AMP 2022-10-23 12:50:00 -07:00