only allow fleet to be initialized once

revert-16555-model_data_cryption_link_all_lib
dongdaxiang 6 years ago
parent 3e38d1db46
commit a58df687a8

@ -29,14 +29,17 @@ class Fleet(object):
self._opt_info = None # for fleet only self._opt_info = None # for fleet only
self.role_maker_ = None self.role_maker_ = None
self.local_ip_ = 0 self.local_ip_ = 0
self.is_initialized_ = False
def init(self): def init(self):
# TODO(guru4elephant) # TODO(guru4elephant)
# this is a temporary solution # this is a temporary solution
# we will support more configurable RoleMaker for users in the future # we will support more configurable RoleMaker for users in the future
self.role_maker_ = MPISymetricRoleMaker() if not self.is_initialized_:
self.role_maker_.generate_role() self.role_maker_ = MPISymetricRoleMaker()
self._fleet_ptr = fluid.core.Fleet() self.role_maker_.generate_role()
self._fleet_ptr = fluid.core.Fleet()
self.is_initialized_ = True
def stop(self): def stop(self):
self.role_maker_.barrier_worker() self.role_maker_.barrier_worker()

Loading…
Cancel
Save