Home Explore Help
Register Sign In
david
/
flash-attention
mirror of https://github.com/Dao-AILab/flash-attention
1
0
Fork 0
Files Issues 0 Wiki
Branch: changes_for_fp8
Branches Tags
flash-attention
/
csrc
/
flash_attn_ck
rocking d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) 4 months ago
..
flash_api.cpp d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) 4 months ago
flash_common.hpp d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) 4 months ago
mha_bwd.cpp d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) 4 months ago
mha_fwd.cpp d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) 4 months ago
mha_varlen_bwd.cpp d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) 4 months ago
mha_varlen_fwd.cpp d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) 4 months ago