[Bugfix] Fix swapped engine_ids in NIXL Llama 4 local attention path (#33795)

Signed-off-by: Yoray Zack <yorayz@nvidia.com>
This commit is contained in:
zackyoray
2026-02-05 19:51:58 +02:00
committed by GitHub
parent 7d8c6804e2
commit 1ee95841bd

View File

@@ -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)