diff --git a/paddle/pten/kernels/eigen/CMakeLists.txt b/paddle/pten/kernels/eigen/CMakeLists.txt index 27e757256881b276d752b25c42e597c34609c930..200b1101956b499de7ece4e6ba9dd3cfb7e27cf6 100644 --- a/paddle/pten/kernels/eigen/CMakeLists.txt +++ b/paddle/pten/kernels/eigen/CMakeLists.txt @@ -1,7 +1,7 @@ if(WITH_GPU) - nv_library(scale_kernel_eigen SRCS scale_kernel.cc DEPS dense_tensor eigen_function) + nv_library(scale_kernel_eigen SRCS scale_kernel.cc DEPS dense_tensor kernel_context kernel_factory eigen_function) elseif(WITH_ROCM) - hip_library(scale_kernel_eigen SRCS scale_kernel.cc DEPS dense_tensor eigen_function) + hip_library(scale_kernel_eigen SRCS scale_kernel.cc DEPS dense_tensor kernel_context kernel_factory eigen_function) else() - cc_library(scale_kernel_eigen SRCS scale_kernel.cc DEPS dense_tensor eigen_function) + cc_library(scale_kernel_eigen SRCS scale_kernel.cc DEPS dense_tensor kernel_context kernel_factory eigen_function) endif()