rocking d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) hace 6 meses
..
flash_api.cpp d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) hace 6 meses
flash_common.hpp d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) hace 6 meses
mha_bwd.cpp d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) hace 6 meses
mha_fwd.cpp d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) hace 6 meses
mha_varlen_bwd.cpp d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) hace 6 meses
mha_varlen_fwd.cpp d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) hace 6 meses