flash-attention/csrc/flash_attn
Grigory Sizov 2a15840f09
Enable paged attention in varlen forward (#831)
* Enable paged attention in varlen forward

* Format + fix padding
2024-03-15 00:48:19 -07:00
..
src Enable headdim 256 backward on consumer GPUs (Ampere, Ada) 2024-02-21 15:56:19 -08:00
flash_api.cpp Enable paged attention in varlen forward (#831) 2024-03-15 00:48:19 -07:00