add scale lost deps (#38237)

66a9d71a · Chen Weihang · GitHub · d80fe268 · 66a9d71a
显示空白变更内容
内联并排

Showing with 3 addition and 3 deletion

paddle/pten/kernels/eigen/CMakeLists.txt paddle/pten/kernels/eigen/CMakeLists.txt +3 -3

未找到文件。
--- a/paddle/pten/kernels/eigen/CMakeLists.txt
+++ b/paddle/pten/kernels/eigen/CMakeLists.txt
 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()