Kezdőlap Felfedezés Súgó
Regisztráció Bejelentkezés
david
/
flash-attention
tükrözi: https://github.com/Dao-AILab/flash-attention
1
0
Másolás 0
Fájlok Problémák 0 Wiki
Branch: ipiszy/used_q
Branch-ok Tag-ek
flash-attention
/
csrc
/
flash_attn_ck
rocking d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) 6 hónapja
..
flash_api.cpp d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) 6 hónapja
flash_common.hpp d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) 6 hónapja
mha_bwd.cpp d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) 6 hónapja
mha_fwd.cpp d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) 6 hónapja
mha_varlen_bwd.cpp d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) 6 hónapja
mha_varlen_fwd.cpp d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) 6 hónapja