@@ -158,7 +158,7 @@ std::shared_ptr<ge::DNNEngine> DNNEngineManager::GetEngine(const std::string &na | |||||
return nullptr; | return nullptr; | ||||
} | } | ||||
bool DNNEngineManager::IsEngineRegistered(const std::string &name) const { | |||||
bool DNNEngineManager::IsEngineRegistered(const std::string &name) { | |||||
auto iter = engines_map_.find(name); | auto iter = engines_map_.find(name); | ||||
if (iter != engines_map_.end()) { | if (iter != engines_map_.end()) { | ||||
return true; | return true; | ||||
@@ -177,7 +177,7 @@ const map<string, uint64_t> &DNNEngineManager::GetCheckSupportCost() const { | |||||
return checksupport_cost_; | return checksupport_cost_; | ||||
} | } | ||||
std::string DNNEngineManager::GetDNNEngineName(const ge::NodePtr &node_ptr) const { | |||||
std::string DNNEngineManager::GetDNNEngineName(const ge::NodePtr &node_ptr) { | |||||
std::lock_guard<std::mutex> lock(mutex_); | 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"); | GE_IF_BOOL_EXEC(node_ptr == nullptr, GELOGE(GE_CLI_GE_NOT_INITIALIZED, "DNNEngineManager: node_ptr is nullptr"); | ||||
@@ -61,9 +61,9 @@ class DNNEngineManager { | |||||
public: | public: | ||||
friend class GELib; | friend class GELib; | ||||
std::shared_ptr<ge::DNNEngine> GetEngine(const std::string &name) const; | std::shared_ptr<ge::DNNEngine> GetEngine(const std::string &name) const; | ||||
bool IsEngineRegistered(const std::string &name) const; | |||||
bool IsEngineRegistered(const std::string &name); | |||||
// If can't find appropriate engine name, return "", report error | // If can't find appropriate engine name, return "", report error | ||||
string GetDNNEngineName(const ge::NodePtr &node_ptr) const; | |||||
string GetDNNEngineName(const ge::NodePtr &node_ptr); | |||||
const map<string, SchedulerConf> &GetSchedulers() const; | const map<string, SchedulerConf> &GetSchedulers() const; | ||||
const map<string, uint64_t> &GetCheckSupportCost() const; | const map<string, uint64_t> &GetCheckSupportCost() const; | ||||
void InitPerformanceStaistic(); | void InitPerformanceStaistic(); | ||||
@@ -48,13 +48,13 @@ class GELib { | |||||
Status Finalize(); | Status Finalize(); | ||||
// get DNNEngineManager object | // get DNNEngineManager object | ||||
const DNNEngineManager &DNNEngineManagerObj() { return engineManager_; } | |||||
DNNEngineManager &DNNEngineManagerObj() { return engineManager_; } | |||||
// get OpsKernelManager object | // get OpsKernelManager object | ||||
const OpsKernelManager &OpsKernelManagerObj() { return opsManager_; } | |||||
OpsKernelManager &OpsKernelManagerObj() { return opsManager_; } | |||||
// get SessionManager object | // get SessionManager object | ||||
const SessionManager &SessionManagerObj() { return sessionManager_; } | |||||
SessionManager &SessionManagerObj() { return sessionManager_; } | |||||
// get Initial flag | // get Initial flag | ||||
bool InitFlag() const { return init_flag_; } | bool InitFlag() const { return init_flag_; } | ||||