diff --git a/paddle/platform/device_context.h b/paddle/platform/device_context.h
index 94f54d705d1aa18097383f579da1a9cdd62adc48..7de07d06bed885d6529a884fb81fedbdaba78f4a 100644
--- a/paddle/platform/device_context.h
+++ b/paddle/platform/device_context.h
@@ -20,6 +20,7 @@ limitations under the License. */
 #define EIGEN_USE_GPU
 #endif
 #include <paddle/platform/place.h>
+#include <memory>
 #include <unsupported/Eigen/CXX11/Tensor>
 
 namespace paddle {