gigl.distributed.graph_store.compute#

Attributes#

`init_compute_process`(local_rank, cluster_info[, ...])	Initializes distributed setup for a compute node in a Graph Store cluster.
`shutdown_compute_proccess`()	Shuts down the distributed setup for a compute node in a Graph Store cluster.

gigl.distributed.graph_store.compute.init_compute_process(local_rank, cluster_info, compute_world_backend=None)[source]#

Initializes distributed setup for a compute node in a Graph Store cluster.

Should be called once per compute process (e.g. one per process per compute node, once per cluster_info.compute_cluster_world_size)

Parameters:

local_rank (int) – The local (process) rank on the compute node.
cluster_info (GraphStoreInfo) – The cluster information.
compute_world_backend (Optional[str]) – The backend for the compute Torch Distributed process group.

Raises:

ValueError – If the process group is already initialized.

Return type:

None

gigl.distributed.graph_store.compute.shutdown_compute_proccess()[source]#

Shuts down the distributed setup for a compute node in a Graph Store cluster.

Should be called once per compute process (e.g. one per process per compute node, once per cluster_info.compute_cluster_world_size)