diff --git a/paddle/framework/type_defs.h b/paddle/framework/type_defs.h
index a5b9472213..6f65a942ba 100644
--- a/paddle/framework/type_defs.h
+++ b/paddle/framework/type_defs.h
@@ -15,6 +15,7 @@
 #pragma once
 #include <functional>
 #include <map>
+#include <memory>
 #include "paddle/platform/variant.h"
 
 namespace paddle {