fix args typo in memory_pool_host (#8662)

76ba5bbe · huangtingwei · GitHub · ed6f7597 · 76ba5bbe
Unverified Commit 76ba5bbe authored Aug 04, 2025 by huangtingwei Committed by GitHub Aug 03, 2025
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

python/sglang/srt/mem_cache/memory_pool_host.py python/sglang/srt/mem_cache/memory_pool_host.py +1 -1

No files found.
--- 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,