w/o cuda.

Author: trivialfis

CMakeLists.txt

@@ -8,6 +8,7 @@ project(pyhwloc LANGUAGES C VERSION 0.0.0)
 list(APPEND CMAKE_MODULE_PATH "${pyhwloc_SOURCE_DIR}/cmake/")
 
 option(PYHWLOC_FETCH_HWLOC "Fetch hwloc from GitHub instead of using system version" OFF)
+option(PYHWLOC_WITH_CUDA "Add CUDA-related components" ON)
 
 set(PYHWLOC_BUILD_HWLOC_CMAKE OFF)
 if(WIN32)
@@ -42,7 +43,7 @@ if(PYHWLOC_FETCH_HWLOC)
         -G ${CMAKE_GENERATOR}
         -DCMAKE_INSTALL_PREFIX=${PYHWLOC_OUTPUT_DIR}
         -DBUILD_SHARED_LIBS=ON
-        -DHWLOC_WITH_CUDA=ON
+        -DHWLOC_WITH_CUDA=${PYHWLOC_WITH_CUDA}
         -DHWLOC_ENABLE_PLUGINS=ON
         WORKING_DIRECTORY ${hwloc_SOURCE_DIR}
       )
@@ -94,26 +95,29 @@ else()
 endif()
 
 message(STATUS "HWLOC_LIBRARY: ${HWLOC_LIBRARY}")
-find_package(CUDAToolkit REQUIRED COMPONENTS nvml cuda_driver cudart_static)
 
 target_link_libraries(pyhwloc PRIVATE ${HWLOC_LIBRARY})
 target_include_directories(pyhwloc PRIVATE ${HWLOC_INCLUDE_DIR})
 list(APPEND PYHWLOC_LIBS pyhwloc)
 
-add_library(pyhwloc_cuda SHARED src/ext/cudr.c)
-target_link_libraries(pyhwloc_cuda PRIVATE CUDA::cuda_driver ${HWLOC_LIBRARY})
-target_include_directories(pyhwloc_cuda PRIVATE ${HWLOC_INCLUDE_DIR})
-list(APPEND PYHWLOC_LIBS pyhwloc_cuda)
+if(PYHWLOC_WITH_CUDA)
+  find_package(CUDAToolkit REQUIRED COMPONENTS nvml cuda_driver cudart_static)
 
-add_library(pyhwloc_cudart SHARED src/ext/cudart.c)
-target_link_libraries(pyhwloc_cudart PRIVATE CUDA::cudart ${HWLOC_LIBRARY})
-target_include_directories(pyhwloc_cudart PRIVATE ${HWLOC_INCLUDE_DIR})
-list(APPEND PYHWLOC_LIBS pyhwloc_cudart)
+  add_library(pyhwloc_cuda SHARED src/ext/cudr.c)
+  target_link_libraries(pyhwloc_cuda PRIVATE CUDA::cuda_driver ${HWLOC_LIBRARY})
+  target_include_directories(pyhwloc_cuda PRIVATE ${HWLOC_INCLUDE_DIR})
+  list(APPEND PYHWLOC_LIBS pyhwloc_cuda)
 
-add_library(pyhwloc_nvml SHARED src/ext/nvml.c)
-target_link_libraries(pyhwloc_nvml PRIVATE CUDA::nvml ${HWLOC_LIBRARY})
-target_include_directories(pyhwloc_nvml PRIVATE ${HWLOC_INCLUDE_DIR})
-list(APPEND PYHWLOC_LIBS pyhwloc_nvml)
+  add_library(pyhwloc_cudart SHARED src/ext/cudart.c)
+  target_link_libraries(pyhwloc_cudart PRIVATE CUDA::cudart ${HWLOC_LIBRARY})
+  target_include_directories(pyhwloc_cudart PRIVATE ${HWLOC_INCLUDE_DIR})
+  list(APPEND PYHWLOC_LIBS pyhwloc_cudart)
+
+  add_library(pyhwloc_nvml SHARED src/ext/nvml.c)
+  target_link_libraries(pyhwloc_nvml PRIVATE CUDA::nvml ${HWLOC_LIBRARY})
+  target_include_directories(pyhwloc_nvml PRIVATE ${HWLOC_INCLUDE_DIR})
+  list(APPEND PYHWLOC_LIBS pyhwloc_nvml)
+endif()
 
 include(GenerateExportHeader)
 

hatch/hook.py

@@ -168,6 +168,10 @@ def initialize(self, version: str, build_data: dict[str, Any]) -> None:
                 raise FileNotFoundError(root_dir)
             cmake_args.append(f"-DHWLOC_ROOT={root_dir}")
 
+        # No CUDA in RTD
+        if os.environ.get("READTHEDOCS", None) == "True":
+            cmake_args.append("-DPYHWLOC_WITH_CUDA=OFF")
+
         # Source path
         if os.environ.get(SRC_KEY, None) is not None:
             src_dir = os.environ[SRC_KEY]