Fix win arg patch for BUILD_SPLIT_CUDA=ON (#3419)

7ffc9520 · peterjc123 · GitHub · 22c548b0 · 7ffc9520
Unverified Commit 7ffc9520 authored Feb 20, 2021 by peterjc123 Committed by GitHub Feb 19, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 9 additions and 1 deletion

CMakeLists.txt CMakeLists.txt +9 -1

No files found.
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -44,7 +44,15 @@ if(MSVC)
      string(APPEND CMAKE_CUDA_FLAGS " -Xcudafe --diag_suppress=${diag}")
    endforeach()
    CUDA_CONVERT_FLAGS(torch_cpu)
-    CUDA_CONVERT_FLAGS(torch_cuda)
+    if(TARGET torch_cuda)
+      CUDA_CONVERT_FLAGS(torch_cuda)
+    endif()
+    if(TARGET torch_cuda_cu)
+      CUDA_CONVERT_FLAGS(torch_cuda_cu)
+    endif()
+    if(TARGET torch_cuda_cpp)
+      CUDA_CONVERT_FLAGS(torch_cuda_cpp)
+    endif()
  endif()
 endif()