Skip to content

Commit 318dc77

Browse files
leofangclaude
andcommitted
Skip cutlass/cute header tests on Python 3.15
nvidia-cutlass is not installed on 3.15 (scipy has no 3.15 wheel), so the pathfinder header discovery tests for cutlass and cute will always fail under all_must_work strictness. Skip those parametrized cases on 3.15+. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1 parent 839f785 commit 318dc77

1 file changed

Lines changed: 13 additions & 1 deletion

File tree

cuda_pathfinder/tests/test_find_nvidia_headers.py

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
import importlib.metadata
1717
import os
1818
import re
19+
import sys
1920
from pathlib import Path
2021

2122
import pytest
@@ -112,7 +113,18 @@ def _fake_cudart_canary_abs_path(ctk_root: Path) -> str:
112113
return str(ctk_root / "lib64" / "libcudart.so.13")
113114

114115

115-
@pytest.mark.parametrize("libname", SUPPORTED_HEADERS_NON_CTK.keys())
116+
# TODO: remove the Python 3.15 guard once 3.15 is officially supported
117+
_CUTLASS_SKIP = pytest.mark.skipif(
118+
sys.version_info >= (3, 15),
119+
reason="nvidia-cutlass not available on Python 3.15 (scipy missing)",
120+
)
121+
_NON_CTK_HEADER_PARAMS = [
122+
pytest.param(name, marks=_CUTLASS_SKIP) if name in ("cutlass", "cute") else name
123+
for name in SUPPORTED_HEADERS_NON_CTK.keys()
124+
]
125+
126+
127+
@pytest.mark.parametrize("libname", _NON_CTK_HEADER_PARAMS)
116128
def test_locate_non_ctk_headers(info_summary_append, libname):
117129
hdr_dir = find_nvidia_header_directory(libname)
118130
located_hdr_dir = locate_nvidia_header_directory(libname)

0 commit comments

Comments
 (0)