diff --git a/python/sglang/srt/mem_cache/memory_pool_host.py b/python/sglang/srt/mem_cache/memory_pool_host.py index fc0ba09bc..02c64c8b3 100644 --- a/python/sglang/srt/mem_cache/memory_pool_host.py +++ b/python/sglang/srt/mem_cache/memory_pool_host.py @@ -618,7 +618,7 @@ class MLATokenToKVPoolHost(HostKVCache): elif self.layout == "page_first": transfer_kv_all_layer_mla_lf_pf( src_layers=device_pool.data_ptrs, - dst_k=self.kv_buffer, + dst=self.kv_buffer, src_indices=device_indices, dst_indices=host_indices, item_size=self.token_stride_size,