squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
b0eac3297f
flash-attention/benchmarks
History
Tri Dao b0c0db81f6 Implement FlashAttention in Triton
2022-10-30 18:09:11 -07:00
..
benchmark_causal.py Implement FlashAttention in Triton 2022-10-30 18:09:11 -07:00
benchmark_flash_attention.py Move benchmark utils, support AMP 2022-10-23 12:50:00 -07:00