Skip to content

Commit 8b88040

Browse files
committed
pathfinder: add cudla header lookup support
Register cudla as a CTK header so locate_nvidia_header_directory() can find cudla.h in the standard cu13 wheel include directory. In strict header tests, skip cudla on hosts where libnvcudla.so is not available so Tegra setups still exercise the real path without making unsupported hosts fail. Made-with: Cursor
1 parent 1a46a79 commit 8b88040

2 files changed

Lines changed: 10 additions & 0 deletions

File tree

cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -134,6 +134,13 @@ class HeaderDescriptorSpec:
134134
site_packages_dirs=("nvidia/cu13/include", "nvidia/cuda_nvcc/nvvm/include"),
135135
anchor_include_rel_dirs=("nvvm/include",),
136136
),
137+
HeaderDescriptorSpec(
138+
name="cudla",
139+
packaged_with="ctk",
140+
header_basename="cudla.h",
141+
site_packages_dirs=("nvidia/cu13/include",),
142+
available_on_windows=False,
143+
),
137144
# -----------------------------------------------------------------------
138145
# Third-party / separately packaged headers
139146
# -----------------------------------------------------------------------

cuda_pathfinder/tests/test_find_nvidia_headers.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
import pytest
2222

2323
import cuda.pathfinder._headers.find_nvidia_headers as find_nvidia_headers_module
24+
from conftest import skip_if_missing_libnvcudla_so
2425
from cuda.pathfinder import LocatedHeaderDir, find_nvidia_header_directory, locate_nvidia_header_directory
2526
from cuda.pathfinder._dynamic_libs.load_nvidia_dynamic_lib import (
2627
_resolve_system_loaded_abs_path_in_subprocess,
@@ -158,6 +159,8 @@ def test_locate_ctk_headers(info_summary_append, libname):
158159
h_filename = SUPPORTED_HEADERS_CTK[libname]
159160
assert os.path.isfile(os.path.join(hdr_dir, h_filename))
160161
if STRICTNESS == "all_must_work":
162+
if libname == "cudla":
163+
skip_if_missing_libnvcudla_so(libname, timeout=30)
161164
assert hdr_dir is not None
162165

163166

0 commit comments

Comments
 (0)