def _recv_forward()

in dualpipe/dualpipev.py [0:0]

• 6 lines of code
• 6 McCabe index (conditional complexity)

    def _recv_forward(self, phase: int) -> None:
        if (self.is_first_rank and phase == 0) or (self.is_last_rank and phase == 1):
            return

        self.current_recv_f_chunk_id[phase] += 1
        tensors = comm.append_irecv(self.comm_ops, self.prev_rank if phase == 0 else self.next_rank, self.group)
        self.input_chunks[phase].append(tensors)