Coverage for cuda / pathfinder / __init__.py: 100%

1# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. 

2# SPDX-License-Identifier: Apache-2.0 

3 

4"""cuda.pathfinder public APIs""" 

5 

6from cuda.pathfinder._dynamic_libs.load_dl_common import DynamicLibNotFoundError as DynamicLibNotFoundError 

7from cuda.pathfinder._dynamic_libs.load_dl_common import LoadedDL as LoadedDL 

8from cuda.pathfinder._dynamic_libs.load_nvidia_dynamic_lib import load_nvidia_dynamic_lib as load_nvidia_dynamic_lib 

9from cuda.pathfinder._dynamic_libs.supported_nvidia_libs import ( 

10 SUPPORTED_LIBNAMES as SUPPORTED_NVIDIA_LIBNAMES, # noqa: F401 

11) 

12from cuda.pathfinder._headers.find_nvidia_headers import find_nvidia_header_directory as find_nvidia_header_directory 

13from cuda.pathfinder._headers.supported_nvidia_headers import SUPPORTED_HEADERS_CTK as _SUPPORTED_HEADERS_CTK 

14from cuda.pathfinder._version import __version__ as __version__ 

15 

16# Indirections to help Sphinx find the docstrings. 

17#: Mapping from short CUDA Toolkit (CTK) library names to their canonical 

18#: header basenames (used to validate a discovered include directory). 

19#: Example: ``"cublas" → "cublas.h"``. The key set is platform-aware 

20#: (e.g., ``"cufile"`` may be Linux-only). 

21SUPPORTED_HEADERS_CTK = _SUPPORTED_HEADERS_CTK 

22 

23# Backward compatibility: _find_nvidia_header_directory was added in release 1.2.2. 

24# It will be removed in release 1.2.4. 

25_find_nvidia_header_directory = find_nvidia_header_directory