gigl.distributed.utils.device# Functions# get_available_device(local_process_rank) Returns the available device for the current process. Module Contents# gigl.distributed.utils.device.get_available_device(local_process_rank)[source]# Returns the available device for the current process. Parameters: local_process_rank (int) – The local rank of the current process within a node. Returns: The device to use. Return type: torch.device
