Add support for H100 gpu's (#669)

Add support for sm90 - enabling implicit to run on H100 gpu's
benfred · Jun 23, 2023 · 91bd56c · 91bd56c
1 parent 2b8ef2d
commit 91bd56c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/implicit/gpu/CMakeLists.txt b/implicit/gpu/CMakeLists.txt
@@ -83,9 +83,13 @@ else()
     else()
         if (${CUDAToolkit_VERSION} VERSION_LESS "11.1.0")
             set_target_properties(_cuda PROPERTIES CUDA_ARCHITECTURES "60;70;80")
-        else()
+        elseif (${CUDAToolkit_VERSION} VERSION_LESS "11.8.0")
             set_target_properties(_cuda PROPERTIES CUDA_ARCHITECTURES "60;70;80;86")
+        else()
+            set_target_properties(_cuda PROPERTIES CUDA_ARCHITECTURES "60;70;80;86;90")
         endif()
+        get_target_property(CUDA_ARCH _cuda CUDA_ARCHITECTURES)
+        message("using cuda architectures ${CUDA_ARCH} for cuda version ${CUDAToolkit_VERSION}")
     endif()
     target_link_libraries(_cuda CUDA::cublas CUDA::curand rmm::rmm)