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
"llama/ggml-cuda/ggml-cuda.cu" did not exist on "7a1c3e62dc6a55b2c6d48e2e20f6eaf1c0b27b44"
9795159082f6e6c847db2bf4284fd17326c31fbd
Switch branch/tag
flash-attention
flash_attn
ops
triton
rotary.py
06 Sep, 2023
1 commit
[Rotary] Set device before launching Triton kernel to avoid error
· 97951590
Tri Dao
authored
Sep 05, 2023
97951590
04 Sep, 2023
1 commit
[Rotary] Implement varlen rotary
· b28ec236
Tri Dao
authored
Sep 03, 2023
b28ec236
03 Sep, 2023
3 commits
[Rotary] Clean up rotary Triton implementation a bit
· 861c8257
Tri Dao
authored
Sep 03, 2023
861c8257
[Rotary] Speed up rotary kernel when interleaved=True
· 1c523c1c
Tri Dao
authored
Sep 03, 2023
1c523c1c
[Rotary] Implement rotary in Triton
· 942fcbf0
Tri Dao
authored
Sep 03, 2023
942fcbf0
