| .. |
|
flash_api.cpp
|
74aed78373
Replace c10::optional with std::optional in flash_attn
|
hai 1 mes |
|
flash_common.cpp
|
e2182cc21d
Support page kvcache in AMD ROCm (#1198)
|
hai 5 meses |
|
flash_common.hpp
|
88d1657a14
[AMD ROCm] Fix KVcache bug and improve performance (#1328)
|
hai 3 meses |
|
mha_bwd.cpp
|
74aed78373
Replace c10::optional with std::optional in flash_attn
|
hai 1 mes |
|
mha_fwd.cpp
|
74aed78373
Replace c10::optional with std::optional in flash_attn
|
hai 1 mes |
|
mha_fwd_kvcache.cpp
|
74aed78373
Replace c10::optional with std::optional in flash_attn
|
hai 1 mes |
|
mha_varlen_bwd.cpp
|
74aed78373
Replace c10::optional with std::optional in flash_attn
|
hai 1 mes |
|
mha_varlen_fwd.cpp
|
74aed78373
Replace c10::optional with std::optional in flash_attn
|
hai 1 mes |
Lu Fang
74aed78373
Replace c10::optional with std::optional in flash_attn