fix bug that cann't find mkldnn(kunlun) (#30394)

4 years ago · cf786d22ec
parent 5ff4f1ad5e
commit cf786d22ec
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/memory/CMakeLists.txt
+++ b/paddle/fluid/memory/CMakeLists.txt
@ -9,7 +9,7 @@ endif()

 cc_library(malloc SRCS malloc.cc DEPS
    place enforce allocator_facade profiler ${MKLDNN_CTX_DEPS})
-cc_library(memcpy SRCS memcpy.cc DEPS place)
+cc_library(memcpy SRCS memcpy.cc DEPS place device_context)

 cc_library(memory DEPS malloc memcpy)