Browse Source

fwkacllib support inference when ge lib is not inited

tags/v1.3.0
lichun 3 years ago
parent
commit
2083a8e059
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      ge/hybrid/node_executor/aicore/aicore_task_compiler.h

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

@@ -18,7 +18,6 @@
#define GE_HYBRID_KERNEL_AICORE_TASK_COMPILER_H_

#include <mutex>
#include <atomic>
#include "opskernel_manager/ops_kernel_manager.h"
#include "aicore_node_executor.h"

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


Loading…
Cancel
Save