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"