From 1ee95841bd251f9081c3a317984c4dcaa003b3c0 Mon Sep 17 00:00:00 2001 From: zackyoray Date: Thu, 5 Feb 2026 19:51:58 +0200 Subject: [PATCH] [Bugfix] Fix swapped engine_ids in NIXL Llama 4 local attention path (#33795) Signed-off-by: Yoray Zack --- .../kv_transfer/kv_connector/v1/nixl_connector.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py b/vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py index 3a8400447..c2777b393 100644 --- a/vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py +++ b/vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py @@ -2323,15 +2323,15 @@ class NixlConnectorWorker: # Get descs ids for the layer. layer_local_desc_ids = self._get_block_descs_ids( - dst_engine_id, + self.engine_id, layer_local_block_ids, layer_idx, + block_size_ratio=block_size_ratio, ) layer_remote_desc_ids = self._get_block_descs_ids( - self.engine_id, + dst_engine_id, layer_remote_block_ids, layer_idx, - block_size_ratio=block_size_ratio, ) local_descs_list.append(layer_local_desc_ids)