From 66a9d71a60951718d4467d853e9d341a01428753 Mon Sep 17 00:00:00 2001 From: Chen Weihang Date: Fri, 17 Dec 2021 04:33:43 -0600 Subject: [PATCH] add scale lost deps (#38237) --- paddle/pten/kernels/eigen/CMakeLists.txt | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/paddle/pten/kernels/eigen/CMakeLists.txt b/paddle/pten/kernels/eigen/CMakeLists.txt index 27e75725688..200b1101956 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() -- GitLab