AlpinDale 61aed092a5 rocm: add support for FP8 KV cache in the custom paged attention kkernels (#1066) | il y a 1 mois | |
---|---|---|
.. | ||
attention.cu | 61aed092a5 rocm: add support for FP8 KV cache in the custom paged attention kkernels (#1066) | il y a 1 mois |
ops.h | 61aed092a5 rocm: add support for FP8 KV cache in the custom paged attention kkernels (#1066) | il y a 1 mois |
torch_bindings.cpp | 61aed092a5 rocm: add support for FP8 KV cache in the custom paged attention kkernels (#1066) | il y a 1 mois |