SUPPORTED_HEADERS_CTK_LINUX_ONLY etc. (for cufile)

Author: rwgk

cuda_pathfinder/cuda/pathfinder/_headers/find_nvidia_headers.py

@@ -7,8 +7,8 @@
 import shutil
 from typing import Optional
 
-from cuda.pathfinder._dynamic_libs.supported_nvidia_libs import IS_WINDOWS
 from cuda.pathfinder._headers import supported_nvidia_headers
+from cuda.pathfinder._headers.supported_nvidia_headers import IS_WINDOWS
 from cuda.pathfinder._utils.env_vars import get_cuda_home_or_path
 from cuda.pathfinder._utils.find_sub_dirs import find_sub_dirs_all_sitepackages
 

cuda_pathfinder/cuda/pathfinder/_headers/supported_nvidia_headers.py

@@ -1,12 +1,15 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
-SUPPORTED_HEADERS_CTK = {
+import sys
+
+IS_WINDOWS = sys.platform == "win32"
+
+SUPPORTED_HEADERS_CTK_COMMON = {
     "cccl": "cuda/std/version",
     "cublas": "cublas.h",
     "cudart": "cuda_runtime.h",
     "cufft": "cufft.h",
-    "cufile": "cufile.h",
     "curand": "curand.h",
     "cusolver": "cusolver_common.h",
     "cusparse": "cusparse.h",
@@ -19,6 +22,19 @@
     "nvvm": "nvvm.h",
 }
 
+SUPPORTED_HEADERS_CTK_LINUX_ONLY = {
+    "cufile": "cufile.h",
+}
+SUPPORTED_HEADERS_CTK_LINUX = SUPPORTED_HEADERS_CTK_COMMON | SUPPORTED_HEADERS_CTK_LINUX_ONLY
+
+SUPPORTED_HEADERS_CTK_WINDOWS_ONLY: dict[str, str] = {}
+SUPPORTED_HEADERS_CTK_WINDOWS = SUPPORTED_HEADERS_CTK_COMMON | SUPPORTED_HEADERS_CTK_WINDOWS_ONLY
+
+SUPPORTED_HEADERS_CTK_ALL = (
+    SUPPORTED_HEADERS_CTK_COMMON | SUPPORTED_HEADERS_CTK_LINUX_ONLY | SUPPORTED_HEADERS_CTK_WINDOWS_ONLY
+)
+SUPPORTED_HEADERS_CTK = SUPPORTED_HEADERS_CTK_WINDOWS if IS_WINDOWS else SUPPORTED_HEADERS_CTK_LINUX
+
 SUPPORTED_SITE_PACKAGE_HEADER_DIRS_CTK = {
     "cccl": (
         "cuda/cccl/headers/include",  # cuda-cccl

cuda_pathfinder/tests/test_find_nvidia_headers.py

@@ -21,9 +21,10 @@
 import pytest
 
 from cuda.pathfinder import _find_nvidia_header_directory as find_nvidia_header_directory
-from cuda.pathfinder._dynamic_libs.supported_nvidia_libs import IS_WINDOWS
 from cuda.pathfinder._headers.supported_nvidia_headers import (
+    IS_WINDOWS,
     SUPPORTED_HEADERS_CTK,
+    SUPPORTED_HEADERS_CTK_ALL,
     SUPPORTED_SITE_PACKAGE_HEADER_DIRS_CTK,
 )
 
@@ -65,7 +66,7 @@ def test_find_libname_nvshmem(info_summary_append):
 
 
 def test_supported_headers_site_packages_ctk_consistency():
-    assert tuple(sorted(SUPPORTED_HEADERS_CTK)) == tuple(sorted(SUPPORTED_SITE_PACKAGE_HEADER_DIRS_CTK.keys()))
+    assert tuple(sorted(SUPPORTED_HEADERS_CTK_ALL)) == tuple(sorted(SUPPORTED_SITE_PACKAGE_HEADER_DIRS_CTK.keys()))
 
 
 @pytest.mark.parametrize("libname", SUPPORTED_HEADERS_CTK.keys())