Inicio Explorar Axuda
Rexistro Iniciar sesión
david
/
flash-attention
réplica de https://github.com/Dao-AILab/flash-attention
1
0
Fork 0
Ficheiros Incidencias 0 Wiki
Rama: ipiszy/local_attn
Ramas Etiquetas
flash-attention
/
csrc
rocking e2182cc21d Support page kvcache in AMD ROCm (#1198) hai 4 meses
..
composable_kernel @ a9b170b541 e2182cc21d Support page kvcache in AMD ROCm (#1198) hai 4 meses
cutlass @ 756c351b49 74b0761ff7 [FA3] BF16 forward hai 6 meses
flash_attn 65f723bb9a Split bwd into more .cu files to speed up compilation hai 5 meses
flash_attn_ck e2182cc21d Support page kvcache in AMD ROCm (#1198) hai 4 meses
ft_attention 50896ec574 Make nvcc threads configurable via environment variable (#885) hai 10 meses
fused_dense_lib 50896ec574 Make nvcc threads configurable via environment variable (#885) hai 10 meses
fused_softmax 50896ec574 Make nvcc threads configurable via environment variable (#885) hai 10 meses
layer_norm 50896ec574 Make nvcc threads configurable via environment variable (#885) hai 10 meses
rotary 50896ec574 Make nvcc threads configurable via environment variable (#885) hai 10 meses
xentropy 50896ec574 Make nvcc threads configurable via environment variable (#885) hai 10 meses