gigl.distributed.graph_store.compute

Attributes

logger

Functions

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.

Module Contents

gigl.distributed.graph_store.compute.init_compute_process(local_rank, cluster_info)

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.

Raises:

ValueError – If the process group is already initialized.

Return type:

None

gigl.distributed.graph_store.compute.shutdown_compute_proccess()

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)

Parameters:

None

Return type:

None

gigl.distributed.graph_store.compute.logger