Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
flash-attention
Commits
ccbb14f38ee1e51a6f65bccbef9e4765acba6d79
Switch branch/tag
flash-attention
csrc
flash_attn
src
kernel_traits.h
16 Sep, 2023
1 commit
Implement rotary embedding in flash_attn_with_kvcache
· ccbb14f3
Tri Dao
authored
Sep 16, 2023
ccbb14f3
04 Sep, 2023
1 commit
Remove unused sdPsum in dot_do_o function
· 5953c4f5
Tri Dao
authored
Sep 03, 2023
5953c4f5
01 Sep, 2023
1 commit
Remove commented out code in bwd (#512)
· 37e32feb
Sophia Wisdom
authored
Sep 01, 2023
* Remove lots of comments * Remove unused traits
37e32feb
29 Aug, 2023
1 commit
Implement splitKV attention
· b1fbbd83
Tri Dao
authored
Aug 29, 2023
b1fbbd83
13 Aug, 2023
1 commit
Prepare for Cutlass 3.2
· dbd79237
Tri Dao
authored
Aug 13, 2023
dbd79237
17 Jul, 2023
1 commit
FlashAttention-2 release
· 4f285b35
Tri Dao
authored
Jul 17, 2023
4f285b35