fix bugs for kunlun ci

YuanRisheng · Nov 2, 2021 · 853b328 · 853b328
1 parent 8b8d114
commit 853b328
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/paddle/pten/api/CMakeLists.txt b/paddle/pten/api/CMakeLists.txt
@@ -4,5 +4,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})