[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:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user