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
Árbore: 9b6cba16c1
Ramas Etiquetas
flash-attention
/
csrc
/
flash_attn_ck
rocking 53a4f34163 Hotfix due to change of upstream api (#1239) hai 2 meses
..
flash_api.cpp 53a4f34163 Hotfix due to change of upstream api (#1239) hai 2 meses
flash_common.cpp e2182cc21d Support page kvcache in AMD ROCm (#1198) hai 3 meses
flash_common.hpp e2182cc21d Support page kvcache in AMD ROCm (#1198) hai 3 meses
mha_bwd.cpp 53a4f34163 Hotfix due to change of upstream api (#1239) hai 2 meses
mha_fwd.cpp 53a4f34163 Hotfix due to change of upstream api (#1239) hai 2 meses
mha_fwd_kvcache.cpp e2182cc21d Support page kvcache in AMD ROCm (#1198) hai 3 meses
mha_varlen_bwd.cpp 53a4f34163 Hotfix due to change of upstream api (#1239) hai 2 meses
mha_varlen_fwd.cpp 53a4f34163 Hotfix due to change of upstream api (#1239) hai 2 meses