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]

Convert partitions of dgl to FusedCSCSamplingGraph of GraphBolt.

This API converts DGLGraph partitions to FusedCSCSamplingGraph which is dedicated for sampling in GraphBolt. New graphs will be stored alongside original graph as fused_csc_sampling_graph.pt.

In the near future, partitions are supposed to be saved as FusedCSCSamplingGraph directly. At that time, this API should be deprecated.

Parameters:

  • part_config (str) – The partition configuration JSON file.

  • store_eids (bool, optional) – Whether to store edge IDs in the new graph. Default: False.

  • store_inner_node (bool, optional) – Whether to store inner node mask in the new graph. Default: False.

  • store_inner_edge (bool, optional) – Whether to store inner edge mask in the new graph. Default: False.