为pytorch添加了对统一CPU/CUDA内存的支持。
项目描述
Torch unified
为torch添加了对统一内存的支持。
用法
import torch
from torch_unified import empty_unified
tensor_cpu, tensor_gpu = empty_unified((10,), torch.float32)
关闭
torch_unified-0.1.tar.gz的哈希值
| 算法 | 哈希摘要 | |
|---|---|---|
| SHA256 | d51dbf837e371d927ce12fd5af1bf579b6fe875087c488247ad27710969a71df |
|
| MD5 | acdbfc334a17da28c66c78805c153752 |
|
| BLAKE2b-256 | 364bc1874559c1218a3121eed473dd9509d2e3fc12826ebba756b1ab0b3003b7 |
