We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 22c548b commit 7ffc952Copy full SHA for 7ffc952
CMakeLists.txt
@@ -44,7 +44,15 @@ if(MSVC)
44
string(APPEND CMAKE_CUDA_FLAGS " -Xcudafe --diag_suppress=${diag}")
45
endforeach()
46
CUDA_CONVERT_FLAGS(torch_cpu)
47
- CUDA_CONVERT_FLAGS(torch_cuda)
+ if(TARGET torch_cuda)
48
+ CUDA_CONVERT_FLAGS(torch_cuda)
49
+ endif()
50
+ if(TARGET torch_cuda_cu)
51
+ CUDA_CONVERT_FLAGS(torch_cuda_cu)
52
53
+ if(TARGET torch_cuda_cpp)
54
+ CUDA_CONVERT_FLAGS(torch_cuda_cpp)
55
56
endif()
57
58
0 commit comments