|
@@ -113,7 +113,10 @@ class Main(object):
|
|
def init_fleet_with_gloo(use_gloo=True):
|
|
def init_fleet_with_gloo(use_gloo=True):
|
|
if use_gloo:
|
|
if use_gloo:
|
|
os.environ["PADDLE_WITH_GLOO"] = "1"
|
|
os.environ["PADDLE_WITH_GLOO"] = "1"
|
|
- role = role_maker.PaddleCloudRoleMaker()
|
|
|
|
|
|
+ role = PaddleCloudRoleMaker(
|
|
|
|
+ is_collective=False,
|
|
|
|
+ init_gloo=True
|
|
|
|
+ )
|
|
fleet.init(role)
|
|
fleet.init(role)
|
|
else:
|
|
else:
|
|
fleet.init()
|
|
fleet.init()
|