def _weight_chunk()

in dualpipe/dualpipe.py [0:0]


    def _weight_chunk(self) -> None:
        if self.forward_only:
            return

        self._commit_and_wait_comm()

        # Assume FIFO
        WeightGradStore.pop()