graphengine/tests/framework/stub_engine/engine/stub_engine.cc

/**
 * Copyright 2020 Huawei Technologies Co., Ltd
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "stub_engine.h"
#include <map>
#include <memory>
#include <string>
#include <securec.h>
#include "framework/common/debug/ge_log.h"
#include "common/ge/ge_util.h"
#include "inc/st_types.h"

namespace ge {
namespace st {
StubEngine &StubEngine::Instance() {
  static StubEngine instance;
  return instance;
}

Status StubEngine::Initialize(const std::map<string, string> &options) {
  for (const auto engine_2_lib : kStubEngine2KernelLib) {
    auto ops_kernel_store = MakeShared<StubOpsKernelInfoStore>(engine_2_lib.second);
    if (ops_kernel_store == nullptr) {
      return FAILED;
    }
    ops_kernel_store_map_.insert(make_pair(engine_2_lib.second, ops_kernel_store));
  }
  return SUCCESS;
}

void StubEngine::GetOpsKernelInfoStores(std::map<std::string, OpsKernelInfoStorePtr> &ops_kernel_map) {
  for (const auto name_2_ops_kernel_store : ops_kernel_store_map_) {
    ops_kernel_map[name_2_ops_kernel_store.first] = name_2_ops_kernel_store.second;
  }
}

void StubEngine::GetGraphOptimizerObjs(std::map<std::string, GraphOptimizerPtr> &) {
  // no optimizer for host cpu engine
}

Status StubEngine::Finalize() {
  return SUCCESS;
}
}  // namespace st
}  // namespace ge

ge::Status Initialize(const std::map<string, string> &options) {
  return ge::st::StubEngine::Instance().Initialize(options);
}

void GetOpsKernelInfoStores(std::map<std::string, OpsKernelInfoStorePtr> &ops_kernel_map) {
  ge::st::StubEngine::Instance().GetOpsKernelInfoStores(ops_kernel_map);
}

void GetGraphOptimizerObjs(std::map<std::string, GraphOptimizerPtr> &graph_optimizers) {
  ge::st::StubEngine::Instance().GetGraphOptimizerObjs(graph_optimizers);
}

ge::Status Finalize() {
  return ge::st::StubEngine::Instance().Finalize();
}