proper key.

Author: trivialfis

docs/source/conf.py

@@ -125,6 +125,7 @@ def build_hwloc_xml() -> str:
                 PROJECT_ROOT,
                 "--config-settings=fetch-hwloc=True",
                 f"--config-settings=hwloc-src-dir={hwloc_src_dir}",
+                "--config-settings=with-cuda=False",
                 "--no-deps",
                 "--no-build-isolation",
             ]

hatch/backend.py

@@ -10,7 +10,7 @@
 
 import hatchling.build
 
-from .hook import BUILD_KEY, FETCH_KEY, ROOT_KEY, SRC_KEY
+from .hook import BUILD_KEY, CUDA_KEY, FETCH_KEY, ROOT_KEY, SRC_KEY
 
 
 @contextmanager
@@ -26,6 +26,10 @@ def build_config(config_settings: dict[str, Any] | None) -> Iterator[None]:
             os.environ[SRC_KEY] = config_settings["hwloc-src-dir"]
         if "hwloc-root-dir" in config_settings:
             os.environ[ROOT_KEY] = config_settings["hwloc-root-dir"]
+        if "with-cuda" in config_settings:
+            v = config_settings["with-cuda"]
+            assert v in ("True", "False")
+            os.environ[FETCH_KEY] = CUDA_KEY
     try:
         yield
     finally:

hatch/hook.py

@@ -16,6 +16,7 @@
 from packaging.tags import platform_tags
 
 FETCH_KEY = "PYHWLOC_FETCH_HWLOC"
+CUDA_KEY = "PYHWLOC_WITH_CUDA"
 BUILD_KEY = "PYHWLOC_BUILD_DIR"
 ROOT_KEY = "PYHWLOC_HWLOC_ROOT_DIR"
 SRC_KEY = "PYHWLOC_HWLOC_SRC_DIR"
@@ -169,8 +170,9 @@ def initialize(self, version: str, build_data: dict[str, Any]) -> None:
             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")
+        with_cuda = os.environ.get(CUDA_KEY, None)
+        if with_cuda == "False":
+            cmake_args.append(f"-D{CUDA_KEY}=OFF")
 
         # Source path
         if os.environ.get(SRC_KEY, None) is not None: