Method pin_memory

accelerator/hpu_accelerator.py:234–235

(self, tensor, align_bytes=1)

        return functools.partial(torch.tensor, dtype=torch.long, device='hpu')

    def pin_memory(self, tensor, align_bytes=1):
        return tensor.pin_memory(self.device())

    def is_pinned(self, tensor):
        return tensor.is_pinned()