diff --git a/mace/kernels/cwise.h b/mace/kernels/cwise.h
index e33528113394509fd976e2e825568df96806d6cf..57beb64f9a9f13d099abc66ba137a5d0f0a29e1a 100644
--- a/mace/kernels/cwise.h
+++ b/mace/kernels/cwise.h
@@ -5,6 +5,7 @@
 #define MACE_KERNELS_CWISE_H_
 
 #include <algorithm>
+#include <memory>
 #include <vector>
 
 #include "mace/core/future.h"
diff --git a/mace/kernels/depth_to_space.h b/mace/kernels/depth_to_space.h
index 648a66d9448941aa71a820cc36b2b39cc80d7910..ab557c91e3101234585c85ba1b01d4b87060134c 100644
--- a/mace/kernels/depth_to_space.h
+++ b/mace/kernels/depth_to_space.h
@@ -4,6 +4,7 @@
 
 #ifndef MACE_KERNELS_DEPTH_TO_SPACE_H_
 #define MACE_KERNELS_DEPTH_TO_SPACE_H_
+#include <memory>
 #include <vector>
 
 #include "mace/core/future.h"
diff --git a/mace/kernels/depthwise_conv2d.h b/mace/kernels/depthwise_conv2d.h
index 8767e8ab57b6fcb86f58d3083bcef1e1d45965fd..cc6ceee507be8d9da0f367a50c69f8ab86d9760e 100644
--- a/mace/kernels/depthwise_conv2d.h
+++ b/mace/kernels/depthwise_conv2d.h
@@ -9,6 +9,7 @@
 #include <arm_neon.h>
 #endif
 #include <algorithm>
+#include <memory>
 #include <vector>
 
 #include "mace/core/future.h"
diff --git a/mace/kernels/eltwise.h b/mace/kernels/eltwise.h
index 13adae8cde402b5ff3db0eefaec030ce1c8c7aa9..faf52ff31edf589c8fb6d89bc914a171f775b8d9 100644
--- a/mace/kernels/eltwise.h
+++ b/mace/kernels/eltwise.h
@@ -5,6 +5,7 @@
 #define MACE_KERNELS_ELTWISE_H_
 
 #include <algorithm>
+#include <memory>
 #include <vector>
 
 #include "mace/core/future.h"
diff --git a/mace/kernels/fully_connected.h b/mace/kernels/fully_connected.h
index 56acf6b64ab5996e1a3af1ce062c6fe8446be8e4..7f37b1735c3f4b20afbff19f7a0c6c663c5eacd7 100644
--- a/mace/kernels/fully_connected.h
+++ b/mace/kernels/fully_connected.h
@@ -5,6 +5,7 @@
 #ifndef MACE_KERNELS_FULLY_CONNECTED_H_
 #define MACE_KERNELS_FULLY_CONNECTED_H_
 
+#include <memory>
 #include <vector>
 
 #include "mace/core/future.h"
diff --git a/mace/kernels/matmul.h b/mace/kernels/matmul.h
index f95ce73fd06ffe8ad5513ff6463f2fa5bd5bb0f9..3cba36947f2c8d0facdbb33b7927dd86713f21c3 100644
--- a/mace/kernels/matmul.h
+++ b/mace/kernels/matmul.h
@@ -9,9 +9,10 @@
 #include <arm_neon.h>
 #endif
 
+#include <algorithm>
+#include <memory>
 #include <string>
 #include <vector>
-#include <algorithm>
 
 #include "mace/core/future.h"
 #include "mace/core/runtime/opencl/cl2_header.h"
diff --git a/mace/kernels/pooling.h b/mace/kernels/pooling.h
index 22910b8d2e9e460c323ff5d25a6b7f95b66aef68..f20c059d116330d38bb419c6e5e4910a47a7ce9b 100644
--- a/mace/kernels/pooling.h
+++ b/mace/kernels/pooling.h
@@ -7,6 +7,7 @@
 
 #include <algorithm>
 #include <limits>
+#include <memory>
 #include <vector>
 
 #include "mace/core/future.h"
diff --git a/mace/kernels/resize_bilinear.h b/mace/kernels/resize_bilinear.h
index 212940489b98f5d31d3e3a3bfbadb5612890a0da..cfc3105ba74eed1bdf8b9a13eeaea284e53e518b 100644
--- a/mace/kernels/resize_bilinear.h
+++ b/mace/kernels/resize_bilinear.h
@@ -5,6 +5,7 @@
 #define MACE_KERNELS_RESIZE_BILINEAR_H_
 
 #include <algorithm>
+#include <memory>
 #include <vector>
 
 #include "mace/core/future.h"
diff --git a/mace/kernels/slice.h b/mace/kernels/slice.h
index cc4c0d295342a4f9c5e327805e090bc2d927180a..0fea6b586fd7c81790538fa4a318980cf072a96d 100644
--- a/mace/kernels/slice.h
+++ b/mace/kernels/slice.h
@@ -5,6 +5,7 @@
 #ifndef MACE_KERNELS_SLICE_H_
 #define MACE_KERNELS_SLICE_H_
 
+#include <memory>
 #include <vector>
 
 #include "mace/core/future.h"
diff --git a/mace/kernels/softmax.h b/mace/kernels/softmax.h
index ce8c820db79842741a3ab308044b461f202e0860..48fe37f20b59961be88de7b26d878023329f606b 100644
--- a/mace/kernels/softmax.h
+++ b/mace/kernels/softmax.h
@@ -7,6 +7,7 @@
 
 #include <algorithm>
 #include <functional>
+#include <memory>
 #include <vector>
 
 #include "mace/core/future.h"
diff --git a/mace/kernels/space_to_batch.h b/mace/kernels/space_to_batch.h
index e8e10e504804d52a1703feb5443b4639128b3564..672ccfca558c439e9b8fa16461638c7bf937afee 100644
--- a/mace/kernels/space_to_batch.h
+++ b/mace/kernels/space_to_batch.h
@@ -5,6 +5,7 @@
 #ifndef MACE_KERNELS_SPACE_TO_BATCH_H_
 #define MACE_KERNELS_SPACE_TO_BATCH_H_
 
+#include <memory>
 #include <vector>
 
 #include "mace/core/future.h"
diff --git a/mace/kernels/winograd_transform.h b/mace/kernels/winograd_transform.h
index 0087cefb4c3db4890ca2d610d705e7b205b9c7b1..b6b45ea162c2abb45fcb98353821b9070542ca62 100644
--- a/mace/kernels/winograd_transform.h
+++ b/mace/kernels/winograd_transform.h
@@ -5,6 +5,7 @@
 #ifndef MACE_KERNELS_WINOGRAD_TRANSFORM_H_
 #define MACE_KERNELS_WINOGRAD_TRANSFORM_H_
 
+#include <memory>
 #include <vector>
 
 #include "mace/core/future.h"