From 1129e2b1abc50edd5038cafb185b7c2456e07e65 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Nicol=C3=B2=20Lucchesi?= Date: Thu, 12 Jun 2025 14:36:14 +0200 Subject: [PATCH] [V1][NixlConnector] Drop `num_blocks` check (#19532) Signed-off-by: NickLucche --- vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py b/vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py index 7552fc889f2f1..bdab4850d4c19 100644 --- a/vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py +++ b/vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py @@ -703,8 +703,6 @@ class NixlConnectorWorker: assert self.block_size == remote_block_size, "Remote P worker with " \ "different block size is not supported" - assert self.num_blocks >= nixl_agent_meta.num_blocks - # Create dst descs and xfer side handles. TP workers have same #blocks. if engine_id in self.dst_num_blocks: assert self.dst_num_blocks[engine_id] == nixl_agent_meta.num_blocks