chore: import upstream snapshot with attribution

2026-07-13 13:35:51 +08:00
commit c36a561cd8
2172 changed files with 455595 additions and 0 deletions
@@ -0,0 +1,30 @@
+"""DGL distributed module"""
+
+from . import optim
+from .dist_context import exit_client, initialize
+from .dist_dataloader import (
+    DistDataLoader,
+    DistEdgeDataLoader,
+    DistNodeDataLoader,
+    EdgeCollator,
+    NodeCollator,
+)
+from .dist_graph import DistGraph, DistGraphServer, edge_split, node_split
+from .dist_tensor import DistTensor
+from .graph_partition_book import GraphPartitionBook, PartitionPolicy
+from .graph_services import *
+from .kvstore import KVClient, KVServer
+from .nn import *
+from .partition import (
+    dgl_partition_to_graphbolt,
+    gb_convert_single_dgl_partition,
+    load_partition,
+    load_partition_book,
+    load_partition_feats,
+    partition_graph,
+)
+from .rpc import *
+from .rpc_client import connect_to_server
+from .rpc_server import start_server
+from .server_state import ServerState
+from .constants import *