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
a5559a0e752ec9f8ce7157e1d4db9f08593638c3
Switch branch/tag
flash-attention
csrc
flash_attn
src
fmha.h
04 Jul, 2022
2 commits
Do P * dP (pointwise) in the bwd in fp32 instead of fp16
· a5559a0e
Tri Dao
authored
Jul 03, 2022
a5559a0e
Implement cross attention
· 6c3a8c65
Tri Dao
authored
Jun 30, 2022
6c3a8c65
12 Jun, 2022
1 commit
Refactor Gmem code to store q, k, v pointers separately
· 5d07483b
Tri Dao
authored
Jun 12, 2022
5d07483b
26 May, 2022
1 commit
Rename, add benchmarking script
· 9dbc491a
Tri Dao
authored
May 26, 2022
9dbc491a
20 May, 2022
1 commit
First release
· 1fcbe6f0
Tri Dao
authored
May 20, 2022
1fcbe6f0