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)[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: True.

  • 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.

  • graph_formats (str or list[str], optional) โ€“ Save partitions in specified formats. It could be any combination of coo, csc. As csc format is mandatory for FusedCSCSamplingGraph, it is not necessary to specify this argument. Itโ€™s mainly for specifying coo format to save edge ID mapping and destination node IDs. If not specified, whether to save coo format is determined by the availability of the format in DGL partitions. Default: None.