[dev] update cuda cmake (#6574)

fdeda8a8 · Rhett Ying · GitHub · 12f02c3c · fdeda8a8
Unverified Commit fdeda8a8 authored Nov 16, 2023 by Rhett Ying Committed by GitHub Nov 16, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 3 deletions

cmake/modules/CUDA.cmake cmake/modules/CUDA.cmake +2 -3

No files found.
--- a/cmake/modules/CUDA.cmake
+++ b/cmake/modules/CUDA.cmake
@@ -16,12 +16,11 @@ if (CUDA_VERSION_MAJOR GREATER_EQUAL "11")
  list(APPEND dgl_known_gpu_archs "80" "86")
  set(dgl_cuda_arch_ptx "80" "86")
 endif()
-# CMake 3.5 doesn't support VERSION_GREATER_EQUAL
+if (CUDA_VERSION VERSION_GREATER_EQUAL "11.8")
-if (NOT CUDA_VERSION VERSION_LESS "11.8")
  list(APPEND dgl_known_gpu_archs "90")
  set(dgl_cuda_arch_ptx "90")
 endif()
-if (NOT CUDA_VERSION VERSION_LESS "12.0")
+if (CUDA_VERSION VERSION_GREATER_EQUAL "12.0")
  list(REMOVE_ITEM dgl_known_gpu_archs "35")
 endif()