diff --git a/paddle/fluid/platform/device/mlu/CMakeLists.txt b/paddle/fluid/platform/device/mlu/CMakeLists.txt
index 43a8f175047502d76c81343e0b93095bc2bf1439..1d4d87b9ec5326effc57d31f6dc3a832f6d8b023 100644
--- a/paddle/fluid/platform/device/mlu/CMakeLists.txt
+++ b/paddle/fluid/platform/device/mlu/CMakeLists.txt
@@ -9,7 +9,7 @@ cc_test(
 cc_library(
   mlu_info
   SRCS mlu_info.cc
-  DEPS enforce glog monitor neuware_lib)
+  DEPS enforce glog malloc monitor neuware_lib)
 cc_library(
   mlu_stream
   SRCS mlu_stream.cc