flash-attention/benchmarks
2023-08-25 14:10:05 -07:00
..
benchmark_causal.py Change causal mask to be aligned to bottom-right instead of top-left 2023-08-24 23:41:07 -07:00
benchmark_flash_attention.py add benchmark for xformers fa2 wrapper (#492) 2023-08-25 14:10:05 -07:00