david/flash-attention

miroir de https://github.com/Dao-AILab/flash-attention

Auteur	SHA1 Message	Date
Tri Dao	a157cc8c9b [FT] Implement MQA/GQA	il y a 1 an
Tri Dao	62e9814466 [Rotary] Make sure frequency calculation is in fp32	il y a 1 an
Tri Dao	48bc6eacd6 [Gen] Add rotary base as an argument to FT attention kernel	il y a 1 an
Tri Dao	a01d1213d7 [Gen] Add kernel from FasterTransformer for benchmarking	il y a 2 ans