Source code for torch_geometric.distributed.dist_context

from dataclasses import dataclass
from enum import Enum


class DistRole(Enum):
    WORKER = 1


[docs]@dataclass
class DistContext:
    r"""Context information of the current process."""
    rank: int
    global_rank: int
    world_size: int
    global_world_size: int
    group_name: str
    role: DistRole = DistRole.WORKER

    @property
    def worker_name(self) -> str:
        return f'{self.group_name}-{self.rank}'