diff --git a/paddle/fluid/operators/cvm_op.cc b/paddle/fluid/operators/cvm_op.cc
index 5f0413797ea1df7914a4455967743756ca6a2208..0d66c10599dbbc7a8c3fe6be6ad38d2601fcc356 100644
--- a/paddle/fluid/operators/cvm_op.cc
+++ b/paddle/fluid/operators/cvm_op.cc
@@ -13,6 +13,7 @@ See the License for the specific language governing permissions and
 limitations under the License. */
 
 #include "paddle/fluid/operators/cvm_op.h"
+#include <memory>
 #include "paddle/fluid/operators/math/math_function.h"
 
 namespace paddle {