xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-21 15:57:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
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
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/v1/kv_connector/unit/test_nixl_connector.py
View File

@ -229,6 +229,9 @@ class FakeNixlConnectorWorker(NixlConnectorWorker):
num_blocks=1, num_blocks=1,
block_len=self.block_len, block_len=self.block_len,
attn_backend_name=self.backend_name, 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) remote_tp_size=remote_tp_size)
return {0: remote_agent_name} return {0: remote_agent_name}