From 48eb44b71972df0f1da88c0ab4672ac952db13f0 Mon Sep 17 00:00:00 2001 From: Tianyu Guo Date: Tue, 6 Oct 2026 11:19:41 +0800 Subject: [PATCH] [Bugfix][KVConnector][NIXL] Fix TP handshake test after per-region FA desc flags (#60157) Signed-off-by: Tianyu Guo --- tests/v1/kv_connector/unit/test_nixl_connector.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/tests/v1/kv_connector/unit/test_nixl_connector.py b/tests/v1/kv_connector/unit/test_nixl_connector.py index 35ac7d887e..18097d220e 100644 --- a/tests/v1/kv_connector/unit/test_nixl_connector.py +++ b/tests/v1/kv_connector/unit/test_nixl_connector.py @@ -901,11 +901,16 @@ class TestNixlHandshake: vllm_config = create_vllm_config() - connector = NixlConnector( - vllm_config, KVConnectorRole.WORKER, make_kv_cache_config(block_size=16) - ) + # num_blocks=1 keeps the fake handshake's region_num_blocks consistent + # with the minimal single-block registration below (num_descs == 1); + # _fa_desc_replicated cross-checks the two. + kv_cache_config = make_kv_cache_config(block_size=16, num_blocks=1) + connector = NixlConnector(vllm_config, KVConnectorRole.WORKER, kv_cache_config) connector.connector_worker = FakeNixlConnectorWorker( - vllm_config, connector.engine_id, hand_shake_latency=0 + vllm_config, + connector.engine_id, + hand_shake_latency=0, + kv_cache_config=kv_cache_config, ) worker = connector.connector_worker