def init(self):
# env init
if not oneflow_api.IsEnvInited():
flow.env.init()
# session init
if not oneflow_api.IsSessionInited():
self._make_config_proto()
session_util._TryCompleteConfigProto(self.config_proto_)
c_api_util.InitLazyGlobalSession(self.config_proto_)
self.status_ = self.SessionStatus.OPEN
def Init(self):
assert self.status_ is SessionStatus.OPEN
self.status_ = SessionStatus.RUNNING
if not oneflow_api.IsEnvInited():
oneflow.env.init()
_TryCompleteConfigProto(self.config_proto)
self.resource_ = self.config_proto.resource
if not oneflow_api.EagerExecutionEnabled():
c_api_util.InitLazyGlobalSession(self.config_proto)
for job_name, func_desc in self.job_name2function_desc_.items():
compiler.Compile(self, func_desc, self.config_proto)
self.existed_module_names_ = set()
self.job_name2var_name2var_blob_ = dict()
assert len(self.job_name2function_desc_.items()) > 0
oneflow_api.StartLazyGlobalSession()
self.inter_user_job_info_ = c_api_util.GetInterUserJobInfo()
# Get latest op_attr and job_name after compiler.Compile
self.UpdateInfo4InterfaceOp()
if not config_util.api_legacy_model_io_enabled():
check_point_v2.Init()
else:
self.eager_config_proto_ctx_ = oneflow_api.LogicalConfigProtoContext(
str(self.config_proto))
return self
def env_initialized(ctx):
assert in_normal_mode(ctx)
return oneflow_api.IsEnvInited()
