Chirag Jain 50896ec574 Make nvcc threads configurable via environment variable (#885)		2024-03-13 20:46:57 -07:00
..
cutlass@bbe579a9e3	Update Cutlass to v3.4.1	2024-02-20 16:28:21 -08:00
flash_attn	Enable headdim 256 backward on consumer GPUs (Ampere, Ada)	2024-02-21 15:56:19 -08: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