Skip to content

Commit 52ab9f6

Browse files
committed
rename symbol
Signed-off-by: Will Eaton <[email protected]>
1 parent 004f610 commit 52ab9f6

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -623,7 +623,7 @@ def __init__(self, vllm_config: VllmConfig, engine_id: str):
623623
# invalid blocks from failed NIXL operations
624624
self._invalid_block_ids: set[int] = set()
625625
# requests that skipped transfer (handshake or transfer failures)
626-
self._skipped_transfer: set[ReqId] = set()
626+
self._failed_recv_reqs: set[ReqId] = set()
627627

628628
# Background thread for handling new handshake requests.
629629
self._nixl_handshake_listener_t: threading.Thread | None = None
@@ -821,7 +821,7 @@ def request_ready(f: Future[Any], entry=(req_id, meta)):
821821
)
822822
if req_meta := self._recving_metadata.get(req_id):
823823
self._invalid_block_ids.update(req_meta.local_block_ids)
824-
self._skipped_transfer.add(req_id)
824+
self._failed_recv_reqs.add(req_id)
825825

826826
fut.add_done_callback(request_ready)
827827

@@ -1230,8 +1230,8 @@ def get_finished(self) -> tuple[set[str], set[str]]:
12301230
done_recving = self._pop_done_transfers(self._recving_transfers)
12311231

12321232
# add requests that skipped transfer to done_recving
1233-
done_recving.update(self._skipped_transfer)
1234-
self._skipped_transfer.clear()
1233+
done_recving.update(self._failed_recv_reqs)
1234+
self._failed_recv_reqs.clear()
12351235

12361236
if len(done_sending) > 0 or len(done_recving) > 0:
12371237
logger.debug(
@@ -1533,7 +1533,7 @@ def _read_blocks(
15331533
self.xfer_stats.record_failed_transfer()
15341534
if handle is not None:
15351535
self.nixl_wrapper.release_xfer_handle(handle)
1536-
self._skipped_transfer.add(request_id)
1536+
self._failed_recv_reqs.add(request_id)
15371537

15381538
def _get_block_descs_ids(
15391539
self, engine_id: str, block_ids: list[int], layer_idx: int | None = None

0 commit comments

Comments
 (0)