From 4164c3356076f716a36f4c8cedd4abe9c00b27a6 Mon Sep 17 00:00:00 2001
From: lichun <lichun30@hisilicon.com>
Date: Wed, 12 May 2021 17:30:16 +0800
Subject: [PATCH] support fwk offline inference when ge_lib is not initialized

---
 ge/hybrid/node_executor/aicore/aicore_task_compiler.h | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ge/hybrid/node_executor/aicore/aicore_task_compiler.h b/ge/hybrid/node_executor/aicore/aicore_task_compiler.h
index 15e8ef1c..4fe2cdc5 100755
--- a/ge/hybrid/node_executor/aicore/aicore_task_compiler.h
+++ b/ge/hybrid/node_executor/aicore/aicore_task_compiler.h
@@ -18,6 +18,7 @@
 #define GE_HYBRID_KERNEL_AICORE_TASK_COMPILER_H_
 
 #include <mutex>
+#include <atomic>
 #include "opskernel_manager/ops_kernel_manager.h"
 #include "aicore_node_executor.h"