def _free_tensors()

in dualpipe/dualpipe.py [0:0]

• 5 lines of code
• 2 McCabe index (conditional complexity)

    def _free_tensors(self) -> None:
        for tensor in self.to_free:
            assert tensor._base is None, f"pipeline stage should not return view tensors {dist.get_rank(), tensor.shape}"
            tensor.data = torch.Tensor()
        self.to_free = []