Correctly initialize architectures when building the C++ library from…

… the Python build.
rapidsai · May 16, 2022 · 627cdbd · 627cdbd
1 parent 41acb32
commit 627cdbd
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/python/cuspatial/CMakeLists.txt b/python/cuspatial/CMakeLists.txt
@@ -40,9 +40,16 @@ else()
 endif()
 
 if(NOT cuspatial_FOUND)
-  # TODO: This will not be necessary once we upgrade to CMake 3.22, which will pull in the required
-  # languages for the C++ project even if this project does not require those languges.
+  # TODO: This will not be necessary once we upgrade to CMake 3.22, which will
+  # pull in the required languages for the C++ project even if this project
+  # does not require those languges.
+  include(rapids-cuda)
+  rapids_cuda_init_architectures(cuspatial)
   enable_language(CUDA)
+  # Since cuspatial only enables CUDA optionally we need to manually include the file that
+  # rapids_cuda_init_architectures relies on `project` including.
+  include("${CMAKE_PROJECT_cuspatial_INCLUDE}")
+
   add_subdirectory(../../cpp cuspatial-cpp)
 
   install(TARGETS cuspatial DESTINATION cuspatial/_lib)