squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
43617deab9
flash-attention/csrc/flash_attn
History
Tri Dao 43617deab9 Remove template for (IsEvenMN=T, IsEvenK=F) to speed up compilation
2023-09-18 12:21:36 -07:00
..
src Remove template for (IsEvenMN=T, IsEvenK=F) to speed up compilation 2023-09-18 12:21:36 -07:00
flash_api.cpp Implement rotary embedding in flash_attn_with_kvcache 2023-09-16 01:20:16 -07:00