cuBLAS

Using real-world code improves tests. This module helps retrieve code from cuBLAS.

class tests.cublas.CuBLAS(cmake_file_api: FileAPI)View on GitHub

Bases: object

CUDA_CUDART: Final[str] = 'CUDA_CUDART'

__init__(cmake_file_api: FileAPI) → NoneView on GitHub

Find the cuBLAS shared library, assuming that the CMake cache populates CUDA_CUDART.

extract(*, arch: NVIDIAArch, cwd: Path, randomly: bool = True) → tuple[Path, ...]View on GitHub

Extract CUDA binary files from libcublas.

Parameters:

• cwd – Where to extract the files.

• randomly – If True, extract only one file (randomly). Otherwise, extract all files.

Raises:

IndexError – If randomly is True but there is no embedded cubin for arch.

libcublas

Shared library path.