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