diff --git a/cuda_bindings/cuda/bindings/utils/_ptx_utils.py b/cuda_bindings/cuda/bindings/utils/_ptx_utils.py index 1395fdfd469..ac61524765f 100644 --- a/cuda_bindings/cuda/bindings/utils/_ptx_utils.py +++ b/cuda_bindings/cuda/bindings/utils/_ptx_utils.py @@ -77,9 +77,9 @@ def get_minimal_required_cuda_ver_from_ptx_ver(ptx_version: str) -> int: Examples -------- - >>> get_minimal_required_driver_ver_from_ptx_ver("8.8") + >>> get_minimal_required_cuda_ver_from_ptx_ver("8.8") 12090 - >>> get_minimal_required_driver_ver_from_ptx_ver("7.0") + >>> get_minimal_required_cuda_ver_from_ptx_ver("7.0") 11000 """ try: diff --git a/cuda_bindings/tests/test_utils.py b/cuda_bindings/tests/test_utils.py index c767996bced..20373a2dec7 100644 --- a/cuda_bindings/tests/test_utils.py +++ b/cuda_bindings/tests/test_utils.py @@ -72,6 +72,18 @@ def test_ptx_utils(kernel, actual_ptx_ver, min_cuda_ver): assert cuda_ver == min_cuda_ver +@pytest.mark.agent_authored(model="claude-opus-5") +def test_ptx_utils_docstring_examples(): + """The examples are rendered into the public docs; they have to run.""" + import doctest + + from cuda.bindings.utils import _ptx_utils + + results = doctest.testmod(_ptx_utils, verbose=False, report=False) + assert results.attempted > 0 + assert results.failed == 0 + + @pytest.mark.parametrize( "target", (