dgl.distributed.dgl_partition_to_graphbolt

dgl.distributed.dgl_partition_to_graphbolt(part_config, *, store_eids=False, store_inner_node=False, store_inner_edge=False)[source]

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

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

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

Parameters:

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

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

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

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