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
Repository
7fc39832e2f1bac23ece617fd51d9ffb5d020791
Switch branch/tag
flash-attention
csrc
flash_attn
src
fmha_fprop_fp16_kernel.sm80.cu
Find file
Blame
History
Permalink
Use block_size=128 for headdim=128 on SM80
· 7fc39832
Tri Dao
authored
Oct 21, 2022
Previously we were using block_size=256.
7fc39832
fmha_fprop_fp16_kernel.sm80.cu
12.5 KB
Edit
Web IDE
Replace fmha_fprop_fp16_kernel.sm80.cu
×
Attach a file by drag & drop or
click to upload
Commit message
Replace fmha_fprop_fp16_kernel.sm80.cu
Replace file
Cancel
A new branch will be created in your fork and a new merge request will be started.