rocking
d8f104e97a
Support AMD ROCm on FlashAttention 2 (#1010)
|
4 сар өмнө | |
|---|---|---|
| .. | ||
| flash_api.cpp | 4 сар өмнө | |
| flash_common.hpp | 4 сар өмнө | |
| mha_bwd.cpp | 4 сар өмнө | |
| mha_fwd.cpp | 4 сар өмнө | |
| mha_varlen_bwd.cpp | 4 сар өмнө | |
| mha_varlen_fwd.cpp | 4 сар өмнө | |