dgl.distributed.dgl_partition_to_graphbolt

dgl.distributed.dgl_partition_to_graphbolt(part_config, *, store_eids=True, store_inner_node=False, store_inner_edge=False, graph_formats=None, n_jobs=1)[源码]

将 DGL 的分区转换为 GraphBolt 的 FusedCSCSamplingGraph。

此 API 将 DGLGraph 分区转换为 FusedCSCSamplingGraph，后者专用于 GraphBolt 中的采样。新的图将与原始图一起存储为 fused_csc_sampling_graph.pt。

在不久的将来，分区将直接存储为 FusedCSCSamplingGraph。届时，此 API 应弃用。

参数：

  • part_config (str) – 分区配置 JSON 文件。

  • store_eids (bool, 可选) – 是否在新图中存储边 ID。默认值：True。

  • store_inner_node (bool, 可选) – 是否在新图中存储内部节点掩码。默认值：False。

  • store_inner_edge (bool, 可选) – 是否在新图中存储内部边掩码。默认值：False。

  • graph_formats (str 或 list[str], 可选) – 以指定的格式保存分区。可以是 coo, csc 的任意组合。由于 csc 格式对于 FusedCSCSamplingGraph 是必需的，因此无需指定此参数。它主要用于指定 coo 格式以保存边 ID 映射和目标节点 ID。如果未指定，是否保存 coo 格式取决于 DGL 分区中该格式的可用性。默认值：None。

  • n_jobs (int) – 分区转换期间并行运行的作业数。最大并行度由分区数量决定。