Unverified Commit 559eb852 authored by youkaichao's avatar youkaichao Committed by GitHub
Browse files

[Core] init_distributed_environment align with init_process_group(#4014)

[Core][Distributed] make init_distributed_environment compatible with init_process_group (#4014)
parent a10d3056
......@@ -39,9 +39,9 @@ _PIPELINE_GLOBAL_RANKS = None
def init_distributed_environment(
world_size: int,
rank: int,
distributed_init_method: Optional[str] = None,
world_size: int = -1,
rank: int = -1,
distributed_init_method: str = "env://",
local_rank: int = -1,
backend: str = "nccl",
):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment