dgl-ascend
Deep Graph Library (DGL) with Ascend NPU support, developed by the BUPT-GAMMA group.
Installation
pip install dgl-ascend
Requirements: Ascend 910B NPU, CANN >= 8.5, aarch64 Linux, Python 3.11.
Quick Start
import torch
import torch_npu
import dgl
device = torch.device("npu:0")
g = dgl.graph(([0, 1, 2], [1, 2, 3])).to(device)
g.update_all(dgl.function.copy_u("x", "m"), dgl.function.sum("m", "h"))
License
Apache License 2.0.
Metadata
Release files for dgl-ascend 0.0.1
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| dgl_ascend-0.0.1-cp311-cp311-manylinux_2_28_aarch64.whl | CPython 3.11 | CPython 3.11 | Linux glibc 2.28+ ARM64 | Details |
Release files / dgl_ascend-0.0.1-cp311-cp311-manylinux_2_28_aarch64.whl
| Download URL | dgl_ascend-0.0.1-cp311-cp311-manylinux_2_28_aarch64.whl |
|---|---|
| Size | 4.8 MB |
| Tags | CPython 3.11 Linux glibc 2.28+ ARM64 |
|
SHA-256 checksum How to use checksums |
8062e02b98cc41540111eb3cc7edfb5a49423340ecb5931dd9488b1087a2289a
|
|
BLAKE2b-256 checksum How to use checksums |
4f7d669fa159520a4300869093d3344243b9fd5849c20e81a6a761766957d0a4
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.11.14
|