Method OneFlowEnv

oneflow/api/cpp/env_impl.cpp:116–129

}  // namespace

OneFlowEnv::OneFlowEnv() {
  of::EnvProto env_proto;
  CompleteEnvProto(env_proto);

  env_ctx_ = std::make_shared<of::EnvGlobalObjectsScope>(env_proto);

  of::ConfigProto config_proto;
  config_proto.mutable_resource()->set_cpu_device_num(1);  // useless, will be set in TryInit
  const int64_t session_id = of::NewSessionId();
  config_proto.set_session_id(session_id);
  CHECK(of::RegsterSessionId(session_id));
  session_ctx_ = std::make_shared<of::MultiClientSessionContext>(env_ctx_);
  CHECK_JUST(session_ctx_->TryInit(config_proto));
}

OneFlowEnv::~OneFlowEnv() {
  session_ctx_.reset();