pylibwholegraph.torch.initialize.init_torch_env#

pylibwholegraph.torch.initialize.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