From d287dae8eb4c4972adf29119d034c8321b3931a4 Mon Sep 17 00:00:00 2001 From: "Corey J. Nolet" Date: Fri, 31 Jul 2026 13:18:50 -0400 Subject: [PATCH] Revert "Merge branch 'main' into release/26.08" This reverts commit ae5226c7ae3ba7e70794b4b63f1bf6cd33750499, reversing changes made to 2db820e7b90c5807c8e532ccd4da0ceba7667c7f. --- .../cuda12.9-conda/devcontainer.json | 6 +- .devcontainer/cuda12.9-pip/devcontainer.json | 8 +- .../cuda13.3-conda/devcontainer.json | 6 +- .devcontainer/cuda13.3-pip/devcontainer.json | 8 +- .github/CODEOWNERS | 5 +- .github/workflows/build.yaml | 46 +- .github/workflows/labeler.yml | 20 +- .github/workflows/pr.yaml | 65 ++- .github/workflows/publish-rust.yaml | 2 +- .github/workflows/test.yaml | 30 +- .../trigger-breaking-change-alert.yaml | 2 +- .pre-commit-config.yaml | 411 +++++++++--------- .yamllint.yaml | 37 -- RAPIDS_BRANCH | 2 +- VERSION | 2 +- cmake/RAPIDS.cmake | 2 - .../all_cuda-129_arch-aarch64.yaml | 5 +- .../all_cuda-129_arch-x86_64.yaml | 5 +- .../all_cuda-133_arch-aarch64.yaml | 5 +- .../all_cuda-133_arch-x86_64.yaml | 5 +- .../bench_ann_cuda-129_arch-aarch64.yaml | 9 +- .../bench_ann_cuda-129_arch-x86_64.yaml | 9 +- .../bench_ann_cuda-133_arch-aarch64.yaml | 9 +- .../bench_ann_cuda-133_arch-x86_64.yaml | 9 +- .../go_cuda-129_arch-aarch64.yaml | 5 +- .../environments/go_cuda-129_arch-x86_64.yaml | 5 +- .../go_cuda-133_arch-aarch64.yaml | 5 +- .../environments/go_cuda-133_arch-x86_64.yaml | 5 +- .../rust_cuda-129_arch-aarch64.yaml | 4 +- .../rust_cuda-129_arch-x86_64.yaml | 4 +- .../rust_cuda-133_arch-aarch64.yaml | 4 +- .../rust_cuda-133_arch-x86_64.yaml | 4 +- conda/recipes/cuvs-bench-cpu/recipe.yaml | 2 +- conda/recipes/cuvs-bench/recipe.yaml | 2 +- conda/recipes/cuvs/recipe.yaml | 2 +- conda/recipes/libcuvs/recipe.yaml | 12 +- dependencies.yaml | 33 +- examples/go/README.md | 2 +- java/benchmarks/pom.xml | 4 +- java/build.sh | 2 +- java/cuvs-java/pom.xml | 2 +- java/examples/README.md | 6 +- java/examples/pom.xml | 4 +- python/cuvs/pyproject.toml | 10 +- .../config/algos/cuvs_cagra_hnswlib.yaml | 2 +- .../config/algos/faiss_cpu_hnsw_flat.yaml | 4 +- .../config/algos/faiss_cpu_ivf_flat.yaml | 4 +- .../config/algos/faiss_gpu_cagra.yaml | 2 +- .../config/algos/faiss_gpu_cagra_hnsw.yaml | 2 +- .../config/algos/faiss_gpu_ivf_flat.yaml | 10 +- .../config/algos/faiss_gpu_ivf_pq.yaml | 40 +- python/cuvs_bench/pyproject.toml | 2 +- python/libcuvs/pyproject.toml | 8 +- rust/Cargo.toml | 4 +- 54 files changed, 425 insertions(+), 478 deletions(-) delete mode 100644 .yamllint.yaml diff --git a/.devcontainer/cuda12.9-conda/devcontainer.json b/.devcontainer/cuda12.9-conda/devcontainer.json index c2eed35f1f..48f66b2a50 100644 --- a/.devcontainer/cuda12.9-conda/devcontainer.json +++ b/.devcontainer/cuda12.9-conda/devcontainer.json @@ -5,7 +5,7 @@ "args": { "CUDA": "12.9", "PYTHON_PACKAGE_MANAGER": "conda", - "BASE": "rapidsai/devcontainers:26.10-cpp-mambaforge" + "BASE": "rapidsai/devcontainers:26.08-cpp-mambaforge" }, "cacheFrom": [ "ghcr.io/rapidsai/cuvs/devcontainer:26.08-cuda12.9-conda" @@ -14,13 +14,13 @@ "runArgs": [ "--rm", "--name", - "${localEnv:USER:anon}-rapids-${localWorkspaceFolderBasename}-26.10-cuda12.9-conda", + "${localEnv:USER:anon}-rapids-${localWorkspaceFolderBasename}-26.08-cuda12.9-conda", "--ulimit", "nofile=500000" ], "hostRequirements": {"gpu": "optional"}, "features": { - "ghcr.io/rapidsai/devcontainers/features/rapids-build-utils:26.10": {} + "ghcr.io/rapidsai/devcontainers/features/rapids-build-utils:26.8": {} }, "overrideFeatureInstallOrder": [ "ghcr.io/rapidsai/devcontainers/features/rapids-build-utils" diff --git a/.devcontainer/cuda12.9-pip/devcontainer.json b/.devcontainer/cuda12.9-pip/devcontainer.json index 60c62ef968..0bf75c1b4b 100644 --- a/.devcontainer/cuda12.9-pip/devcontainer.json +++ b/.devcontainer/cuda12.9-pip/devcontainer.json @@ -5,7 +5,7 @@ "args": { "CUDA": "12.9", "PYTHON_PACKAGE_MANAGER": "pip", - "BASE": "rapidsai/devcontainers:26.10-cpp-cuda12.9-ucx1.19.0-openmpi5.0.10" + "BASE": "rapidsai/devcontainers:26.08-cpp-cuda12.9-ucx1.19.0-openmpi5.0.10" }, "cacheFrom": [ "ghcr.io/rapidsai/cuvs/devcontainer:26.08-cuda12.9-pip" @@ -14,20 +14,20 @@ "runArgs": [ "--rm", "--name", - "${localEnv:USER:anon}-rapids-${localWorkspaceFolderBasename}-26.10-cuda12.9-pip", + "${localEnv:USER:anon}-rapids-${localWorkspaceFolderBasename}-26.08-cuda12.9-pip", "--ulimit", "nofile=500000" ], "hostRequirements": {"gpu": "optional"}, "features": { - "ghcr.io/rapidsai/devcontainers/features/cuda:26.10": { + "ghcr.io/rapidsai/devcontainers/features/cuda:26.8": { "version": "12.9", "installcuBLAS": true, "installcuSOLVER": true, "installcuRAND": true, "installcuSPARSE": true }, - "ghcr.io/rapidsai/devcontainers/features/rapids-build-utils:26.10": {} + "ghcr.io/rapidsai/devcontainers/features/rapids-build-utils:26.8": {} }, "overrideFeatureInstallOrder": [ "ghcr.io/rapidsai/devcontainers/features/ucx", diff --git a/.devcontainer/cuda13.3-conda/devcontainer.json b/.devcontainer/cuda13.3-conda/devcontainer.json index dc3038689f..4c9fdd9501 100644 --- a/.devcontainer/cuda13.3-conda/devcontainer.json +++ b/.devcontainer/cuda13.3-conda/devcontainer.json @@ -5,7 +5,7 @@ "args": { "CUDA": "13.3", "PYTHON_PACKAGE_MANAGER": "conda", - "BASE": "rapidsai/devcontainers:26.10-cpp-mambaforge" + "BASE": "rapidsai/devcontainers:26.08-cpp-mambaforge" }, "cacheFrom": [ "ghcr.io/rapidsai/cuvs/devcontainer:26.08-cuda13.3-conda" @@ -14,13 +14,13 @@ "runArgs": [ "--rm", "--name", - "${localEnv:USER:anon}-rapids-${localWorkspaceFolderBasename}-26.10-cuda13.3-conda", + "${localEnv:USER:anon}-rapids-${localWorkspaceFolderBasename}-26.08-cuda13.3-conda", "--ulimit", "nofile=500000" ], "hostRequirements": {"gpu": "optional"}, "features": { - "ghcr.io/rapidsai/devcontainers/features/rapids-build-utils:26.10": {} + "ghcr.io/rapidsai/devcontainers/features/rapids-build-utils:26.8": {} }, "overrideFeatureInstallOrder": [ "ghcr.io/rapidsai/devcontainers/features/rapids-build-utils" diff --git a/.devcontainer/cuda13.3-pip/devcontainer.json b/.devcontainer/cuda13.3-pip/devcontainer.json index ef47a3c6c4..94dbbe3c1b 100644 --- a/.devcontainer/cuda13.3-pip/devcontainer.json +++ b/.devcontainer/cuda13.3-pip/devcontainer.json @@ -5,7 +5,7 @@ "args": { "CUDA": "13.3", "PYTHON_PACKAGE_MANAGER": "pip", - "BASE": "rapidsai/devcontainers:26.10-cpp-cuda13.3-ucx1.19.0-openmpi5.0.10" + "BASE": "rapidsai/devcontainers:26.08-cpp-cuda13.3-ucx1.19.0-openmpi5.0.10" }, "cacheFrom": [ "ghcr.io/rapidsai/cuvs/devcontainer:26.08-cuda13.3-pip" @@ -14,20 +14,20 @@ "runArgs": [ "--rm", "--name", - "${localEnv:USER:anon}-rapids-${localWorkspaceFolderBasename}-26.10-cuda13.3-pip", + "${localEnv:USER:anon}-rapids-${localWorkspaceFolderBasename}-26.08-cuda13.3-pip", "--ulimit", "nofile=500000" ], "hostRequirements": {"gpu": "optional"}, "features": { - "ghcr.io/rapidsai/devcontainers/features/cuda:26.10": { + "ghcr.io/rapidsai/devcontainers/features/cuda:26.8": { "version": "13.3", "installcuBLAS": true, "installcuSOLVER": true, "installcuRAND": true, "installcuSPARSE": true }, - "ghcr.io/rapidsai/devcontainers/features/rapids-build-utils:26.10": {} + "ghcr.io/rapidsai/devcontainers/features/rapids-build-utils:26.8": {} }, "overrideFeatureInstallOrder": [ "ghcr.io/rapidsai/devcontainers/features/ucx", diff --git a/.github/CODEOWNERS b/.github/CODEOWNERS index cba4d15f88..b897d49c50 100755 --- a/.github/CODEOWNERS +++ b/.github/CODEOWNERS @@ -28,11 +28,10 @@ CMakeLists.txt @NVIDIA/cuvs-cmake-codeowners cpp/scripts/run-cmake-format.sh @NVIDIA/cuvs-cmake-codeowners #CI code owners -/.coderabbit.yaml @NVIDIA/adi-ci-codeowners /.github/ @NVIDIA/adi-ci-codeowners -/.shellcheckrc @NVIDIA/adi-ci-codeowners -/.yamllint.yaml @NVIDIA/adi-ci-codeowners /ci/ @NVIDIA/adi-ci-codeowners +/.shellcheckrc @NVIDIA/adi-ci-codeowners +/.coderabbit.yaml @NVIDIA/adi-ci-codeowners #packaging code owners /.pre-commit-config.yaml @NVIDIA/adi-packaging-codeowners diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index c209d0c0ad..ec53a86eda 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -43,7 +43,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/conda-cpp-build.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/conda-cpp-build.yaml@release/26.08 with: build_type: ${{ inputs.build_type || 'branch' }} branch: ${{ inputs.branch }} @@ -55,7 +55,7 @@ jobs: rocky8-clib-standalone-build-matrix: permissions: contents: read - uses: rapidsai/shared-workflows/.github/workflows/compute-matrix.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/compute-matrix.yaml@release/26.08 with: build_type: pull-request matrix_name: conda-cpp-build @@ -68,7 +68,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@release/26.08 strategy: fail-fast: false matrix: ${{ fromJSON(needs.rocky8-clib-standalone-build-matrix.outputs.matrix) }} @@ -77,7 +77,7 @@ jobs: branch: ${{ inputs.branch }} arch: "${{matrix.ARCH}}" date: ${{ inputs.date }} - container_image: "rapidsai/ci-wheel:26.10-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" + container_image: "rapidsai/ci-wheel:26.08-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" node_type: "cpu16" requires_license_builder: true script: "ci/build_standalone_c.sh" @@ -88,7 +88,7 @@ jobs: needs: cpp-build permissions: contents: read - uses: rapidsai/shared-workflows/.github/workflows/compute-matrix.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/compute-matrix.yaml@release/26.08 with: build_type: pull-request matrix_name: conda-cpp-build @@ -102,7 +102,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@release/26.08 # Artifacts are not published from these jobs, so it's safe to run for multiple CUDA versions. # If these jobs start producing artifacts, the names will have to differentiate between CUDA versions. strategy: @@ -113,7 +113,7 @@ jobs: branch: ${{ inputs.branch }} arch: "amd64" date: ${{ inputs.date }} - container_image: "rapidsai/ci-conda:26.10-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" + container_image: "rapidsai/ci-conda:26.08-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" node_type: "gpu-l4-latest-1" script: "ci/build_rust.sh" sha: ${{ inputs.sha }} @@ -121,7 +121,7 @@ jobs: needs: cpp-build permissions: contents: read - uses: rapidsai/shared-workflows/.github/workflows/compute-matrix.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/compute-matrix.yaml@release/26.08 with: build_type: pull-request matrix_name: conda-cpp-build @@ -135,7 +135,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@release/26.08 # Artifacts are not published from these jobs, so it's safe to run for multiple CUDA versions. # If these jobs start producing artifacts, the names will have to differentiate between CUDA versions. strategy: @@ -146,7 +146,7 @@ jobs: branch: ${{ inputs.branch }} arch: "amd64" date: ${{ inputs.date }} - container_image: "rapidsai/ci-conda:26.10-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" + container_image: "rapidsai/ci-conda:26.08-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" node_type: "gpu-l4-latest-1" script: "ci/build_go.sh" sha: ${{ inputs.sha }} @@ -154,7 +154,7 @@ jobs: needs: cpp-build permissions: contents: read - uses: rapidsai/shared-workflows/.github/workflows/compute-matrix.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/compute-matrix.yaml@release/26.08 with: build_type: pull-request matrix_name: conda-cpp-build @@ -168,7 +168,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@release/26.08 # Artifacts are not published from these jobs, so it's safe to run for multiple CUDA versions. # If these jobs start producing artifacts, the names will have to differentiate between CUDA versions. strategy: @@ -179,7 +179,7 @@ jobs: branch: ${{ inputs.branch }} arch: "amd64" date: ${{ inputs.date }} - container_image: "rapidsai/ci-conda:26.10-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" + container_image: "rapidsai/ci-conda:26.08-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" script: "ci/build_java.sh" artifact-name: "cuvs-java-cuda${{ matrix.CUDA_VER }}" file_to_upload: "java/cuvs-java/target/" @@ -193,7 +193,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/conda-python-build.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/conda-python-build.yaml@release/26.08 with: build_type: ${{ inputs.build_type || 'branch' }} branch: ${{ inputs.branch }} @@ -210,7 +210,7 @@ jobs: id-token: write packages: read pull-requests: read - uses: rapidsai/shared-workflows/.github/workflows/conda-upload-packages.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/conda-upload-packages.yaml@release/26.08 secrets: CONDA_RAPIDSAI_NIGHTLY_TOKEN: ${{ secrets.CONDA_RAPIDSAI_NIGHTLY_TOKEN }} CONDA_RAPIDSAI_TOKEN: ${{ secrets.CONDA_RAPIDSAI_TOKEN }} @@ -225,7 +225,7 @@ jobs: permissions: contents: read secrets: - GPUTESTER_CRATES_TOKEN: ${{ secrets.GPUTESTER_CRATES_TOKEN }} + GPUTESTER_CRATES_TOKEN: ${{ secrets.GPUTESTER_CRATES_TOKEN }} uses: ./.github/workflows/publish-rust.yaml docs-build: if: github.ref_type == 'branch' @@ -236,12 +236,12 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@release/26.08 with: arch: "amd64" branch: ${{ inputs.branch }} build_type: ${{ inputs.build_type || 'branch' }} - container_image: "rapidsai/ci-conda:26.10-latest" + container_image: "rapidsai/ci-conda:26.08-latest" date: ${{ inputs.date }} node_type: "gpu-l4-latest-1" script: "ci/build_docs.sh" @@ -254,7 +254,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/wheels-build.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/wheels-build.yaml@release/26.08 with: build_type: ${{ inputs.build_type || 'branch' }} branch: ${{ inputs.branch }} @@ -274,7 +274,7 @@ jobs: id-token: write packages: read pull-requests: read - uses: rapidsai/shared-workflows/.github/workflows/wheels-publish.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/wheels-publish.yaml@release/26.08 secrets: CONDA_RAPIDSAI_WHEELS_NIGHTLY_TOKEN: ${{ secrets.CONDA_RAPIDSAI_WHEELS_NIGHTLY_TOKEN }} RAPIDSAI_PYPI_TOKEN: ${{ secrets.RAPIDSAI_PYPI_TOKEN }} @@ -295,7 +295,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/wheels-build.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/wheels-build.yaml@release/26.08 with: build_type: ${{ inputs.build_type || 'branch' }} branch: ${{ inputs.branch }} @@ -315,7 +315,7 @@ jobs: id-token: write packages: read pull-requests: read - uses: rapidsai/shared-workflows/.github/workflows/wheels-publish.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/wheels-publish.yaml@release/26.08 secrets: CONDA_RAPIDSAI_WHEELS_NIGHTLY_TOKEN: ${{ secrets.CONDA_RAPIDSAI_WHEELS_NIGHTLY_TOKEN }} RAPIDSAI_PYPI_TOKEN: ${{ secrets.RAPIDSAI_PYPI_TOKEN }} @@ -330,7 +330,7 @@ jobs: devcontainers: name: Build devcontainers secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/build-devcontainers.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/build-devcontainers.yaml@release/26.08 permissions: packages: write with: diff --git a/.github/workflows/labeler.yml b/.github/workflows/labeler.yml index 502f673735..6f1e52d8d7 100644 --- a/.github/workflows/labeler.yml +++ b/.github/workflows/labeler.yml @@ -3,15 +3,15 @@ name: "Pull Request Labeler" # incorrectly. These usages are ok and don't execute any PR-specific # code (and so aren't susceptible to exploits from forked PRs) on: # zizmor: ignore[dangerous-triggers] - - pull_request_target +- pull_request_target jobs: - triage: - runs-on: ubuntu-latest - permissions: - contents: read - pull-requests: write - steps: - - uses: actions/labeler@8558fd74291d67161a8a78ce36a881fa63b766a9 # v5 - with: - repo-token: "${{ secrets.GITHUB_TOKEN }}" + triage: + runs-on: ubuntu-latest + permissions: + contents: read + pull-requests: write + steps: + - uses: actions/labeler@8558fd74291d67161a8a78ce36a881fa63b766a9 # v5 + with: + repo-token: "${{ secrets.GITHUB_TOKEN }}" diff --git a/.github/workflows/pr.yaml b/.github/workflows/pr.yaml index 12d2d7f8eb..84b70af1fc 100644 --- a/.github/workflows/pr.yaml +++ b/.github/workflows/pr.yaml @@ -35,7 +35,7 @@ jobs: - wheel-tests-cuvs - devcontainer - telemetry-setup - uses: rapidsai/shared-workflows/.github/workflows/pr-builder.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/pr-builder.yaml@release/26.08 permissions: contents: read if: always() @@ -78,7 +78,7 @@ jobs: contents: read packages: read pull-requests: read - uses: rapidsai/shared-workflows/.github/workflows/changed-files.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/changed-files.yaml@release/26.08 with: files_yaml: | build_docs: @@ -97,7 +97,6 @@ jobs: - '!.github/workflows/trigger-breaking-change-alert.yaml' - '!.pre-commit-config.yaml' - '!.shellcheckrc' - - '!.yamllint.yaml' - '!Dockerfile' - '!SECURITY.md' - '!ci/build_wheel*.sh' @@ -123,7 +122,6 @@ jobs: - '!.github/workflows/trigger-breaking-change-alert.yaml' - '!.pre-commit-config.yaml' - '!.shellcheckrc' - - '!.yamllint.yaml' - '!Dockerfile' - '!README.md' - '!SECURITY.md' @@ -162,7 +160,6 @@ jobs: - '!.github/workflows/trigger-breaking-change-alert.yaml' - '!.pre-commit-config.yaml' - '!.shellcheckrc' - - '!.yamllint.yaml' - '!CODEOWNERS' - '!Dockerfile' - '!README.md' @@ -212,7 +209,6 @@ jobs: - '!.github/workflows/update-c-abi-baseline.yaml' - '!.pre-commit-config.yaml' - '!.shellcheckrc' - - '!.yamllint.yaml' - '!ci/build_go.sh' - '!ci/build_java.sh' - '!ci/build_rust.sh' @@ -255,7 +251,6 @@ jobs: - '!.github/workflows/trigger-breaking-change-alert.yaml' - '!.pre-commit-config.yaml' - '!.shellcheckrc' - - '!.yamllint.yaml' - '!ci/build_cpp.sh' - '!ci/build_docs.sh' - '!ci/build_go.sh' @@ -303,7 +298,6 @@ jobs: - '!.github/workflows/trigger-breaking-change-alert.yaml' - '!.pre-commit-config.yaml' - '!.shellcheckrc' - - '!.yamllint.yaml' - '!CODEOWNERS' - '!Dockerfile' - '!README.md' @@ -346,7 +340,6 @@ jobs: - '!.github/workflows/trigger-breaking-change-alert.yaml' - '!.pre-commit-config.yaml' - '!.shellcheckrc' - - '!.yamllint.yaml' - '!CODEOWNERS' - '!Dockerfile' - '!README.md' @@ -376,7 +369,7 @@ jobs: - '!thirdparty/LICENSES/**' checks: needs: telemetry-setup - uses: rapidsai/shared-workflows/.github/workflows/checks.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/checks.yaml@release/26.08 permissions: contents: read with: @@ -399,7 +392,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/conda-cpp-build.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/conda-cpp-build.yaml@release/26.08 with: build_type: pull-request node_type: cpu16 @@ -413,7 +406,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/conda-cpp-tests.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/conda-cpp-tests.yaml@release/26.08 if: fromJSON(needs.changed-files.outputs.changed_file_groups).test_cpp with: build_type: pull-request @@ -426,7 +419,7 @@ jobs: id-token: write packages: read pull-requests: read - uses: rapidsai/shared-workflows/.github/workflows/conda-cpp-post-build-checks.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/conda-cpp-post-build-checks.yaml@release/26.08 with: build_type: pull-request symbol_exclusions: (void (thrust::|cub::)) @@ -440,7 +433,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/conda-python-build.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/conda-python-build.yaml@release/26.08 with: build_type: pull-request script: ci/build_python.sh @@ -455,7 +448,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/conda-python-tests.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/conda-python-tests.yaml@release/26.08 if: fromJSON(needs.changed-files.outputs.changed_file_groups).test_python_conda with: build_type: pull-request @@ -464,7 +457,7 @@ jobs: needs: checks permissions: contents: read - uses: rapidsai/shared-workflows/.github/workflows/compute-matrix.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/compute-matrix.yaml@release/26.08 with: build_type: pull-request matrix_name: conda-cpp-build @@ -477,7 +470,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@release/26.08 strategy: fail-fast: false matrix: ${{ fromJSON(needs.rocky8-clib-standalone-build-matrix.outputs.matrix) }} @@ -485,7 +478,7 @@ jobs: build_type: pull-request arch: "${{matrix.arch}}" date: ${{ inputs.date }}_c - container_image: "rapidsai/ci-wheel:26.10-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" + container_image: "rapidsai/ci-wheel:26.08-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" node_type: "cpu16" requires_license_builder: true script: "ci/build_standalone_c.sh --build-tests" @@ -496,7 +489,7 @@ jobs: needs: [rocky8-clib-standalone-build, changed-files] permissions: contents: read - uses: rapidsai/shared-workflows/.github/workflows/compute-matrix.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/compute-matrix.yaml@release/26.08 if: fromJSON(needs.changed-files.outputs.changed_file_groups).test_cpp with: build_type: pull-request @@ -510,7 +503,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@release/26.08 strategy: fail-fast: false matrix: ${{ fromJSON(needs.rocky8-clib-tests-matrix.outputs.matrix) }} @@ -519,14 +512,14 @@ jobs: node_type: "gpu-l4-latest-1" arch: "${{matrix.arch}}" date: ${{ inputs.date }}_c - container_image: "rapidsai/ci-wheel:26.10-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" + container_image: "rapidsai/ci-wheel:26.08-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" script: "ci/test_standalone_c.sh libcuvs_c_${{ matrix.CUDA_VER }}_${{ matrix.ARCH }}.tar.gz" sha: ${{ inputs.sha }} conda-java-build-and-tests-matrix: needs: [conda-cpp-build, changed-files] permissions: contents: read - uses: rapidsai/shared-workflows/.github/workflows/compute-matrix.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/compute-matrix.yaml@release/26.08 if: fromJSON(needs.changed-files.outputs.changed_file_groups).test_java || fromJSON(needs.changed-files.outputs.changed_file_groups).test_cpp with: build_type: pull-request @@ -541,7 +534,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@release/26.08 # Artifacts are not published from these jobs, so it's safe to run for multiple CUDA versions. # If these jobs start producing artifacts, the names will have to differentiate between CUDA versions. strategy: @@ -551,7 +544,7 @@ jobs: build_type: pull-request node_type: "gpu-l4-latest-1" arch: "amd64" - container_image: "rapidsai/ci-conda:26.10-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" + container_image: "rapidsai/ci-conda:26.08-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" script: "ci/test_java.sh" artifact-name: "cuvs-java-cuda${{ matrix.CUDA_VER }}" file_to_upload: "java/cuvs-java/target/" @@ -559,7 +552,7 @@ jobs: needs: [conda-cpp-build, changed-files] permissions: contents: read - uses: rapidsai/shared-workflows/.github/workflows/compute-matrix.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/compute-matrix.yaml@release/26.08 if: fromJSON(needs.changed-files.outputs.changed_file_groups).test_rust || fromJSON(needs.changed-files.outputs.changed_file_groups).test_cpp with: build_type: pull-request @@ -574,7 +567,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@release/26.08 # Artifacts are not published from these jobs, so it's safe to run for multiple CUDA versions. # If these jobs start producing artifacts, the names will have to differentiate between CUDA versions. strategy: @@ -584,13 +577,13 @@ jobs: build_type: pull-request node_type: "gpu-l4-latest-1" arch: "amd64" - container_image: "rapidsai/ci-conda:26.10-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" + container_image: "rapidsai/ci-conda:26.08-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" script: "ci/build_rust.sh" go-build-matrix: needs: [conda-cpp-build, changed-files] permissions: contents: read - uses: rapidsai/shared-workflows/.github/workflows/compute-matrix.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/compute-matrix.yaml@release/26.08 if: fromJSON(needs.changed-files.outputs.changed_file_groups).test_go || fromJSON(needs.changed-files.outputs.changed_file_groups).test_cpp with: build_type: pull-request @@ -605,7 +598,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@release/26.08 # Artifacts are not published from these jobs, so it's safe to run for multiple CUDA versions. # If these jobs start producing artifacts, the names will have to differentiate between CUDA versions. strategy: @@ -615,7 +608,7 @@ jobs: build_type: pull-request node_type: "gpu-l4-latest-1" arch: "amd64" - container_image: "rapidsai/ci-conda:26.10-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" + container_image: "rapidsai/ci-conda:26.08-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" script: "ci/build_go.sh" docs-build: permissions: @@ -625,12 +618,12 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@release/26.08 with: build_type: pull-request node_type: "gpu-l4-latest-1" arch: "amd64" - container_image: "rapidsai/ci-conda:26.10-latest" + container_image: "rapidsai/ci-conda:26.08-latest" script: "ci/build_docs.sh" wheel-build-libcuvs: needs: checks @@ -641,7 +634,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/wheels-build.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/wheels-build.yaml@release/26.08 with: build_type: pull-request node_type: cpu16 @@ -659,7 +652,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/wheels-build.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/wheels-build.yaml@release/26.08 with: build_type: pull-request node_type: cpu8 @@ -677,7 +670,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/wheels-test.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/wheels-test.yaml@release/26.08 if: fromJSON(needs.changed-files.outputs.changed_file_groups).test_python_wheels with: build_type: pull-request @@ -691,7 +684,7 @@ jobs: pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] needs: telemetry-setup - uses: rapidsai/shared-workflows/.github/workflows/build-in-devcontainer.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/build-in-devcontainer.yaml@release/26.08 with: arch: '["amd64", "arm64"]' cuda: '["13.3"]' diff --git a/.github/workflows/publish-rust.yaml b/.github/workflows/publish-rust.yaml index 97ace8bf73..8b21ff7a13 100644 --- a/.github/workflows/publish-rust.yaml +++ b/.github/workflows/publish-rust.yaml @@ -19,7 +19,7 @@ jobs: cuda_version: - '13.3.0' container: - image: "rapidsai/ci-conda:26.10-cuda${{ matrix.cuda_version }}-ubuntu24.04-py3.13" + image: "rapidsai/ci-conda:26.08-cuda${{ matrix.cuda_version }}-ubuntu24.04-py3.13" steps: - uses: actions/checkout@34e114876b0b11c390a56381ad16ebd13914f8d5 # v4.3.1 with: diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml index 26739e2deb..9e881e416c 100644 --- a/.github/workflows/test.yaml +++ b/.github/workflows/test.yaml @@ -32,7 +32,7 @@ jobs: id-token: write packages: read pull-requests: read - uses: rapidsai/shared-workflows/.github/workflows/conda-cpp-post-build-checks.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/conda-cpp-post-build-checks.yaml@release/26.08 with: build_type: ${{ inputs.build_type }} branch: ${{ inputs.branch }} @@ -48,7 +48,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/conda-cpp-tests.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/conda-cpp-tests.yaml@release/26.08 with: build_type: ${{ inputs.build_type }} branch: ${{ inputs.branch }} @@ -63,23 +63,14 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/conda-python-tests.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/conda-python-tests.yaml@release/26.08 with: build_type: ${{ inputs.build_type }} branch: ${{ inputs.branch }} date: ${{ inputs.date }} script: ci/test_python.sh sha: ${{ inputs.sha }} - conda-java-build-and-tests-matrix: - permissions: - contents: read - uses: rapidsai/shared-workflows/.github/workflows/compute-matrix.yaml@main - with: - build_type: ${{ inputs.build_type }} - matrix_name: conda-cpp-build - matrix_filter: map(select(.ARCH == "amd64")) - conda-java-build-and-tests: - needs: conda-java-build-and-tests-matrix + conda-java-tests: permissions: actions: read contents: read @@ -87,12 +78,15 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@release/26.08 # Artifacts are not published from these jobs, so it's safe to run for multiple CUDA versions. # If these jobs start producing artifacts, the names will have to differentiate between CUDA versions. strategy: fail-fast: false - matrix: ${{ fromJSON(needs.conda-java-build-and-tests-matrix.outputs.matrix) }} + matrix: + cuda_version: + - '12.9.1' + - '13.3.0' with: build_type: ${{ inputs.build_type }} branch: ${{ inputs.branch }} @@ -100,10 +94,8 @@ jobs: sha: ${{ inputs.sha }} node_type: "gpu-l4-latest-1" arch: "amd64" - container_image: "rapidsai/ci-conda:26.10-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" + container_image: "rapidsai/ci-conda:26.08-cuda${{ matrix.cuda_version }}-ubuntu24.04-py3.13" script: "ci/test_java.sh" - artifact-name: "cuvs-java-cuda${{ matrix.CUDA_VER }}" - file_to_upload: "java/cuvs-java/target/" wheel-tests-cuvs: permissions: actions: read @@ -112,7 +104,7 @@ jobs: packages: read pull-requests: read secrets: inherit # zizmor: ignore[secrets-inherit] - uses: rapidsai/shared-workflows/.github/workflows/wheels-test.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/wheels-test.yaml@release/26.08 with: build_type: ${{ inputs.build_type }} branch: ${{ inputs.branch }} diff --git a/.github/workflows/trigger-breaking-change-alert.yaml b/.github/workflows/trigger-breaking-change-alert.yaml index 5a92e2c761..bb1f333bb7 100644 --- a/.github/workflows/trigger-breaking-change-alert.yaml +++ b/.github/workflows/trigger-breaking-change-alert.yaml @@ -18,7 +18,7 @@ jobs: if: contains(github.event.pull_request.labels.*.name, 'breaking') permissions: contents: read - uses: rapidsai/shared-workflows/.github/workflows/breaking-change-alert.yaml@main + uses: rapidsai/shared-workflows/.github/workflows/breaking-change-alert.yaml@release/26.08 secrets: slack-webhook-url: ${{ secrets.NV_SLACK_BREAKING_CHANGE_NOTIFIER_APP }} with: diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 5b7a192fcf..aa65eec100 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -2,216 +2,205 @@ # SPDX-License-Identifier: Apache-2.0 repos: - - repo: https://github.com/pre-commit/pre-commit-hooks - rev: v6.0.0 - hooks: - - id: check-json - - id: trailing-whitespace - - id: end-of-file-fixer - - id: check-symlinks - - id: check-xml - - repo: https://github.com/PyCQA/isort - rev: 5.12.0 - hooks: - - id: isort - # Use the config file specific to each subproject so that each - # project can specify its own first/third-party packages. - args: ["--config-root=python/", "--resolve-all-configs"] - files: python/.* - types: [cython] - - repo: https://github.com/astral-sh/ruff-pre-commit - rev: v0.14.3 - hooks: - - id: ruff-check - args: [--fix] - - id: ruff-format - - repo: https://github.com/pre-commit/mirrors-mypy - rev: 'v1.19.1' - hooks: - - id: mypy - additional_dependencies: [types-cachetools] - args: ["--config-file=pyproject.toml", - "python/cuvs/cuvs", - "cpp/cmake/modules", - "cpp/tests/python", - "ci/checks"] - pass_filenames: false - - repo: https://github.com/PyCQA/pydocstyle - rev: 6.1.1 - hooks: - - id: pydocstyle - # https://github.com/PyCQA/pydocstyle/issues/603 - additional_dependencies: [toml] - args: ["--config=pyproject.toml"] - - repo: https://github.com/pre-commit/mirrors-clang-format - rev: v20.1.8 - hooks: - - id: clang-format - types_or: [c, c++, cuda] - args: ["-fallback-style=none", "-style=file", "-i"] - - repo: local - hooks: - - id: no-deprecationwarning - name: no-deprecationwarning - description: 'Enforce that DeprecationWarning is not introduced (use FutureWarning instead)' - entry: '(category=|\s)DeprecationWarning[,)]' - language: pygrep - types_or: [python, cython] - - id: cmake-format - name: cmake-format - entry: ./cpp/scripts/run-cmake-format.sh cmake-format - language: python - types: [cmake] - exclude: .*/thirdparty/.*|.*FindAVX.cmake.* - # Note that pre-commit autoupdate does not update the versions - # of dependencies, so we'll have to update this manually. - additional_dependencies: - - cmakelang==0.6.13 - verbose: true - require_serial: true - - id: cmake-lint - name: cmake-lint - entry: ./cpp/scripts/run-cmake-format.sh cmake-lint - language: python - types: [cmake] - # Note that pre-commit autoupdate does not update the versions - # of dependencies, so we'll have to update this manually. - additional_dependencies: - - cmakelang==0.6.13 - verbose: true - require_serial: true - exclude: .*/thirdparty/.* - - id: include-check - name: include-check - entry: python ./cpp/scripts/include_checker.py cpp/bench cpp/include cpp/tests - pass_filenames: false - language: python - additional_dependencies: [gitpython] - - id: cargo-fmt - name: cargo-fmt - entry: cargo fmt --manifest-path rust/Cargo.toml --all - pass_filenames: false - files: rust/.* - language: rust - - id: clang-format-with-cmake-placeholders - name: clang-format-with-cmake-placeholders - entry: python3 ci/checks/clang_format_with_cmake_placeholders.py - language: python - additional_dependencies: [clang_format==20.1.8] - files: | - (?x) - [.](cpp|cu|hpp|cuh)[.]in$ - # clang-format struggles with one of the placeholders in register_fatbin.cpp.in, so exclude it - exclude: | - (?x) - ^cpp/cmake/modules/register_fatbin[.]cpp[.]in$ - - repo: https://github.com/codespell-project/codespell - rev: v2.4.1 - hooks: - - id: codespell - additional_dependencies: [tomli] - args: ["--toml", "pyproject.toml"] - exclude: | - (?x) - ^CHANGELOG[.]md$| - ^cpp/cmake/patches/cutlass/build-export[.]patch$| - ^rust/cuvs-sys/src/bindings[.]rs$ - - repo: https://github.com/rapidsai/pre-commit-hooks - rev: v1.6.0 - hooks: - - id: verify-copyright - name: verify-copyright-cuvs - args: [--fix, --spdx] - files: | - (?x) - [.](cmake|c|cpp([.]in)?|cu([.]in)?|cuh([.]in)?|h|hpp([.]in)?|sh|pxd|py|pyx|rs|java)$| - CMakeLists[.]txt$| - CMakeLists_standalone[.]txt$| - meta[.]yaml$| - pyproject[.]toml$| - ^python/cuvs_bench/cuvs_bench/split_groundtruth/split_groundtruth[.]pl$| - Dockerfile$| - pom[.]xml$| - ^java/cuvs-java/src/assembly/native-with-deps[.]xml$| - ^java/docker-build/build-in-docker$| - ^java/docker-build/run-in-docker$| - ^[.]flake8$| - recipe[.]yaml$| - ^[.]pre-commit-config[.]yaml$ - exclude: | - (?x)^( - docs/source/sphinxext/github_link[.]py$| - cpp/cmake/modules/FindAVX[.]cmake$| - cpp/src/neighbors/detail/faiss_distance_utils[.]h$| - cpp/src/distance/detail/fused_distance_nn/gemm[.]h$| - cpp/src/distance/detail/fused_distance_nn/epilogue[.]cuh$| - cpp/src/distance/detail/fused_distance_nn/predicated_tile_iterator_normvec_smem[.]h$| - cpp/src/distance/detail/fused_distance_nn/persistent_gemm[.]h$| - cpp/src/distance/detail/fused_distance_nn/epilogue_elementwise[.]cuh$| - cpp/src/distance/detail/fused_distance_nn/predicated_tile_iterator_reduced_vec[.]h$| - cpp/src/distance/detail/fused_distance_nn/custom_epilogue_with_broadcast[.]h$| - rust/cuvs-sys/src/bindings[.]rs$ - ) - - id: verify-copyright - name: verify-copyright-scikit-learn - args: [--fix, --spdx, "--spdx-license-identifier=Apache-2.0 AND BSD-3-Clause"] - files: | - (?x)^( - docs/source/sphinxext/github_link[.]py$ - ) - - id: verify-copyright - name: verify-copyright-cutlass - args: [--fix, --spdx, "--spdx-license-identifier=Apache-2.0 AND BSD-3-Clause"] - files: | - (?x)^( - cpp/src/distance/detail/fused_distance_nn/gemm[.]h$| - cpp/src/distance/detail/fused_distance_nn/epilogue[.]cuh$| - cpp/src/distance/detail/fused_distance_nn/predicated_tile_iterator_normvec_smem[.]h$| - cpp/src/distance/detail/fused_distance_nn/persistent_gemm[.]h$| - cpp/src/distance/detail/fused_distance_nn/epilogue_elementwise[.]cuh$| - cpp/src/distance/detail/fused_distance_nn/predicated_tile_iterator_reduced_vec[.]h$| - cpp/src/distance/detail/fused_distance_nn/custom_epilogue_with_broadcast[.]h$ - ) - - id: verify-copyright - name: verify-copyright-faiss - args: [--fix, --spdx, "--spdx-license-identifier=Apache-2.0 AND MIT"] - files: | - (?x)^( - cpp/src/neighbors/detail/faiss_distance_utils[.]h$ - ) - - id: verify-alpha-spec - - id: verify-codeowners - args: [--fix, --org=NVIDIA, --project-prefix=cuvs] - - id: verify-pyproject-license - # ignore the top-level pyproject.toml, which doesn't - # have or need a [project] table - exclude: | - (?x) - ^pyproject[.]toml$ - - repo: https://github.com/rapidsai/dependency-file-generator - rev: v1.20.0 - hooks: - - id: rapids-dependency-file-generator - args: ["--clean", "--warn-all", "--strict"] - - repo: https://github.com/shellcheck-py/shellcheck-py - rev: v0.10.0.1 - hooks: - - id: shellcheck - - repo: https://github.com/adrienverge/yamllint - rev: v1.38.0 - hooks: - - id: yamllint - additional_dependencies: [pyyaml] - exclude: | - (?x)^( - conda/environments/.*| - cpp/[.]clang-format$| - cpp/[.]clang-tidy$ - ) - - repo: https://github.com/zizmorcore/zizmor-pre-commit - rev: v1.24.1 - hooks: - - id: zizmor + - repo: https://github.com/pre-commit/pre-commit-hooks + rev: v6.0.0 + hooks: + - id: check-json + - id: trailing-whitespace + - id: end-of-file-fixer + - id: check-symlinks + - id: check-xml + - repo: https://github.com/PyCQA/isort + rev: 5.12.0 + hooks: + - id: isort + # Use the config file specific to each subproject so that each + # project can specify its own first/third-party packages. + args: ["--config-root=python/", "--resolve-all-configs"] + files: python/.* + types: [cython] + - repo: https://github.com/astral-sh/ruff-pre-commit + rev: v0.14.3 + hooks: + - id: ruff-check + args: [--fix] + - id: ruff-format + - repo: https://github.com/pre-commit/mirrors-mypy + rev: 'v1.19.1' + hooks: + - id: mypy + additional_dependencies: [types-cachetools] + args: ["--config-file=pyproject.toml", + "python/cuvs/cuvs", + "cpp/cmake/modules", + "cpp/tests/python", + "ci/checks"] + pass_filenames: false + - repo: https://github.com/PyCQA/pydocstyle + rev: 6.1.1 + hooks: + - id: pydocstyle + # https://github.com/PyCQA/pydocstyle/issues/603 + additional_dependencies: [toml] + args: ["--config=pyproject.toml"] + - repo: https://github.com/pre-commit/mirrors-clang-format + rev: v20.1.8 + hooks: + - id: clang-format + types_or: [c, c++, cuda] + args: ["-fallback-style=none", "-style=file", "-i"] + - repo: local + hooks: + - id: no-deprecationwarning + name: no-deprecationwarning + description: 'Enforce that DeprecationWarning is not introduced (use FutureWarning instead)' + entry: '(category=|\s)DeprecationWarning[,)]' + language: pygrep + types_or: [python, cython] + - id: cmake-format + name: cmake-format + entry: ./cpp/scripts/run-cmake-format.sh cmake-format + language: python + types: [cmake] + exclude: .*/thirdparty/.*|.*FindAVX.cmake.* + # Note that pre-commit autoupdate does not update the versions + # of dependencies, so we'll have to update this manually. + additional_dependencies: + - cmakelang==0.6.13 + verbose: true + require_serial: true + - id: cmake-lint + name: cmake-lint + entry: ./cpp/scripts/run-cmake-format.sh cmake-lint + language: python + types: [cmake] + # Note that pre-commit autoupdate does not update the versions + # of dependencies, so we'll have to update this manually. + additional_dependencies: + - cmakelang==0.6.13 + verbose: true + require_serial: true + exclude: .*/thirdparty/.* + - id: include-check + name: include-check + entry: python ./cpp/scripts/include_checker.py cpp/bench cpp/include cpp/tests + pass_filenames: false + language: python + additional_dependencies: [gitpython] + - id: cargo-fmt + name: cargo-fmt + entry: cargo fmt --manifest-path rust/Cargo.toml --all + pass_filenames: false + files: rust/.* + language: rust + - id: clang-format-with-cmake-placeholders + name: clang-format-with-cmake-placeholders + entry: python3 ci/checks/clang_format_with_cmake_placeholders.py + language: python + additional_dependencies: [clang_format==20.1.8] + files: | + (?x) + [.](cpp|cu|hpp|cuh)[.]in$ + # clang-format struggles with one of the placeholders in register_fatbin.cpp.in, so exclude it + exclude: | + (?x) + ^cpp/cmake/modules/register_fatbin[.]cpp[.]in$ + - repo: https://github.com/codespell-project/codespell + rev: v2.4.1 + hooks: + - id: codespell + additional_dependencies: [tomli] + args: ["--toml", "pyproject.toml"] + exclude: | + (?x) + ^CHANGELOG[.]md$| + ^cpp/cmake/patches/cutlass/build-export[.]patch$| + ^rust/cuvs-sys/src/bindings[.]rs$ + - repo: https://github.com/rapidsai/pre-commit-hooks + rev: v1.6.0 + hooks: + - id: verify-copyright + name: verify-copyright-cuvs + args: [--fix, --spdx] + files: | + (?x) + [.](cmake|c|cpp([.]in)?|cu([.]in)?|cuh([.]in)?|h|hpp([.]in)?|sh|pxd|py|pyx|rs|java)$| + CMakeLists[.]txt$| + CMakeLists_standalone[.]txt$| + meta[.]yaml$| + pyproject[.]toml$| + ^python/cuvs_bench/cuvs_bench/split_groundtruth/split_groundtruth[.]pl$| + Dockerfile$| + pom[.]xml$| + ^java/cuvs-java/src/assembly/native-with-deps[.]xml$| + ^java/docker-build/build-in-docker$| + ^java/docker-build/run-in-docker$| + ^[.]flake8$| + recipe[.]yaml$| + ^[.]pre-commit-config[.]yaml$ + exclude: | + (?x)^( + docs/source/sphinxext/github_link[.]py$| + cpp/cmake/modules/FindAVX[.]cmake$| + cpp/src/neighbors/detail/faiss_distance_utils[.]h$| + cpp/src/distance/detail/fused_distance_nn/gemm[.]h$| + cpp/src/distance/detail/fused_distance_nn/epilogue[.]cuh$| + cpp/src/distance/detail/fused_distance_nn/predicated_tile_iterator_normvec_smem[.]h$| + cpp/src/distance/detail/fused_distance_nn/persistent_gemm[.]h$| + cpp/src/distance/detail/fused_distance_nn/epilogue_elementwise[.]cuh$| + cpp/src/distance/detail/fused_distance_nn/predicated_tile_iterator_reduced_vec[.]h$| + cpp/src/distance/detail/fused_distance_nn/custom_epilogue_with_broadcast[.]h$| + rust/cuvs-sys/src/bindings[.]rs$ + ) + - id: verify-copyright + name: verify-copyright-scikit-learn + args: [--fix, --spdx, "--spdx-license-identifier=Apache-2.0 AND BSD-3-Clause"] + files: | + (?x)^( + docs/source/sphinxext/github_link[.]py$ + ) + - id: verify-copyright + name: verify-copyright-cutlass + args: [--fix, --spdx, "--spdx-license-identifier=Apache-2.0 AND BSD-3-Clause"] + files: | + (?x)^( + cpp/src/distance/detail/fused_distance_nn/gemm[.]h$| + cpp/src/distance/detail/fused_distance_nn/epilogue[.]cuh$| + cpp/src/distance/detail/fused_distance_nn/predicated_tile_iterator_normvec_smem[.]h$| + cpp/src/distance/detail/fused_distance_nn/persistent_gemm[.]h$| + cpp/src/distance/detail/fused_distance_nn/epilogue_elementwise[.]cuh$| + cpp/src/distance/detail/fused_distance_nn/predicated_tile_iterator_reduced_vec[.]h$| + cpp/src/distance/detail/fused_distance_nn/custom_epilogue_with_broadcast[.]h$ + ) + - id: verify-copyright + name: verify-copyright-faiss + args: [--fix, --spdx, "--spdx-license-identifier=Apache-2.0 AND MIT"] + files: | + (?x)^( + cpp/src/neighbors/detail/faiss_distance_utils[.]h$ + ) + - id: verify-alpha-spec + - id: verify-codeowners + args: [--fix, --org=NVIDIA, --project-prefix=cuvs] + - id: verify-pyproject-license + # ignore the top-level pyproject.toml, which doesn't + # have or need a [project] table + exclude: | + (?x) + ^pyproject[.]toml$ + - repo: https://github.com/rapidsai/dependency-file-generator + rev: v1.20.0 + hooks: + - id: rapids-dependency-file-generator + args: ["--clean", "--warn-all", "--strict"] + - repo: https://github.com/shellcheck-py/shellcheck-py + rev: v0.10.0.1 + hooks: + - id: shellcheck + - repo: https://github.com/zizmorcore/zizmor-pre-commit + rev: v1.24.1 + hooks: + - id: zizmor default_language_version: - python: python3 + python: python3 diff --git a/.yamllint.yaml b/.yamllint.yaml deleted file mode 100644 index 5e30d95bce..0000000000 --- a/.yamllint.yaml +++ /dev/null @@ -1,37 +0,0 @@ -# SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved. -# SPDX-License-Identifier: Apache-2.0 -extends: default - -rules: - anchors: - forbid-undeclared-aliases: true - forbid-duplicated-anchors: true - forbid-unused-anchors: true - braces: - forbid: false - min-spaces-inside: 0 - # allow 1 space for jinja templating in conda recipes - max-spaces-inside: 1 - min-spaces-inside-empty: -1 - max-spaces-inside-empty: -1 - brackets: enable - colons: - max-spaces-before: 0 - max-spaces-after: 1 - commas: - max-spaces-before: 0 - min-spaces-after: 1 - max-spaces-after: 1 - comments: disable - comments-indentation: disable - document-end: disable - document-start: disable - key-duplicates: - forbid-duplicated-merge-keys: true - line-length: disable - truthy: - allowed-values: ['false', 'true'] - # having problematic value in keys is rare... and also - # GitHub Actions' choice of 'on:' triggers this check - # ref: https://github.com/adrienverge/yamllint/issues/430 - check-keys: false diff --git a/RAPIDS_BRANCH b/RAPIDS_BRANCH index ba2906d066..8ce6b73837 100644 --- a/RAPIDS_BRANCH +++ b/RAPIDS_BRANCH @@ -1 +1 @@ -main +release/26.08 diff --git a/VERSION b/VERSION index 6549ba6527..4e6864b4ca 100644 --- a/VERSION +++ b/VERSION @@ -1 +1 @@ -26.10.00 +26.08.00 diff --git a/cmake/RAPIDS.cmake b/cmake/RAPIDS.cmake index 03fb805894..7ddec419bc 100644 --- a/cmake/RAPIDS.cmake +++ b/cmake/RAPIDS.cmake @@ -67,11 +67,9 @@ if(rapids-cmake-fetch-via-git) GIT_REPOSITORY "${rapids-cmake-url}" GIT_TAG "${rapids-cmake-value-to-clone}" ) - message(STATUS "Fetching rapids-cmake from ${rapids-cmake-url}@${rapids-cmake-value-to-clone}") else() string(APPEND rapids-cmake-url "${rapids-cmake-value-to-clone}") FetchContent_Declare(rapids-cmake URL "${rapids-cmake-url}") - message(STATUS "Fetching rapids-cmake from ${rapids-cmake-url}") endif() FetchContent_GetProperties(rapids-cmake) if(rapids-cmake_POPULATED) diff --git a/conda/environments/all_cuda-129_arch-aarch64.yaml b/conda/environments/all_cuda-129_arch-aarch64.yaml index 955fbf4b3b..5ac1fc363a 100644 --- a/conda/environments/all_cuda-129_arch-aarch64.yaml +++ b/conda/environments/all_cuda-129_arch-aarch64.yaml @@ -22,13 +22,14 @@ dependencies: - dlpack>=0.8,<1.0 - gcc_linux-aarch64=14.* - go +- libclang==20.1.8 - libcublas-dev - libcurand-dev - libcusolver-dev - libcusparse-dev - libnvjitlink-dev - libopenblas<=0.3.30 -- librmm==26.10.*,>=0.0.0a0 +- librmm==26.8.*,>=0.0.0a0 - make - nccl>=2.19 - ninja @@ -36,7 +37,7 @@ dependencies: - numpy>=2.0,<3.0 - openblas - pre-commit -- pylibraft==26.10.*,>=0.0.0a0 +- pylibraft==26.8.*,>=0.0.0a0 - pytest - pytest-cov - rapids-build-backend>=0.4.0,<0.5.0 diff --git a/conda/environments/all_cuda-129_arch-x86_64.yaml b/conda/environments/all_cuda-129_arch-x86_64.yaml index e0b706e97e..55ae0a47de 100644 --- a/conda/environments/all_cuda-129_arch-x86_64.yaml +++ b/conda/environments/all_cuda-129_arch-x86_64.yaml @@ -22,12 +22,13 @@ dependencies: - dlpack>=0.8,<1.0 - gcc_linux-64=14.* - go +- libclang==20.1.8 - libcublas-dev - libcurand-dev - libcusolver-dev - libcusparse-dev - libnvjitlink-dev -- librmm==26.10.*,>=0.0.0a0 +- librmm==26.8.*,>=0.0.0a0 - make - nccl>=2.19 - ninja @@ -35,7 +36,7 @@ dependencies: - numpy>=2.0,<3.0 - openblas - pre-commit -- pylibraft==26.10.*,>=0.0.0a0 +- pylibraft==26.8.*,>=0.0.0a0 - pytest - pytest-cov - rapids-build-backend>=0.4.0,<0.5.0 diff --git a/conda/environments/all_cuda-133_arch-aarch64.yaml b/conda/environments/all_cuda-133_arch-aarch64.yaml index e6972b30dc..b22f353fce 100644 --- a/conda/environments/all_cuda-133_arch-aarch64.yaml +++ b/conda/environments/all_cuda-133_arch-aarch64.yaml @@ -22,13 +22,14 @@ dependencies: - dlpack>=0.8,<1.0 - gcc_linux-aarch64=14.* - go +- libclang==20.1.8 - libcublas-dev - libcurand-dev - libcusolver-dev - libcusparse-dev - libnvjitlink-dev - libopenblas<=0.3.30 -- librmm==26.10.*,>=0.0.0a0 +- librmm==26.8.*,>=0.0.0a0 - make - nccl>=2.19 - ninja @@ -36,7 +37,7 @@ dependencies: - numpy>=2.0,<3.0 - openblas - pre-commit -- pylibraft==26.10.*,>=0.0.0a0 +- pylibraft==26.8.*,>=0.0.0a0 - pytest - pytest-cov - rapids-build-backend>=0.4.0,<0.5.0 diff --git a/conda/environments/all_cuda-133_arch-x86_64.yaml b/conda/environments/all_cuda-133_arch-x86_64.yaml index bab866e313..a17ac5e358 100644 --- a/conda/environments/all_cuda-133_arch-x86_64.yaml +++ b/conda/environments/all_cuda-133_arch-x86_64.yaml @@ -22,12 +22,13 @@ dependencies: - dlpack>=0.8,<1.0 - gcc_linux-64=14.* - go +- libclang==20.1.8 - libcublas-dev - libcurand-dev - libcusolver-dev - libcusparse-dev - libnvjitlink-dev -- librmm==26.10.*,>=0.0.0a0 +- librmm==26.8.*,>=0.0.0a0 - make - nccl>=2.19 - ninja @@ -35,7 +36,7 @@ dependencies: - numpy>=2.0,<3.0 - openblas - pre-commit -- pylibraft==26.10.*,>=0.0.0a0 +- pylibraft==26.8.*,>=0.0.0a0 - pytest - pytest-cov - rapids-build-backend>=0.4.0,<0.5.0 diff --git a/conda/environments/bench_ann_cuda-129_arch-aarch64.yaml b/conda/environments/bench_ann_cuda-129_arch-aarch64.yaml index b844e1d93d..d495ad32ed 100644 --- a/conda/environments/bench_ann_cuda-129_arch-aarch64.yaml +++ b/conda/environments/bench_ann_cuda-129_arch-aarch64.yaml @@ -17,20 +17,21 @@ dependencies: - cuda-profiler-api - cuda-version=12.9 - cupy>=14.0.1,!=14.1.0 -- cuvs==26.10.*,>=0.0.0a0 +- cuvs==26.8.*,>=0.0.0a0 - cxx-compiler - cython>=3.2.2 - dlpack>=0.8,<1.0 - gcc_linux-aarch64=14.* - glog>=0.6.0 - h5py>=3.8.0 +- libclang==20.1.8 - libcublas-dev - libcurand-dev - libcusolver-dev - libcusparse-dev -- libcuvs==26.10.*,>=0.0.0a0 +- libcuvs==26.8.*,>=0.0.0a0 - libnvjitlink-dev -- librmm==26.10.*,>=0.0.0a0 +- librmm==26.8.*,>=0.0.0a0 - matplotlib-base>=3.9 - nccl>=2.19 - ninja @@ -38,7 +39,7 @@ dependencies: - openblas - opensearch-py>=2.4.0 - pandas -- pylibraft==26.10.*,>=0.0.0a0 +- pylibraft==26.8.*,>=0.0.0a0 - pyyaml - rapids-build-backend>=0.4.0,<0.5.0 - requests diff --git a/conda/environments/bench_ann_cuda-129_arch-x86_64.yaml b/conda/environments/bench_ann_cuda-129_arch-x86_64.yaml index 2340c9d7a0..651a372dc1 100644 --- a/conda/environments/bench_ann_cuda-129_arch-x86_64.yaml +++ b/conda/environments/bench_ann_cuda-129_arch-x86_64.yaml @@ -17,7 +17,7 @@ dependencies: - cuda-profiler-api - cuda-version=12.9 - cupy>=14.0.1,!=14.1.0 -- cuvs==26.10.*,>=0.0.0a0 +- cuvs==26.8.*,>=0.0.0a0 - cxx-compiler - cython>=3.2.2 - dlpack>=0.8,<1.0 @@ -26,13 +26,14 @@ dependencies: - h5py>=3.8.0 - libaio - libboost-devel=1.87 +- libclang==20.1.8 - libcublas-dev - libcurand-dev - libcusolver-dev - libcusparse-dev -- libcuvs==26.10.*,>=0.0.0a0 +- libcuvs==26.8.*,>=0.0.0a0 - libnvjitlink-dev -- librmm==26.10.*,>=0.0.0a0 +- librmm==26.8.*,>=0.0.0a0 - matplotlib-base>=3.9 - mkl-devel=2023 - nccl>=2.19 @@ -41,7 +42,7 @@ dependencies: - openblas - opensearch-py>=2.4.0 - pandas -- pylibraft==26.10.*,>=0.0.0a0 +- pylibraft==26.8.*,>=0.0.0a0 - pyyaml - rapids-build-backend>=0.4.0,<0.5.0 - requests diff --git a/conda/environments/bench_ann_cuda-133_arch-aarch64.yaml b/conda/environments/bench_ann_cuda-133_arch-aarch64.yaml index 321a892555..05b0429254 100644 --- a/conda/environments/bench_ann_cuda-133_arch-aarch64.yaml +++ b/conda/environments/bench_ann_cuda-133_arch-aarch64.yaml @@ -17,20 +17,21 @@ dependencies: - cuda-profiler-api - cuda-version=13.3 - cupy>=14.0.1,!=14.1.0 -- cuvs==26.10.*,>=0.0.0a0 +- cuvs==26.8.*,>=0.0.0a0 - cxx-compiler - cython>=3.2.2 - dlpack>=0.8,<1.0 - gcc_linux-aarch64=14.* - glog>=0.6.0 - h5py>=3.8.0 +- libclang==20.1.8 - libcublas-dev - libcurand-dev - libcusolver-dev - libcusparse-dev -- libcuvs==26.10.*,>=0.0.0a0 +- libcuvs==26.8.*,>=0.0.0a0 - libnvjitlink-dev -- librmm==26.10.*,>=0.0.0a0 +- librmm==26.8.*,>=0.0.0a0 - matplotlib-base>=3.9 - nccl>=2.19 - ninja @@ -38,7 +39,7 @@ dependencies: - openblas - opensearch-py>=2.4.0 - pandas -- pylibraft==26.10.*,>=0.0.0a0 +- pylibraft==26.8.*,>=0.0.0a0 - pyyaml - rapids-build-backend>=0.4.0,<0.5.0 - requests diff --git a/conda/environments/bench_ann_cuda-133_arch-x86_64.yaml b/conda/environments/bench_ann_cuda-133_arch-x86_64.yaml index 179b4a4a2f..398c612088 100644 --- a/conda/environments/bench_ann_cuda-133_arch-x86_64.yaml +++ b/conda/environments/bench_ann_cuda-133_arch-x86_64.yaml @@ -17,7 +17,7 @@ dependencies: - cuda-profiler-api - cuda-version=13.3 - cupy>=14.0.1,!=14.1.0 -- cuvs==26.10.*,>=0.0.0a0 +- cuvs==26.8.*,>=0.0.0a0 - cxx-compiler - cython>=3.2.2 - dlpack>=0.8,<1.0 @@ -26,13 +26,14 @@ dependencies: - h5py>=3.8.0 - libaio - libboost-devel=1.87 +- libclang==20.1.8 - libcublas-dev - libcurand-dev - libcusolver-dev - libcusparse-dev -- libcuvs==26.10.*,>=0.0.0a0 +- libcuvs==26.8.*,>=0.0.0a0 - libnvjitlink-dev -- librmm==26.10.*,>=0.0.0a0 +- librmm==26.8.*,>=0.0.0a0 - matplotlib-base>=3.9 - mkl-devel=2023 - nccl>=2.19 @@ -41,7 +42,7 @@ dependencies: - openblas - opensearch-py>=2.4.0 - pandas -- pylibraft==26.10.*,>=0.0.0a0 +- pylibraft==26.8.*,>=0.0.0a0 - pyyaml - rapids-build-backend>=0.4.0,<0.5.0 - requests diff --git a/conda/environments/go_cuda-129_arch-aarch64.yaml b/conda/environments/go_cuda-129_arch-aarch64.yaml index e95660174c..eb6ae9f5f7 100644 --- a/conda/environments/go_cuda-129_arch-aarch64.yaml +++ b/conda/environments/go_cuda-129_arch-aarch64.yaml @@ -20,13 +20,14 @@ dependencies: - dlpack>=0.8,<1.0 - gcc_linux-aarch64=14.* - go +- libclang==20.1.8 - libcublas-dev - libcurand-dev - libcusolver-dev - libcusparse-dev -- libcuvs==26.10.*,>=0.0.0a0 +- libcuvs==26.8.*,>=0.0.0a0 - libnvjitlink-dev -- libraft==26.10.*,>=0.0.0a0 +- libraft==26.8.*,>=0.0.0a0 - nccl>=2.19 - ninja - sysroot_linux-aarch64==2.28 diff --git a/conda/environments/go_cuda-129_arch-x86_64.yaml b/conda/environments/go_cuda-129_arch-x86_64.yaml index ba96a1c457..e99615ab2c 100644 --- a/conda/environments/go_cuda-129_arch-x86_64.yaml +++ b/conda/environments/go_cuda-129_arch-x86_64.yaml @@ -20,13 +20,14 @@ dependencies: - dlpack>=0.8,<1.0 - gcc_linux-64=14.* - go +- libclang==20.1.8 - libcublas-dev - libcurand-dev - libcusolver-dev - libcusparse-dev -- libcuvs==26.10.*,>=0.0.0a0 +- libcuvs==26.8.*,>=0.0.0a0 - libnvjitlink-dev -- libraft==26.10.*,>=0.0.0a0 +- libraft==26.8.*,>=0.0.0a0 - nccl>=2.19 - ninja - sysroot_linux-64==2.28 diff --git a/conda/environments/go_cuda-133_arch-aarch64.yaml b/conda/environments/go_cuda-133_arch-aarch64.yaml index 8098f29222..eeb5511537 100644 --- a/conda/environments/go_cuda-133_arch-aarch64.yaml +++ b/conda/environments/go_cuda-133_arch-aarch64.yaml @@ -20,13 +20,14 @@ dependencies: - dlpack>=0.8,<1.0 - gcc_linux-aarch64=14.* - go +- libclang==20.1.8 - libcublas-dev - libcurand-dev - libcusolver-dev - libcusparse-dev -- libcuvs==26.10.*,>=0.0.0a0 +- libcuvs==26.8.*,>=0.0.0a0 - libnvjitlink-dev -- libraft==26.10.*,>=0.0.0a0 +- libraft==26.8.*,>=0.0.0a0 - nccl>=2.19 - ninja - sysroot_linux-aarch64==2.28 diff --git a/conda/environments/go_cuda-133_arch-x86_64.yaml b/conda/environments/go_cuda-133_arch-x86_64.yaml index e68b40440e..5fd2153ffa 100644 --- a/conda/environments/go_cuda-133_arch-x86_64.yaml +++ b/conda/environments/go_cuda-133_arch-x86_64.yaml @@ -20,13 +20,14 @@ dependencies: - dlpack>=0.8,<1.0 - gcc_linux-64=14.* - go +- libclang==20.1.8 - libcublas-dev - libcurand-dev - libcusolver-dev - libcusparse-dev -- libcuvs==26.10.*,>=0.0.0a0 +- libcuvs==26.8.*,>=0.0.0a0 - libnvjitlink-dev -- libraft==26.10.*,>=0.0.0a0 +- libraft==26.8.*,>=0.0.0a0 - nccl>=2.19 - ninja - sysroot_linux-64==2.28 diff --git a/conda/environments/rust_cuda-129_arch-aarch64.yaml b/conda/environments/rust_cuda-129_arch-aarch64.yaml index a859251b85..5849886871 100644 --- a/conda/environments/rust_cuda-129_arch-aarch64.yaml +++ b/conda/environments/rust_cuda-129_arch-aarch64.yaml @@ -19,9 +19,9 @@ dependencies: - libcurand-dev - libcusolver-dev - libcusparse-dev -- libcuvs==26.10.*,>=0.0.0a0 +- libcuvs==26.8.*,>=0.0.0a0 - libnvjitlink-dev -- libraft==26.10.*,>=0.0.0a0 +- libraft==26.8.*,>=0.0.0a0 - make - nccl>=2.19 - ninja diff --git a/conda/environments/rust_cuda-129_arch-x86_64.yaml b/conda/environments/rust_cuda-129_arch-x86_64.yaml index 16c61ef52a..a9e9e2142c 100644 --- a/conda/environments/rust_cuda-129_arch-x86_64.yaml +++ b/conda/environments/rust_cuda-129_arch-x86_64.yaml @@ -19,9 +19,9 @@ dependencies: - libcurand-dev - libcusolver-dev - libcusparse-dev -- libcuvs==26.10.*,>=0.0.0a0 +- libcuvs==26.8.*,>=0.0.0a0 - libnvjitlink-dev -- libraft==26.10.*,>=0.0.0a0 +- libraft==26.8.*,>=0.0.0a0 - make - nccl>=2.19 - ninja diff --git a/conda/environments/rust_cuda-133_arch-aarch64.yaml b/conda/environments/rust_cuda-133_arch-aarch64.yaml index d4d9095d6a..3daa10d2be 100644 --- a/conda/environments/rust_cuda-133_arch-aarch64.yaml +++ b/conda/environments/rust_cuda-133_arch-aarch64.yaml @@ -19,9 +19,9 @@ dependencies: - libcurand-dev - libcusolver-dev - libcusparse-dev -- libcuvs==26.10.*,>=0.0.0a0 +- libcuvs==26.8.*,>=0.0.0a0 - libnvjitlink-dev -- libraft==26.10.*,>=0.0.0a0 +- libraft==26.8.*,>=0.0.0a0 - make - nccl>=2.19 - ninja diff --git a/conda/environments/rust_cuda-133_arch-x86_64.yaml b/conda/environments/rust_cuda-133_arch-x86_64.yaml index fd92bdaafd..9e0fe2d529 100644 --- a/conda/environments/rust_cuda-133_arch-x86_64.yaml +++ b/conda/environments/rust_cuda-133_arch-x86_64.yaml @@ -19,9 +19,9 @@ dependencies: - libcurand-dev - libcusolver-dev - libcusparse-dev -- libcuvs==26.10.*,>=0.0.0a0 +- libcuvs==26.8.*,>=0.0.0a0 - libnvjitlink-dev -- libraft==26.10.*,>=0.0.0a0 +- libraft==26.8.*,>=0.0.0a0 - make - nccl>=2.19 - ninja diff --git a/conda/recipes/cuvs-bench-cpu/recipe.yaml b/conda/recipes/cuvs-bench-cpu/recipe.yaml index fc57688dd4..baf8fcfbc0 100644 --- a/conda/recipes/cuvs-bench-cpu/recipe.yaml +++ b/conda/recipes/cuvs-bench-cpu/recipe.yaml @@ -8,7 +8,7 @@ context: head_rev: '${{ git.head_rev(".")[:8] }}' linux64: ${{ linux and x86_64 }} py_abi_min: ${{ env.get("RAPIDS_PY_VERSION") }} - py_buildstring: ${{ py_abi_min | version_to_buildstring }} + py_buildstring : ${{ py_abi_min | version_to_buildstring }} py_runtime_latest: "3.14" package: diff --git a/conda/recipes/cuvs-bench/recipe.yaml b/conda/recipes/cuvs-bench/recipe.yaml index ece9235a33..a2c5f2426a 100644 --- a/conda/recipes/cuvs-bench/recipe.yaml +++ b/conda/recipes/cuvs-bench/recipe.yaml @@ -10,7 +10,7 @@ context: date_string: '${{ env.get("RAPIDS_DATE_STRING") }}' head_rev: '${{ git.head_rev(".")[:8] }}' py_abi_min: ${{ env.get("RAPIDS_PY_VERSION") }} - py_buildstring: ${{ py_abi_min | version_to_buildstring }} + py_buildstring : ${{ py_abi_min | version_to_buildstring }} py_runtime_latest: "3.14" package: diff --git a/conda/recipes/cuvs/recipe.yaml b/conda/recipes/cuvs/recipe.yaml index 4a32b9f8d2..2b2cca67cb 100644 --- a/conda/recipes/cuvs/recipe.yaml +++ b/conda/recipes/cuvs/recipe.yaml @@ -10,7 +10,7 @@ context: date_string: '${{ env.get("RAPIDS_DATE_STRING") }}' head_rev: '${{ git.head_rev(".")[:8] }}' py_abi_min: ${{ env.get("RAPIDS_PY_VERSION") }} - py_buildstring: ${{ py_abi_min | version_to_buildstring }} + py_buildstring : ${{ py_abi_min | version_to_buildstring }} py_runtime_latest: "3.14" package: diff --git a/conda/recipes/libcuvs/recipe.yaml b/conda/recipes/libcuvs/recipe.yaml index b8ebd8ebec..1041dd074a 100644 --- a/conda/recipes/libcuvs/recipe.yaml +++ b/conda/recipes/libcuvs/recipe.yaml @@ -101,7 +101,7 @@ outputs: dynamic_linking: overlinking_behavior: "error" prefix_detection: - ignore_binary_files: true + ignore_binary_files: True script: content: | cmake --install cpp/build --component cuvs_cpp_headers @@ -159,7 +159,7 @@ outputs: dynamic_linking: overlinking_behavior: "error" prefix_detection: - ignore_binary_files: true + ignore_binary_files: True script: content: | cmake --install cpp/build --component cuvs @@ -222,7 +222,7 @@ outputs: dynamic_linking: overlinking_behavior: "error" prefix_detection: - ignore_binary_files: true + ignore_binary_files: True script: content: | cmake --install cpp/build --component cuvs_static @@ -284,7 +284,7 @@ outputs: dynamic_linking: overlinking_behavior: "error" prefix_detection: - ignore_binary_files: true + ignore_binary_files: True requirements: build: - cmake ${{ cmake_version }} @@ -340,7 +340,7 @@ outputs: dynamic_linking: overlinking_behavior: "error" prefix_detection: - ignore_binary_files: true + ignore_binary_files: True script: content: | # Remove `-fdebug-prefix-map` line from CFLAGS and CXXFLAGS so the @@ -452,7 +452,7 @@ outputs: dynamic_linking: overlinking_behavior: "error" prefix_detection: - ignore_binary_files: true + ignore_binary_files: True requirements: build: - cmake ${{ cmake_version }} diff --git a/dependencies.yaml b/dependencies.yaml index 970b173328..8faea8d8ed 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -303,6 +303,7 @@ dependencies: packages: - clang==20.1.8 - clang-tools==20.1.8 + - libclang==20.1.8 # 'cuda_version' intentionally does not contain fallback entries... we want # a loud error if an unsupported 'cuda' value is passed cuda_version: @@ -617,7 +618,7 @@ dependencies: - output_types: [conda, pyproject, requirements] packages: - click - - cuvs==26.10.*,>=0.0.0a0 + - cuvs==26.8.*,>=0.0.0a0 - pandas - pyyaml - requests @@ -655,17 +656,17 @@ dependencies: common: - output_types: conda packages: - - cuvs==26.10.*,>=0.0.0a0 + - cuvs==26.8.*,>=0.0.0a0 depends_on_cuvs_bench: common: - output_types: conda packages: - - cuvs-bench==26.10.*,>=0.0.0a0 + - cuvs-bench==26.8.*,>=0.0.0a0 depends_on_libcuvs: common: - output_types: conda packages: - - &libcuvs_unsuffixed libcuvs==26.10.*,>=0.0.0a0 + - &libcuvs_unsuffixed libcuvs==26.8.*,>=0.0.0a0 - output_types: requirements packages: # pip recognizes the index as a global option for the requirements.txt file @@ -677,23 +678,23 @@ dependencies: cuda: "12.*" cuda_suffixed: "true" packages: - - libcuvs-cu12==26.10.*,>=0.0.0a0 + - libcuvs-cu12==26.8.*,>=0.0.0a0 - matrix: cuda: "13.*" cuda_suffixed: "true" packages: - - libcuvs-cu13==26.10.*,>=0.0.0a0 + - libcuvs-cu13==26.8.*,>=0.0.0a0 - {matrix: null, packages: [*libcuvs_unsuffixed]} depends_on_libcuvs_tests: common: - output_types: conda packages: - - libcuvs-tests==26.10.*,>=0.0.0a0 + - libcuvs-tests==26.8.*,>=0.0.0a0 depends_on_libraft: common: - output_types: conda packages: - - &libraft_unsuffixed libraft==26.10.*,>=0.0.0a0 + - &libraft_unsuffixed libraft==26.8.*,>=0.0.0a0 - output_types: requirements packages: # pip recognizes the index as a global option for the requirements.txt file @@ -705,18 +706,18 @@ dependencies: cuda: "12.*" cuda_suffixed: "true" packages: - - libraft-cu12==26.10.*,>=0.0.0a0 + - libraft-cu12==26.8.*,>=0.0.0a0 - matrix: cuda: "13.*" cuda_suffixed: "true" packages: - - libraft-cu13==26.10.*,>=0.0.0a0 + - libraft-cu13==26.8.*,>=0.0.0a0 - {matrix: null, packages: [*libraft_unsuffixed]} depends_on_librmm: common: - output_types: conda packages: - - &librmm_unsuffixed librmm==26.10.*,>=0.0.0a0 + - &librmm_unsuffixed librmm==26.8.*,>=0.0.0a0 - output_types: requirements packages: # pip recognizes the index as a global option for the requirements.txt file @@ -728,18 +729,18 @@ dependencies: cuda: "12.*" cuda_suffixed: "true" packages: - - librmm-cu12==26.10.*,>=0.0.0a0 + - librmm-cu12==26.8.*,>=0.0.0a0 - matrix: cuda: "13.*" cuda_suffixed: "true" packages: - - librmm-cu13==26.10.*,>=0.0.0a0 + - librmm-cu13==26.8.*,>=0.0.0a0 - {matrix: null, packages: [*librmm_unsuffixed]} depends_on_pylibraft: common: - output_types: conda packages: - - &pylibraft_unsuffixed pylibraft==26.10.*,>=0.0.0a0 + - &pylibraft_unsuffixed pylibraft==26.8.*,>=0.0.0a0 - output_types: requirements packages: # pip recognizes the index as a global option for the requirements.txt file @@ -751,12 +752,12 @@ dependencies: cuda: "12.*" cuda_suffixed: "true" packages: - - pylibraft-cu12==26.10.*,>=0.0.0a0 + - pylibraft-cu12==26.8.*,>=0.0.0a0 - matrix: cuda: "13.*" cuda_suffixed: "true" packages: - - pylibraft-cu13==26.10.*,>=0.0.0a0 + - pylibraft-cu13==26.8.*,>=0.0.0a0 - {matrix: null, packages: [*pylibraft_unsuffixed]} depends_on_nccl: common: diff --git a/examples/go/README.md b/examples/go/README.md index 9f43aaed79..d5d3ac34f0 100644 --- a/examples/go/README.md +++ b/examples/go/README.md @@ -24,7 +24,7 @@ export CC=clang 2. Install the Go module: ```bash -go get github.com/rapidsai/cuvs/go@v26.10.00 # 25.02.00 being your desired version, selected from https://github.com/rapidsai/cuvs/tags +go get github.com/rapidsai/cuvs/go@v26.08.00 # 25.02.00 being your desired version, selected from https://github.com/rapidsai/cuvs/tags ``` Then you can build your project with the usual `go build`. diff --git a/java/benchmarks/pom.xml b/java/benchmarks/pom.xml index 3a5ef75eb6..23c4825010 100644 --- a/java/benchmarks/pom.xml +++ b/java/benchmarks/pom.xml @@ -10,7 +10,7 @@ com.nvidia.cuvs benchmarks - 26.10.0 + 26.08.0 jar cuvs-java-benchmarks @@ -84,7 +84,7 @@ com.nvidia.cuvs cuvs-java - 26.10.0 + 26.08.0 jar diff --git a/java/build.sh b/java/build.sh index 37487fef26..ca63cabe26 100755 --- a/java/build.sh +++ b/java/build.sh @@ -8,7 +8,7 @@ set -e -u -o pipefail ARGS="$*" NUMARGS=$# -VERSION="26.10.0" # Note: The version is updated automatically when ci/release/update-version.sh is invoked +VERSION="26.08.0" # Note: The version is updated automatically when ci/release/update-version.sh is invoked GROUP_ID="com.nvidia.cuvs" # Identify CUDA major version. diff --git a/java/cuvs-java/pom.xml b/java/cuvs-java/pom.xml index 7d00a7e9b7..246a6bcb5a 100644 --- a/java/cuvs-java/pom.xml +++ b/java/cuvs-java/pom.xml @@ -9,7 +9,7 @@ SPDX-License-Identifier: Apache-2.0 com.nvidia.cuvs cuvs-java - 26.10.0 + 26.08.0 cuvs-java This project provides Java bindings for cuVS, enabling approximate nearest neighbors search and clustering diff --git a/java/examples/README.md b/java/examples/README.md index e32670246f..4dde184871 100644 --- a/java/examples/README.md +++ b/java/examples/README.md @@ -14,7 +14,7 @@ This maven project contains examples for CAGRA, HNSW, and Bruteforce algorithms. In the current directory do: ``` -mvn package && java --enable-native-access=ALL-UNNAMED -cp target/cuvs-java-examples-26.10.0.jar:$HOME/.m2/repository/com/nvidia/cuvs/cuvs-java/26.10.0/cuvs-java-26.10.0.jar com.nvidia.cuvs.examples.CagraExample +mvn package && java --enable-native-access=ALL-UNNAMED -cp target/cuvs-java-examples-26.08.0.jar:$HOME/.m2/repository/com/nvidia/cuvs/cuvs-java/26.08.0/cuvs-java-26.08.0.jar com.nvidia.cuvs.examples.CagraExample ``` ### HNSW Example @@ -22,7 +22,7 @@ mvn package && java --enable-native-access=ALL-UNNAMED -cp target/cuvs-java-exam In the current directory do: ``` -mvn package && java --enable-native-access=ALL-UNNAMED -cp target/cuvs-java-examples-26.10.0.jar:$HOME/.m2/repository/com/nvidia/cuvs/cuvs-java/26.10.0/cuvs-java-26.10.0.jar com.nvidia.cuvs.examples.HnswExample +mvn package && java --enable-native-access=ALL-UNNAMED -cp target/cuvs-java-examples-26.08.0.jar:$HOME/.m2/repository/com/nvidia/cuvs/cuvs-java/26.08.0/cuvs-java-26.08.0.jar com.nvidia.cuvs.examples.HnswExample ``` ### Bruteforce Example @@ -30,5 +30,5 @@ mvn package && java --enable-native-access=ALL-UNNAMED -cp target/cuvs-java-exam In the current directory do: ``` -mvn package && java --enable-native-access=ALL-UNNAMED -cp target/cuvs-java-examples-26.10.0.jar:$HOME/.m2/repository/com/nvidia/cuvs/cuvs-java/26.10.0/cuvs-java-26.10.0.jar com.nvidia.cuvs.examples.BruteForceExample +mvn package && java --enable-native-access=ALL-UNNAMED -cp target/cuvs-java-examples-26.08.0.jar:$HOME/.m2/repository/com/nvidia/cuvs/cuvs-java/26.08.0/cuvs-java-26.08.0.jar com.nvidia.cuvs.examples.BruteForceExample ``` diff --git a/java/examples/pom.xml b/java/examples/pom.xml index e0c327edde..ea9927c020 100644 --- a/java/examples/pom.xml +++ b/java/examples/pom.xml @@ -8,7 +8,7 @@ SPDX-License-Identifier: Apache-2.0 com.nvidia.cuvs.examples cuvs-java-examples - 26.10.0 + 26.08.0 cuvs-java-examples @@ -75,7 +75,7 @@ SPDX-License-Identifier: Apache-2.0 com.nvidia.cuvs cuvs-java - 26.10.0 + 26.08.0 diff --git a/python/cuvs/pyproject.toml b/python/cuvs/pyproject.toml index bc2047d877..29a9d424ec 100644 --- a/python/cuvs/pyproject.toml +++ b/python/cuvs/pyproject.toml @@ -21,9 +21,9 @@ license = "Apache-2.0" requires-python = ">=3.11" dependencies = [ "cuda-bindings>=13.0.1,<14.0", - "libcuvs==26.10.*,>=0.0.0a0", + "libcuvs==26.8.*,>=0.0.0a0", "numpy>=2.0,<3.0", - "pylibraft==26.10.*,>=0.0.0a0", + "pylibraft==26.8.*,>=0.0.0a0", ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. classifiers = [ "Intended Audience :: Developers", @@ -109,9 +109,9 @@ requires = [ "cmake>=4.0", "cuda-bindings>=13.0.1,<14.0", "cython>=3.2.2", - "libcuvs==26.10.*,>=0.0.0a0", - "libraft==26.10.*,>=0.0.0a0", - "librmm==26.10.*,>=0.0.0a0", + "libcuvs==26.8.*,>=0.0.0a0", + "libraft==26.8.*,>=0.0.0a0", + "librmm==26.8.*,>=0.0.0a0", "ninja", ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. build-backend = "scikit_build_core.build" diff --git a/python/cuvs_bench/cuvs_bench/config/algos/cuvs_cagra_hnswlib.yaml b/python/cuvs_bench/cuvs_bench/config/algos/cuvs_cagra_hnswlib.yaml index f64f06225d..0635022909 100644 --- a/python/cuvs_bench/cuvs_bench/config/algos/cuvs_cagra_hnswlib.yaml +++ b/python/cuvs_bench/cuvs_bench/config/algos/cuvs_cagra_hnswlib.yaml @@ -20,4 +20,4 @@ groups: hierarchy: ["none", "cpu"] ef_construction: [64] search: - ef: [10, 20] + ef: [10,20] diff --git a/python/cuvs_bench/cuvs_bench/config/algos/faiss_cpu_hnsw_flat.yaml b/python/cuvs_bench/cuvs_bench/config/algos/faiss_cpu_hnsw_flat.yaml index f453392e66..403cd29e91 100644 --- a/python/cuvs_bench/cuvs_bench/config/algos/faiss_cpu_hnsw_flat.yaml +++ b/python/cuvs_bench/cuvs_bench/config/algos/faiss_cpu_hnsw_flat.yaml @@ -5,6 +5,6 @@ groups: M: [16, 32] efConstruction: [32, 64, 128, 256, 512] search: - bounded_queue: [false, true] - check_relative_distance: [false, true] + bounded_queue: [False, True] + check_relative_distance: [False, True] efSearch: [8, 16, 32, 64, 128, 256, 512, 768, 1024] diff --git a/python/cuvs_bench/cuvs_bench/config/algos/faiss_cpu_ivf_flat.yaml b/python/cuvs_bench/cuvs_bench/config/algos/faiss_cpu_ivf_flat.yaml index d5a3f4a868..6e11df8d70 100644 --- a/python/cuvs_bench/cuvs_bench/config/algos/faiss_cpu_ivf_flat.yaml +++ b/python/cuvs_bench/cuvs_bench/config/algos/faiss_cpu_ivf_flat.yaml @@ -4,7 +4,7 @@ groups: build: nlist: [2048] ratio: [10] - useFloat16: [false] + useFloat16: [False] search: nprobe: [1, 5, 10, 50, 100, 200] refine_ratio: [1] @@ -12,7 +12,7 @@ groups: build: nlist: [2048] ratio: [10] - useFloat16: [false] + useFloat16: [False] search: nprobe: [1, 5] refine_ratio: [1] diff --git a/python/cuvs_bench/cuvs_bench/config/algos/faiss_gpu_cagra.yaml b/python/cuvs_bench/cuvs_bench/config/algos/faiss_gpu_cagra.yaml index b74715dea0..47a92e3d20 100644 --- a/python/cuvs_bench/cuvs_bench/config/algos/faiss_gpu_cagra.yaml +++ b/python/cuvs_bench/cuvs_bench/config/algos/faiss_gpu_cagra.yaml @@ -10,7 +10,7 @@ groups: cagra_build_algo: ["NN_DESCENT"] search: itopk: [32, 64, 128, 256, 512, 768, 1024] - search_width: [1, 2, 4, 8, 16, 32] + search_width: [1 ,2 ,4 ,8, 16, 32] max_iterations: [16, 20, 24, 28, 32, 64, 128] test: build: diff --git a/python/cuvs_bench/cuvs_bench/config/algos/faiss_gpu_cagra_hnsw.yaml b/python/cuvs_bench/cuvs_bench/config/algos/faiss_gpu_cagra_hnsw.yaml index 2baedf0ff5..097a20aa32 100644 --- a/python/cuvs_bench/cuvs_bench/config/algos/faiss_gpu_cagra_hnsw.yaml +++ b/python/cuvs_bench/cuvs_bench/config/algos/faiss_gpu_cagra_hnsw.yaml @@ -7,6 +7,6 @@ groups: graph_degree: [32, 64] intermediate_graph_degree: [64, 128] cagra_build_algo: ["NN_DESCENT"] - base_level_only: [false, true] + base_level_only: [False, True] search: efSearch: [16, 32, 64, 128, 256, 512, 768, 1024] diff --git a/python/cuvs_bench/cuvs_bench/config/algos/faiss_gpu_ivf_flat.yaml b/python/cuvs_bench/cuvs_bench/config/algos/faiss_gpu_ivf_flat.yaml index 423882a693..148121ec3f 100644 --- a/python/cuvs_bench/cuvs_bench/config/algos/faiss_gpu_ivf_flat.yaml +++ b/python/cuvs_bench/cuvs_bench/config/algos/faiss_gpu_ivf_flat.yaml @@ -4,34 +4,34 @@ groups: build: nlist: [1024, 2048, 4096] ratio: [4] - use_cuvs: [false] + use_cuvs: [False] search: nprobe: [1, 5, 10, 50, 100, 200] basecuvs: build: nlist: [1024, 2048, 4096] ratio: [4] - use_cuvs: [true] + use_cuvs: [True] search: nprobe: [1, 5, 10, 50, 100, 200] large: build: nlist: [8192, 16384, 32768] ratio: [4] - use_cuvs: [false] + use_cuvs: [False] search: nprobe: [10, 20, 30, 50, 100, 200, 500, 1000] largecuvs: build: nlist: [8192, 16384, 32768] ratio: [4] - use_cuvs: [true] + use_cuvs: [True] search: nprobe: [10, 20, 30, 50, 100, 200, 500, 1000] test: build: nlist: [1024] ratio: [4] - use_cuvs: [false] + use_cuvs: [False] search: nprobe: [1, 5] diff --git a/python/cuvs_bench/cuvs_bench/config/algos/faiss_gpu_ivf_pq.yaml b/python/cuvs_bench/cuvs_bench/config/algos/faiss_gpu_ivf_pq.yaml index c34ff29384..a533ff6d8c 100644 --- a/python/cuvs_bench/cuvs_bench/config/algos/faiss_gpu_ivf_pq.yaml +++ b/python/cuvs_bench/cuvs_bench/config/algos/faiss_gpu_ivf_pq.yaml @@ -8,9 +8,9 @@ groups: nlist: [1024, 2048, 4096] M: [96, 64] ratio: [4] - usePrecomputed: [true] - useFloat16: [true] - use_cuvs: [false] + usePrecomputed: [True] + useFloat16: [True] + use_cuvs: [False] bitsPerCode: [8] search: nprobe: [10, 20, 50, 100, 200] @@ -20,9 +20,9 @@ groups: nlist: [1024, 2048, 4096] M: [96, 64] ratio: [4] - usePrecomputed: [false] - useFloat16: [false] - use_cuvs: [true] + usePrecomputed: [False] + useFloat16: [False] + use_cuvs: [True] bitsPerCode: [8, 6, 5, 4] search: nprobe: [10, 20, 50, 100, 200] @@ -32,9 +32,9 @@ groups: nlist: [8192, 16384, 32768] M: [96, 48, 32] ratio: [4] - usePrecomputed: [false] - useFloat16: [false, true] - use_cuvs: [false] + usePrecomputed: [False] + useFloat16: [False, True] + use_cuvs: [False] bitsPerCode: [8] search: nprobe: [20, 30, 50, 100, 200, 500, 1000] @@ -44,9 +44,9 @@ groups: nlist: [8192, 16384, 32768] M: [96, 48, 32] ratio: [4] - usePrecomputed: [false] - useFloat16: [false] - use_cuvs: [true] + usePrecomputed: [False] + useFloat16: [False] + use_cuvs: [True] bitsPerCode: [8, 6, 5, 4] search: nprobe: [20, 30, 50, 100, 200, 500, 1000] @@ -56,9 +56,9 @@ groups: nlist: [8192, 16384, 32768] M: [96, 48] ratio: [10] - usePrecomputed: [false, true] - useFloat16: [true] - use_cuvs: [false] + usePrecomputed: [False, True] + useFloat16: [True] + use_cuvs: [False] bitsPerCode: [8] search: nprobe: [20, 30, 40, 50, 100, 200, 500, 1000] @@ -68,8 +68,8 @@ groups: nlist: [8192, 16384, 50000, 100000] M: [96, 48] ratio: [10] - useFloat16: [false, true] - use_cuvs: [true] + useFloat16: [False, True] + use_cuvs: [True] bitsPerCode: [8, 6, 5, 4] search: nprobe: [20, 30, 40, 50, 100, 200, 500, 1000] @@ -79,9 +79,9 @@ groups: nlist: [1024] M: [96] ratio: [4] - usePrecomputed: [true] - useFloat16: [true] - use_cuvs: [false] + usePrecomputed: [True] + useFloat16: [True] + use_cuvs: [False] bitsPerCode: [8] search: nprobe: [10] diff --git a/python/cuvs_bench/pyproject.toml b/python/cuvs_bench/pyproject.toml index 2db75baf38..a946e7fd08 100644 --- a/python/cuvs_bench/pyproject.toml +++ b/python/cuvs_bench/pyproject.toml @@ -20,7 +20,7 @@ license = "Apache-2.0" requires-python = ">=3.11" dependencies = [ "click", - "cuvs==26.10.*,>=0.0.0a0", + "cuvs==26.8.*,>=0.0.0a0", "matplotlib>=3.9", "pandas", "pyyaml", diff --git a/python/libcuvs/pyproject.toml b/python/libcuvs/pyproject.toml index 48be98cc3c..388809bcc3 100644 --- a/python/libcuvs/pyproject.toml +++ b/python/libcuvs/pyproject.toml @@ -20,8 +20,8 @@ license = "Apache-2.0" requires-python = ">=3.11" dependencies = [ "cuda-toolkit[cublas,curand,cusolver,cusparse,nvrtc]==13.*", - "libraft==26.10.*,>=0.0.0a0", - "librmm==26.10.*,>=0.0.0a0", + "libraft==26.8.*,>=0.0.0a0", + "librmm==26.8.*,>=0.0.0a0", "nvidia-nvjitlink>=13.3,<14", ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. classifiers = [ @@ -81,8 +81,8 @@ regex = "(?P.*)" build-backend = "scikit_build_core.build" requires = [ "cmake>=4.0", - "libraft==26.10.*,>=0.0.0a0", - "librmm==26.10.*,>=0.0.0a0", + "libraft==26.8.*,>=0.0.0a0", + "librmm==26.8.*,>=0.0.0a0", "ninja", ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. dependencies-file = "../../dependencies.yaml" diff --git a/rust/Cargo.toml b/rust/Cargo.toml index 4900076023..e5b6809ff9 100644 --- a/rust/Cargo.toml +++ b/rust/Cargo.toml @@ -9,7 +9,7 @@ members = [ resolver = "2" [workspace.package] -version = "26.10.0" +version = "26.8.0" edition = "2024" repository = "https://github.com/rapidsai/cuvs" homepage = "https://github.com/rapidsai/cuvs" @@ -18,4 +18,4 @@ authors = ["NVIDIA Corporation"] license = "Apache-2.0" [workspace.dependencies] -cuvs-sys = { path = "cuvs-sys", version = "26.10.0" } +cuvs-sys = { path = "cuvs-sys", version = "26.8.0" }