try to fix compile on windows

PaddlePaddle · Jan 16, 2024 · db73478 · db73478
1 parent 49a42be
commit db73478
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/paddle/phi/kernels/CMakeLists.txt b/paddle/phi/kernels/CMakeLists.txt
@@ -242,14 +242,17 @@ set(cc_search_pattern
     "strings/cpu/*.cc"
     "fusion/*.cc"
     "stride/*.cc"
-    "fusion/cpu/*.cc"
-    "custom/*.cc")
+    "fusion/cpu/*.cc")
 
 if(WITH_MKLDNN)
   set(cc_search_pattern ${cc_search_pattern} "legacy/onednn/*.cc" "onednn/*.cc"
                         "fusion/onednn/*.cc")
 endif()
 
+if(WITH_CUSTOM_DEVICE)
+  set(cc_search_pattern ${cc_search_pattern} "custom/*.cc")
+endif()
+
 file(
   GLOB kernel_cc
   RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}"