add pten_transpose dependence device_context (#37705)

5747fd1e · chentianyu03 · GitHub · d93ee063 · 5747fd1e · 5747fd1e
隐藏空白更改
内联并排

Showing with 3 addition and 3 deletion

paddle/pten/kernels/math/cpu/CMakeLists.txt paddle/pten/kernels/math/cpu/CMakeLists.txt +1 -1

paddle/pten/kernels/math/cuda/CMakeLists.txt paddle/pten/kernels/math/cuda/CMakeLists.txt +2 -2

未找到文件。
--- a/paddle/pten/kernels/math/cpu/CMakeLists.txt
+++ b/paddle/pten/kernels/math/cpu/CMakeLists.txt
-cc_library(pten_transpose_cpu SRCS transpose.cc DEPS dense_tensor)
+cc_library(pten_transpose_cpu SRCS transpose.cc DEPS dense_tensor device_context)
--- a/paddle/pten/kernels/math/cuda/CMakeLists.txt
+++ b/paddle/pten/kernels/math/cuda/CMakeLists.txt
 if(WITH_GPU)
-  nv_library(pten_transpose_cuda SRCS transpose.cu DEPS dense_tensor malloc)
+  nv_library(pten_transpose_cuda SRCS transpose.cu DEPS dense_tensor malloc device_context)
 elseif(WITH_ROCM)
-  hip_library(pten_transpose_cuda SRCS transpose.cu DEPS dense_tensor malloc)
+  hip_library(pten_transpose_cuda SRCS transpose.cu DEPS dense_tensor malloc device_context)
 endif()