Tags: NVIDIA/cuda-python
Tags
`cuda.pathfinder._find_nvidia_header_directory()`: add support for CT… …K libs (#956) * New supported_nvidia_headers.py, starting with just SUPPORTED_HEADERS_CTK, SUPPORTED_SITE_PACKAGE_HEADER_DIRS_CTK dicts. * Add _find_ctk_header_directory(), currently for site-packages only. * Factor out get_cuda_home_or_path() into new cuda/pathfinder/_utils/env_vars.py * Add CUDA_HOME code in find_nvidia_headers.py * Formalize supported_nvidia_headers.CCCL_LIBNAMES * Add CONDA_PREFIX code in find_nvidia_headers.py * Add `shutil.which("nvcc")` code in find_nvidia_headers.py * Cleanup: add _joined_isfile() helper * find_nvidia_header_directory(): return _abs_norm() * Bump pathfinder version to 1.2.3a0 * Replace libcudacxx,cub,thrust with cccl. Add cuda-toolkit[cccl] to nvidia_wheels_cu12, nvidia_wheels_cu13 * SUPPORTED_HEADERS_CTK_LINUX_ONLY etc. (for cufile) * Insert _find_based_on_conda_layout() * Remove `shutil.which("nvcc")` code (it finds all includes on Windows with conda) * Remove cccl code * conda windows support * Replace cusolver_common.h → cusolverDn.h * UserWarning: Both CUDA_HOME and CUDA_PATH are set but differ * Remove `cccl` again in pyproject.toml * Revert "Remove `cccl` again in pyproject.toml" This reverts commit 826398d. * Revert "Remove cccl code" This reverts commit 4d4ff77. * Remove `cuda-cccl` include path in SUPPORTED_SITE_PACKAGE_HEADER_DIRS_CTK * Apply reviewer suggestion: #956 (comment) * Add find_nvidia_header_directory docstring. * Add _SUPPORTED_HEADERS_CTK to public API * Add cuda_pathfinder 1.2.3 Release notes * Remove leading underscores: _SUPPORTED_HEADERS_CTK, _find_nvidia_header_directory * docstring in __init__.py, using `#: ` Sphinx-specific markup A triple-quoted docstring worked for Sphinx but tripped up the check-docstring-first pre-commit check. * Bump pathfinder version to 1.2.3 (for release) and change release date to Sep 17 * Make comment less ambiguous. * Remove subtitle as suggested by reviewer.
Initial version of `cuda.pathfinder._find_nvidia_headers` for `nvshme… …m` (#661) * find_nvidia_headers.py initial version (untested). * Add tests/test_path_finder_find_headers.py, with hard-coded paths. * Better error message: UNKNOWN libname='unknown-libname' * if libname == "nvshmem" and IS_WINDOWS: return None * Move find_nvidia_headers.py → _headers/find_nvidia_headers.py * test_find_nvidia_headers.py: removed hard-wired paths, comments with more complete commands for setting up manual testing. * Make _find_nvidia_header_directory private for now. * test_find_nvidia_headers.py: Move comments with installation commands up. * Add `have_nvidia_nvshmem_package()` function to enable `assert hdr_dir is not None` * Add nvidia-nvshmem-cu12,13 in pyproject.toml * assert site-packages or dist-packages * Add CUDA_PATHFINDER_TEST_FIND_NVIDIA_HEADERS_STRICTNESS * Transfer `ci/`, `.github/` changes from PR #864 * Add CUDA_PATHFINDER_TEST_FIND_NVIDIA_HEADERS_STRICTNESS in `ci/`, `.github/` * reverse=True in sorting of "/usr/include/nvshmem_*" (find newest first) * Fix: assert site-packages or dist-packages only if have_nvidia_nvshmem_package() * pytest.skip("nvshmem has no Windows support.") * Add new cuda/pathfinder/_utils/conda_env.py and use from find_nvidia_headers.py * Add new cuda/pathfinder/_utils/env_vars_for_include.py and use from find_nvidia_headers.py * Revert "Add new cuda/pathfinder/_utils/env_vars_for_include.py and use from find_nvidia_headers.py" This reverts commit c90c393. * Revert "Add new cuda/pathfinder/_utils/conda_env.py and use from find_nvidia_headers.py" This reverts commit eb2e78a. * Bump pathfinder version to 1.2.2 and add release/1.2.2-notes.rst * Remove os.path.isdir() tests that are not strictly needed. * test_find_nvidia_headers.py: remove check for `dist-packages` because a .deb that installs into dist-packages does not exist. * Additional testing
Add support for cuDSS lib (#931) * Fix cudss in conda * Add cudss depenencies * Fix windows * Update version * Add `cudss` to `SITE_PACKAGES_LIBDIRS_LINUX_OTHER`; add `nvidia-cudss-cu12` in pyproject.toml * Release notes update for 1.2.1 * Update pathfinder version to 1.2.1 (for release) * Add nvidia logo in cuda_pathfinder/DESCRIPTION.rst Preview: https://test.pypi.org/project/cuda-pathfinder/1.2.1.dev202508291538/ --------- Co-authored-by: Ralf W. Grosse-Kunstleve <[email protected]>
PreviousNext