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)
1 maintainer
Metadata
1 maintainer
Release files for torch-unified 0.1
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| torch_unified-0.1.tar.gz | 4.6 kB | Details |
Release files / torch_unified-0.1.tar.gz
| Download URL | torch_unified-0.1.tar.gz |
|---|---|
| Size | 4.6 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
d51dbf837e371d927ce12fd5af1bf579b6fe875087c488247ad27710969a71df
|
|
BLAKE2b-256 checksum How to use checksums |
364bc1874559c1218a3121eed473dd9509d2e3fc12826ebba756b1ab0b3003b7
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.0 CPython/3.9.7
|
