pylibwholegraph.torch.init_torch_env

pylibwholegraph.torch.init_torch_env(

world_rank: int,

world_size: int,

local_rank: int,

local_size: int,

wm_log_level='info',

)

Init WholeGraph environment for PyTorch. :param world_rank: world rank of current process :param world_size: world size of all processes :param local_rank: local rank of current process :param local_size: local size :return: None