@johnsmith0031 Hey have you seen the new version? Its got a huge speed boost and LINEAR attention :))))
FlashAttention-2 currently supports:
Ampere, Ada, or Hopper GPUs (e.g., A100, RTX 3090, RTX 4090, H100). Support for Turing GPUs (T4, RTX 2080) is coming soon, please use FlashAttention 1.x for Turing GPUs for now.
Datatype fp16 and bf16 (bf16 requires Ampere, Ada, or Hopper GPUs).
All head dimensions up to 256. Head dim > 192 backward requires A100/A800 or H100/H800.
@johnsmith0031 Hey have you seen the new version? Its got a huge speed boost and LINEAR attention :))))
https://github.com/Dao-AILab/flash-attention#installation-and-features