Browse Source

support fwk offline inference when ge_lib is not initialized

pull/1636/head
lichun 4 years ago
parent
commit
a11c6d1066
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/hybrid/node_executor/aicore/aicore_task_compiler.h

+ 1
- 1
ge/hybrid/node_executor/aicore/aicore_task_compiler.h View File

@@ -35,7 +35,7 @@ class AiCoreTaskCompiler : public TaskCompiler {
Status DoCompileOp(const NodePtr &node) const; Status DoCompileOp(const NodePtr &node) const;
Status DoGenerateTask(const Node &node, std::vector<domi::TaskDef> &tasks); Status DoGenerateTask(const Node &node, std::vector<domi::TaskDef> &tasks);
OpsKernelInfoStorePtr aic_kernel_store_; OpsKernelInfoStorePtr aic_kernel_store_;
std::atomic_bool is_initialized_ = false;
std::atomic_bool is_initialized_{false};
static std::mutex mu_; static std::mutex mu_;
}; };
} // namespace hybrid } // namespace hybrid


Loading…
Cancel
Save