load_dgl_partition_data

class load_dgl_partition_data(partition_json_file: str, own_partition_idx: int, device: device)

Loads partition data created by DGL’s partition_graph function

Parameters:
  • partition_json_file (str) – Path to the .json file containing partitioning data

  • own_partition_idx (int) – The index of the partition to load. This is typically the worker/machine rank

  • device (torch.device) – Device on which to place the loaded partition data

Returns:

The loaded partition data