Browse Source

Fix code check

tags/v1.2.0
dongduo 4 years ago
parent
commit
6d73d06877
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; 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");


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

@@ -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();


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

@@ -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_; }


Loading…
Cancel
Save