david/flash-attention

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

نویسنده	SHA1 پیام	تاریخ
Tri Dao	2c996ca25f Use SeqlenInfo for bwd and epilogue	1 هفته پیش
Tri Dao	82dc825759 Don't use the unsafe convert_type function	2 هفته پیش
Tri Dao	e7b93e3902 Clean up mha_combine kernel	3 هفته پیش
Tri Dao	3d0d147940 Early stop on actual num_splits in mha_combine kernel	3 هفته پیش
Tri Dao	9fd6b977bb Precompute the pointers in mha_combine kernel	3 هفته پیش
Tri Dao	01de45b8f8 Add combine kernel	1 ماه پیش