diff --git a/ge/engine_manager/dnnengine_manager.cc b/ge/engine_manager/dnnengine_manager.cc index b23993b6..777d2b38 100644 --- a/ge/engine_manager/dnnengine_manager.cc +++ b/ge/engine_manager/dnnengine_manager.cc @@ -158,7 +158,7 @@ std::shared_ptr 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 &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 lock(mutex_); GE_IF_BOOL_EXEC(node_ptr == nullptr, GELOGE(GE_CLI_GE_NOT_INITIALIZED, "DNNEngineManager: node_ptr is nullptr"); diff --git a/ge/engine_manager/dnnengine_manager.h b/ge/engine_manager/dnnengine_manager.h index c3ae5b95..4cc0b8cf 100755 --- a/ge/engine_manager/dnnengine_manager.h +++ b/ge/engine_manager/dnnengine_manager.h @@ -61,9 +61,9 @@ class DNNEngineManager { public: friend class GELib; std::shared_ptr 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 &GetSchedulers() const; const map &GetCheckSupportCost() const; void InitPerformanceStaistic(); diff --git a/ge/init/gelib.h b/ge/init/gelib.h index e52b8dd6..a2aca99d 100644 --- a/ge/init/gelib.h +++ b/ge/init/gelib.h @@ -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_; }