Skip to main content

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)

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 details)

Uploaded Source

File details

Details for the file torch_unified-0.1.tar.gz.

File metadata

  • Download URL: torch_unified-0.1.tar.gz
  • Upload date:
  • Size: 4.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.0 CPython/3.9.7

File hashes

Hashes for torch_unified-0.1.tar.gz
Algorithm Hash digest
SHA256 d51dbf837e371d927ce12fd5af1bf579b6fe875087c488247ad27710969a71df
MD5 acdbfc334a17da28c66c78805c153752
BLAKE2b-256 364bc1874559c1218a3121eed473dd9509d2e3fc12826ebba756b1ab0b3003b7

See more details on using hashes here.

Release history Release notifications | RSS feed

This release

0.1

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page