WholeGraph API

WholeGraph is maintained in the cuGraph-GNN repository and has two API layers:

• pylibwholegraph provides Python and PyTorch interfaces for initialization, communicators, distributed tensors, embeddings, graph storage, sampling, and launch helpers.

• libwholegraph provides the native C/CUDA API used by pylibwholegraph and other native applications.

WholeGraph APIs

• pylibwholegraph API
  • Initialization
  • Communicators
  • Tensors
  • Embeddings and optimizers
  • Graph storage and operations
  • Distributed launch
  • Training and data-loading helpers
• libwholegraph API
  • Initialization, communicators, and memory
  • Tensor descriptions and handles
  • Tensor and graph operations
  • Embeddings
  • Environment callbacks