fix compile bug

YuanRisheng · Nov 3, 2021 · bfb723c · bfb723c
1 parent 11bd75b
commit bfb723c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/paddle/pten/CMakeLists.txt b/paddle/pten/CMakeLists.txt
@@ -17,5 +17,7 @@ set(PTEN_DEPS ${PTEN_DEPS} unary binary)
 if(WITH_GPU OR WITH_ROCM)
   set(PTEN_DEPS ${PTEN_DEPS} math_cuda linalg_cuda creation_cuda manipulation_cuda)
 endif()
-
+if(WITH_XPU)
+  set(PTEN_DEPS ${PTEN_DEPS} manipulation_xpu)
+endif()
 cc_library(pten SRCS all.cc DEPS ${PTEN_DEPS})