Skip to content

Commit 8a1dffe

Browse files
committed
Add HeaderDescriptorSpec for mathdx
1 parent 16427f8 commit 8a1dffe

2 files changed

Lines changed: 9 additions & 0 deletions

File tree

cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -153,6 +153,14 @@ class HeaderDescriptorSpec:
153153
conda_targets_layout=False,
154154
use_ctk_root_canary=False,
155155
),
156+
HeaderDescriptorSpec(
157+
name="mathdx",
158+
packaged_with="other",
159+
header_basename="libmathdx.h",
160+
site_packages_dirs=("nvidia/cu13/include", "nvidia/cu12/include"),
161+
conda_targets_layout=False,
162+
use_ctk_root_canary=False,
163+
),
156164
HeaderDescriptorSpec(
157165
name="nvshmem",
158166
packaged_with="other",

cuda_pathfinder/tests/test_find_nvidia_headers.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@
4242
NON_CTK_IMPORTLIB_METADATA_DISTRIBUTIONS_NAMES = {
4343
"cusparseLt": r"^nvidia-cusparselt-.*$",
4444
"cutensor": r"^cutensor-.*$",
45+
"mathdx": r"^nvidia-libmathdx-.*$",
4546
"nvshmem": r"^nvidia-nvshmem-.*$",
4647
}
4748

0 commit comments

Comments
 (0)