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
beb2bf2a32
flash-attention
/
csrc
History
Nicolas Patry
8f873cc6ac
Implement softcapping. (
#1025
)
...
* Softcap v2 (fwd only). * Some missing interface + remove overrides in tests.
2024-07-08 11:24:48 -07:00
..
cutlass
@
7d49e6c7e2
Update to Cutlass 3.5
2024-05-26 12:49:33 -07:00
flash_attn
Implement softcapping. (
#1025
)
2024-07-08 11:24:48 -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