fix cuda_stream missing mkldnn depending error (#34260)

c963a21d · chentianyu03 · GitHub · 1f6f2235 · c963a21d
隐藏空白更改
内联并排

Showing with 7 addition and 1 deletion

paddle/fluid/platform/stream/CMakeLists.txt paddle/fluid/platform/stream/CMakeLists.txt +7 -1

未找到文件。
--- a/paddle/fluid/platform/stream/CMakeLists.txt
+++ b/paddle/fluid/platform/stream/CMakeLists.txt
+IF(WITH_MKLDNN)
+    set(MKLDNN_CTX_DEPS mkldnn)
+ELSE()
+    set(MKLDNN_CTX_DEPS)
+ENDIF()
+
 IF(WITH_GPU OR WITH_ROCM)
-cc_library(cuda_stream SRCS cuda_stream.cc DEPS enforce boost)
+cc_library(cuda_stream SRCS cuda_stream.cc DEPS enforce boost ${MKLDNN_CTX_DEPS})
 ENDIF()

 IF(WITH_ASCEND_CL)