diff --git a/paddle/fluid/platform/stream/CMakeLists.txt b/paddle/fluid/platform/stream/CMakeLists.txt index cf219caa9f5c9cea9b8c17f8c56acc8ee53af5df..d38a0c3c21fe5bb21887008a4eeaf5b926c1b9ed 100644 --- a/paddle/fluid/platform/stream/CMakeLists.txt +++ b/paddle/fluid/platform/stream/CMakeLists.txt @@ -5,7 +5,7 @@ ELSE() ENDIF() IF(WITH_GPU OR WITH_ROCM) -cc_library(cuda_stream SRCS cuda_stream.cc DEPS enforce boost ${MKLDNN_CTX_DEPS}) +cc_library(cuda_stream SRCS cuda_stream.cc DEPS enforce boost eigen3 ${MKLDNN_CTX_DEPS}) ENDIF() IF(WITH_ASCEND_CL)