add support for unified CPU/CUDA memory to pytorch.
Project description
Torch unified
Adds support for unified memory to torch.
Usage
import torch
from torch_unified import empty_unified
tensor_cpu, tensor_gpu = empty_unified((10,), torch.float32)
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
torch_unified-0.1.tar.gz
(4.6 kB
view hashes)