Skip to content
Merged
Changes from all commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion python/paddle/distributed/fleet/base/role_maker.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
import warnings
from multiprocessing import Process, Manager

import paddle
import paddle.fluid as fluid
from paddle.distributed.fleet.base.private_helper_function import wait_server_ready

Expand Down Expand Up @@ -869,7 +870,8 @@ def _generate_role(self):
else:
self._collective_env()
self._role_is_generated = True
self._gloo_init()
if not paddle.fluid.framework.in_dygraph_mode():
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

it seems in dygraph mode, we do not need to do gloo_init, but actually it is initialized in other places. In general, this is not a good design, and should be refactored with init_parallel_env in the future.

self._gloo_init()


class UserDefinedRoleMaker(PaddleCloudRoleMaker):
Expand Down