MLA optimization with flashattention for llama.cpp,MLA + FA now only uses K-cache - 47% saving on KV-cache size

News

MLA + FA now only uses K-cache - 47% saving on KV-cache size (only for use with #13435 for now) by jukofyork · Pull Request #13529 · ggml-org/llama.cpp

llama_kv_cache_unified: kv_size = 163840, type_k = 'f16', type_v = 'f16', n_layer = 61, can_shift = 0, padding = 256

llama_kv_cache_unified: CUDA0 KV buffer size = 10980.00 MiB

llama_kv_cache_unified: KV self size = 10980.00 MiB, K (f16): 10980.00 MiB, V (f16): 0.00 MiB

The full context of 160k tokens now takes up less than 11GB without kquants

Posted May 14, 2025, 9:42 PM UTC

13435

Comments aren't archived yet

This post had 35 comments on Reddit. Comments for this period haven't been loaded into the archive yet.