gigl.distributed.utils.get_available_device

gigl.distributed.utils.device.get_available_device(local_process_rank: int) → device

Returns the available device for the current process.

Args:

local_process_rank (int): The local rank of the current process within a node.

Returns:

torch.device: The device to use.