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