diff --git a/paddle/pten/kernels/math/cpu/CMakeLists.txt b/paddle/pten/kernels/math/cpu/CMakeLists.txt
index b30d6d96bdd2d1bf681b5aa1f8b7085fb4ec4c59..235a49a5e4af513b0536f2fbc909bb6188914a89 100644
--- a/paddle/pten/kernels/math/cpu/CMakeLists.txt
+++ b/paddle/pten/kernels/math/cpu/CMakeLists.txt
@@ -1 +1 @@
-cc_library(pten_transpose_cpu SRCS transpose.cc DEPS dense_tensor)
+cc_library(pten_transpose_cpu SRCS transpose.cc DEPS dense_tensor device_context)
diff --git a/paddle/pten/kernels/math/cuda/CMakeLists.txt b/paddle/pten/kernels/math/cuda/CMakeLists.txt
index 5e14f759a6dfcc34d078960ef44cbb0fc67767ee..b0be23bb092416db3601d17e9e1a5d9fdecd472d 100644
--- a/paddle/pten/kernels/math/cuda/CMakeLists.txt
+++ b/paddle/pten/kernels/math/cuda/CMakeLists.txt
@@ -1,5 +1,5 @@
 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()