dmlc--dgl
nv-dlasalle
ae8dbe6d3c
* Split NCCL wrapper from sparse optimizer and sparse embedding * Add more unit tests for single node nccl * Fix unit test for tf * Switch to device histogram * Fix histgram issues * Finish migration to histogram * Handle cases with zero send/recieve data * Start on partition object * Get compiling * Updates * Add unit tests * Switch to partition object * Fix linting issues * Rename partition file * Add python doc * Fix python assert and finish doxygen comments * Remove stubs for range based partition to satisfy pylint * Wrap unit test in GPU only * Wrap explicit cuda call in ifdef * Merge with partition.py * update docstrings * Cleanup partition_op * Add Workspace object * Switch to using workspace object * Move last remainder based function out of nccl_api * Add error messages * Update docs with examples * Fix linting erros Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
46 行
1.3 KiB
Python
46 行
1.3 KiB
Python
"""
|
|
The ``dgl`` package contains data structure for storing structural and feature data
|
|
(i.e., the :class:`DGLGraph` class) and also utilities for generating, manipulating
|
|
and transforming graphs.
|
|
"""
|
|
|
|
|
|
# Windows compatibility
|
|
# This initializes Winsock and performs cleanup at termination as required
|
|
import socket
|
|
|
|
# Should import backend before importing anything else
|
|
from .backend import load_backend, backend_name
|
|
|
|
from . import function
|
|
from . import contrib
|
|
from . import container
|
|
from . import distributed
|
|
from . import random
|
|
from . import sampling
|
|
from . import dataloading
|
|
from . import ops
|
|
from . import cuda
|
|
|
|
from ._ffi.runtime_ctypes import TypeCode
|
|
from ._ffi.function import register_func, get_global_func, list_global_func_names, extract_ext_funcs
|
|
from ._ffi.base import DGLError, __version__
|
|
|
|
from .base import ALL, NTYPE, NID, ETYPE, EID
|
|
from .readout import *
|
|
from .batch import *
|
|
from .convert import *
|
|
from .generators import *
|
|
from .heterograph import DGLHeteroGraph
|
|
from .heterograph import DGLHeteroGraph as DGLGraph # pylint: disable=reimported
|
|
from .subgraph import *
|
|
from .traversal import *
|
|
from .transform import *
|
|
from .propagate import *
|
|
from .random import *
|
|
from .data.utils import save_graphs, load_graphs
|
|
from . import optim
|
|
|
|
from ._deprecate.graph import DGLGraph as DGLGraphStale
|
|
from ._deprecate.nodeflow import *
|