pylibwholegraph
WholeGraph supports PyTorch and provides a distributed graph and kv store.
cuGraph-PyG can leverage WholeGraph for even greater scalability.
Using an RMM memory resource
WholeGraph can use RMM's current per-device memory resource for distributed and hierarchy tensors stored on the device. Configure RMM after selecting the process's CUDA device, then enable RMM allocation before creating WholeMemory tensors:
import rmm
import pylibwholegraph.torch as wgth
rmm.reinitialize(pool_allocator=True, initial_pool_size=2**30)
wgth.set_rmm_enabled(True)
WholeMemory looks up RMM's current resource when each supported allocation is created, so normal
RMM per-device resource configuration applies. Call wgth.set_rmm_enabled(False) to use
WholeMemory's default allocator for future allocations.
Chunked, continuous, and NVSHMEM device tensors require specialized CUDA allocation mechanisms. When RMM is enabled, these tensors emit a warning and use their existing allocation path.
Metadata
Release files for pylibwholegraph-cu12 26.10.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Built distributions (wheels)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| pylibwholegraph_cu12-26.10.0-cp311-abi3-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl | CPython 3.11 | abi3 | Linux glibc 2.28+ x86-64, Linux glibc 2.17+ x86-64 | Details |
| pylibwholegraph_cu12-26.10.0-cp311-abi3-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl | CPython 3.11 | abi3 | Linux glibc 2.17+ ARM64, Linux glibc 2.28+ ARM64 | Details |
Total release size: 595.5 kB
Release files / pylibwholegraph_cu12-26.10.0-cp311-abi3-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl
| Download URL | pylibwholegraph_cu12-26.10.0-cp311-abi3-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl |
|---|---|
| Size | 307.9 kB |
| Tags | CPython 3.11 Linux glibc 2.17+ x86-64 Linux glibc 2.28+ x86-64 abi3 |
|
SHA-256 checksum How to use checksums |
e9a44e82277667b6c9adbc23ef363cb5fff6b16f9545602de5637632cb59f885
|
|
BLAKE2b-256 checksum How to use checksums |
b926321bbdd29e9a3921be110f83d0d50f3128aa59877aa1050424f2ca9d499f
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.8
|
Release files / pylibwholegraph_cu12-26.10.0-cp311-abi3-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl
| Download URL | pylibwholegraph_cu12-26.10.0-cp311-abi3-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl |
|---|---|
| Size | 287.6 kB |
| Tags | CPython 3.11 Linux glibc 2.17+ ARM64 Linux glibc 2.28+ ARM64 abi3 |
|
SHA-256 checksum How to use checksums |
d991a8c4b55abc6169c50f8225702d557caab1d260ba2c11ea0d0e9e091fb771
|
|
BLAKE2b-256 checksum How to use checksums |
f3d30920c3d48198268d84e4a08204416ec6da8431b427ed0261f25dda5c4380
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.8
|
