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
b4a9dd6c9c
flash-attention
/
csrc
History
Tri Dao
b4a9dd6c9c
Temporarily switch to cutlass fork for more shapes
2024-07-11 09:29:21 -07:00
..
cutlass
@
fa4f635906
Temporarily switch to cutlass fork for more shapes
2024-07-11 09:29:21 -07:00
flash_attn
Implement cache_leftpad
2024-07-11 08:17:15 -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