flash-attention/csrc
2024-03-15 00:48:54 -07:00
..
cutlass@bbe579a9e3 Update Cutlass to v3.4.1 2024-02-20 16:28:21 -08:00
flash_attn Add in, macrosf for defining __grid_constant__ (#852) 2024-03-15 00:48:54 -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