david/flash-attention

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

Autor	SHA1 Mensagem	Data
Tri Dao	1e3208566a Tune tile sizes for compilation	há 1 mês atrás
Tri Dao	df5fe55264 Change tile sizes for Sm8x to reduce stack frame	há 1 mês atrás
Tri Dao	5f525322ec Only pass sm_90a compile flag to Sm90 kernels, same w Sm89 kernels	há 1 mês atrás
Tri Dao	180ff782dd Template for Sm86	há 1 mês atrás
Tri Dao	7a802796e1 Big refactor and update	há 1 mês atrás