Skip to content

Commit 83ee91e

Browse files
[None][fix] Fix 6522 mpi.pkl5.intracomm.Request has wait not Wait (#6646)
Signed-off-by: Netanel Haber <[email protected]>
1 parent 3036d49 commit 83ee91e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorrt_llm/_torch/pyexecutor/py_executor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -800,7 +800,7 @@ def _executor_loop_pp(self):
800800
# Second last rank does not need to since last rank has original decoded tokens
801801
if not self.dist.is_second_last_pp_rank:
802802
if self.send_handles[prev_microbatch_id] is not None:
803-
self.send_handles[prev_microbatch_id].Wait()
803+
self.send_handles[prev_microbatch_id].wait()
804804
needs_logits = (
805805
self._need_return_logits(scheduled_batch)
806806
or (self._need_return_log_probs(scheduled_batch)

0 commit comments

Comments
 (0)