Skip to content

Commit ee072e5

Browse files
qiaoxj07Shixiaowei02
authored andcommitted
[None][fix] Return 3-tuple from check_gen_transfer_status fast path (#14162)
Signed-off-by: Xianjie Qiao <xqiao@nvidia.com> Signed-off-by: Xianjie <5410381+qiaoxj07@users.noreply.github.com>
1 parent 9ad0b97 commit ee072e5

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tensorrt_llm/_torch/disaggregation/transceiver.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -518,7 +518,7 @@ def check_context_transfer_status(
518518
def check_gen_transfer_status(self, at_least_request_num: Optional[int]):
519519
# Skip the allgather in _gen_consensus when this transceiver never receives (pure CTX role).
520520
if not self._ever_had_recv_session:
521-
return [], []
521+
return [], [], []
522522
block_all = at_least_request_num is None
523523
wait_num = at_least_request_num if not block_all else 0
524524

0 commit comments

Comments
 (0)