[Nixl][CI] Fix tests (#22806)

Signed-off-by: NickLucche <nlucches@redhat.com>
This commit is contained in:
Nicolò Lucchesi 2025-08-13 15:03:53 +02:00 committed by GitHub
parent 98deac3879
commit 6b794c756c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -229,6 +229,9 @@ class FakeNixlConnectorWorker(NixlConnectorWorker):
num_blocks=1,
block_len=self.block_len,
attn_backend_name=self.backend_name,
# `self.kv_cache_layout` is only forced to HND when vllm engine
# is started. We mock HND here.
kv_cache_layout="HND",
),
remote_tp_size=remote_tp_size)
return {0: remote_agent_name}