This website requires JavaScript.
Explore
Help
Register
Sign In
squall
/
flash-attention
Watch
1
Star
0
Fork
0
You've already forked flash-attention
Code
Issues
Pull Requests
Actions
Packages
Projects
Releases
Wiki
Activity
9c0e9ee86d
flash-attention
/
csrc
History
Tri Dao
656daef4ea
Use Cute's local_tile to get gQ, gK, gV
2024-04-07 20:10:19 -07:00
..
cutlass
@
bbe579a9e3
Update Cutlass to v3.4.1
2024-02-20 16:28:21 -08:00
flash_attn
Use Cute's local_tile to get gQ, gK, gV
2024-04-07 20:10:19 -07:00
ft_attention
Make nvcc threads configurable via environment variable (
#885
)
2024-03-13 20:46:57 -07:00
fused_dense_lib
Make nvcc threads configurable via environment variable (
#885
)
2024-03-13 20:46:57 -07:00
fused_softmax
Make nvcc threads configurable via environment variable (
#885
)
2024-03-13 20:46:57 -07:00
layer_norm
Make nvcc threads configurable via environment variable (
#885
)
2024-03-13 20:46:57 -07:00
rotary
Make nvcc threads configurable via environment variable (
#885
)
2024-03-13 20:46:57 -07:00
xentropy
Make nvcc threads configurable via environment variable (
#885
)
2024-03-13 20:46:57 -07:00