if libname == "nvshmem" and IS_WINDOWS: return None

Author: rwgk

cuda_bindings/cuda/bindings/_path_finder/find_nvidia_headers.py

@@ -14,7 +14,9 @@ def find_nvidia_header_directory(libname: str) -> str:
     if libname != "nvshmem":
         raise RuntimeError(f"UNKNOWN {libname=}")
 
-    assert not IS_WINDOWS
+    if libname == "nvshmem" and IS_WINDOWS:
+        # nvshmem has no Windows support.
+        return None
 
     # Installed from a wheel
     nvidia_sub_dirs = ("nvidia", "nvshmem", "include")

cuda_bindings/tests/test_path_finder_find_headers.py

@@ -11,14 +11,20 @@ def test_find_nvidia_header_directory(info_summary_append):
         find_nvidia_headers.find_nvidia_header_directory("unknown-libname")
 
     hdr_dir = find_nvidia_headers.find_nvidia_header_directory("nvshmem")
+    # TODO: Find ways to test more meaningfully, and how to avoid HARD-WIRED PATHS in particular.
     assert hdr_dir in [
         # pip install nvidia-nvshmem-cu12
         "/home/rgrossekunst/forked/cuda-python/venvs/scratch/lib/python3.12/site-packages/nvidia/nvshmem/include",
+        #
         # conda create -y -n nvshmem python=3.12
         # conda activate nvshmem
         # conda install -y conda-forge::libnvshmem3 conda-forge::libnvshmem-dev
         "/home/rgrossekunst/miniforge3/envs/nvshmem/include",
+        #
         # sudo apt install libnvshmem3-cuda-12 libnvshmem3-dev-cuda-12
         "/usr/include/nvshmem_12",
+        #
+        # nvshmem not available
+        None,
     ]
     info_summary_append(f"{hdr_dir=!r}")