def _commit_and_wait_comm()

in dualpipe/dualpipe.py [0:0]

• 8 lines of code
• 3 McCabe index (conditional complexity)

    def _commit_and_wait_comm(self) -> None:
        if not self.comm_ops:
            return
        reqs = dist.batch_isend_irecv(self.comm_ops)
        for req in reqs:
            req.wait()
        self.comm_ops = []
        self._free_tensors()