Tri Dao 26a6e0a048 Switch to Cutlass 3.6.0 1 месяц назад
..
composable_kernel @ a9b170b541 e2182cc21d Support page kvcache in AMD ROCm (#1198) 3 месяцев назад
cutlass @ 19f51596e8 26a6e0a048 Switch to Cutlass 3.6.0 1 месяц назад
flash_attn 83e41b3ca4 Add custom ops for compatibility with PT Compile (#1139) 2 месяцев назад
flash_attn_ck 53a4f34163 Hotfix due to change of upstream api (#1239) 2 месяцев назад
ft_attention 50896ec574 Make nvcc threads configurable via environment variable (#885) 9 месяцев назад
fused_dense_lib 50896ec574 Make nvcc threads configurable via environment variable (#885) 9 месяцев назад
fused_softmax 50896ec574 Make nvcc threads configurable via environment variable (#885) 9 месяцев назад
layer_norm 50896ec574 Make nvcc threads configurable via environment variable (#885) 9 месяцев назад
rotary 50896ec574 Make nvcc threads configurable via environment variable (#885) 9 месяцев назад
xentropy 50896ec574 Make nvcc threads configurable via environment variable (#885) 9 месяцев назад