feat: enable Windows build

XuehaiPan · XuehaiPan · commit abdb3db9591b · 2022-09-05T16:39:23.000+08:00
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -55,7 +55,7 @@ if(USE_FP16)
 endif()
 
 find_package(CUDA)
-if(CUDA_FOUND AND NOT WIN32)
+if(CUDA_FOUND)
     message(STATUS "Found CUDA, enabling CUDA support.")
     enable_language(CUDA)
     set(CMAKE_CUDA_STANDARD "${CMAKE_CXX_STANDARD}")
@@ -110,10 +110,8 @@ if(CUDA_FOUND AND NOT WIN32)
     else()
         set(CMAKE_CUDA_FLAGS_RELEASE "${CMAKE_CUDA_FLAGS_RELEASE} -O3")
     endif()
-elseif(NOT CUDA_FOUND)
-    message(STATUS "CUDA not found, build for CPU-only.")
 else()
-    message(STATUS "CUDA found, but build for CPU-only on Windows.")
+    message(STATUS "CUDA not found, build for CPU-only.")
 endif()
 
 function(system)
