Skip to content

GitLab

Unverified Commit 8e22c38b authored by Tongping Liu's avatar Tongping Liu Committed by GitHub
Browse files

[hotfix] Fixing the bug related to ipv6 support 


Co-authored-by: default avatarByteDance <tongping.liu@bytedance.com>
parent ac85a180
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
colossalai/context/parallel_context.py
View file @ 8e22c38b
...@@ -370,7 +370,7 @@ class ParallelContext(metaclass=SingletonMeta): ...@@ -370,7 +370,7 @@ class ParallelContext(metaclass=SingletonMeta):
port (str): the master port for distributed training port (str): the master port for distributed training
""" """
# initialize the default process group # initialize the default process group
init_method = f'tcp://{host}:{port}' init_method = f'tcp://[{host}]:{port}'
dist.init_process_group(rank=rank, world_size=world_size, backend=backend, init_method=init_method) dist.init_process_group(rank=rank, world_size=world_size, backend=backend, init_method=init_method)
# None will give the default global process group for pytorch dist operations # None will give the default global process group for pytorch dist operations
......
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