Skip to content

GitLab

Unverified Commit 1129e2b1 authored by Nicolò Lucchesi's avatar Nicolò Lucchesi Committed by GitHub
Browse files

[V1][NixlConnector] Drop `num_blocks` check (#19532) 


Signed-off-by: default avatarNickLucche <nlucches@redhat.com>
parent c742438f
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py
View file @ 1129e2b1
...@@ -703,8 +703,6 @@ class NixlConnectorWorker: ...@@ -703,8 +703,6 @@ class NixlConnectorWorker:
assert self.block_size == remote_block_size, "Remote P worker with " \ assert self.block_size == remote_block_size, "Remote P worker with " \
"different block size is not supported" "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. # Create dst descs and xfer side handles. TP workers have same #blocks.
if engine_id in self.dst_num_blocks: if engine_id in self.dst_num_blocks:
assert self.dst_num_blocks[engine_id] == nixl_agent_meta.num_blocks assert self.dst_num_blocks[engine_id] == nixl_agent_meta.num_blocks
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment