Browse Source

Fix code check

tags/v1.2.0
dongduo 4 years ago
parent
commit
9b3f6bd090
3 changed files with 7 additions and 7 deletions
  1. +2
    -2
      ge/engine_manager/dnnengine_manager.cc
  2. +2
    -2
      ge/engine_manager/dnnengine_manager.h
  3. +3
    -3
      ge/init/gelib.h

+ 2
- 2
ge/engine_manager/dnnengine_manager.cc View File

@@ -158,7 +158,7 @@ std::shared_ptr<ge::DNNEngine> DNNEngineManager::GetEngine(const std::string &na
return nullptr;
}

bool DNNEngineManager::IsEngineRegistered(const std::string &name) {
bool DNNEngineManager::IsEngineRegistered(const std::string &name) const {
auto iter = engines_map_.find(name);
if (iter != engines_map_.end()) {
return true;
@@ -177,7 +177,7 @@ const map<string, uint64_t> &DNNEngineManager::GetCheckSupportCost() const {
return checksupport_cost_;
}

std::string DNNEngineManager::GetDNNEngineName(const ge::NodePtr &node_ptr) {
std::string DNNEngineManager::GetDNNEngineName(const ge::NodePtr &node_ptr) const {
std::lock_guard<std::mutex> lock(mutex_);

GE_IF_BOOL_EXEC(node_ptr == nullptr, GELOGE(GE_CLI_GE_NOT_INITIALIZED, "DNNEngineManager: node_ptr is nullptr");


+ 2
- 2
ge/engine_manager/dnnengine_manager.h View File

@@ -61,9 +61,9 @@ class DNNEngineManager {
public:
friend class GELib;
std::shared_ptr<ge::DNNEngine> GetEngine(const std::string &name) const;
bool IsEngineRegistered(const std::string &name);
bool IsEngineRegistered(const std::string &name) const;
// If can't find appropriate engine name, return "", report error
string GetDNNEngineName(const ge::NodePtr &node_ptr);
string GetDNNEngineName(const ge::NodePtr &node_ptr) const;
const map<string, SchedulerConf> &GetSchedulers() const;
const map<string, uint64_t> &GetCheckSupportCost() const;
void InitPerformanceStaistic();


+ 3
- 3
ge/init/gelib.h View File

@@ -48,13 +48,13 @@ class GELib {
Status Finalize();

// get DNNEngineManager object
DNNEngineManager &DNNEngineManagerObj() { return engineManager_; }
const DNNEngineManager &DNNEngineManagerObj() { return engineManager_; }

// get OpsKernelManager object
OpsKernelManager &OpsKernelManagerObj() { return opsManager_; }
const OpsKernelManager &OpsKernelManagerObj() { return opsManager_; }

// get SessionManager object
SessionManager &SessionManagerObj() { return sessionManager_; }
const SessionManager &SessionManagerObj() { return sessionManager_; }

// get Initial flag
bool InitFlag() const { return init_flag_; }


Loading…
Cancel
Save