Skip to content

feat: add cuVS accelerator support #6366

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
Xuanwo wants to merge 8 commits into
base: main
Choose a base branch
from
Open

feat: add cuVS accelerator support #6366

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
f2be9cc
python: add explicit cuvs accelerator path
Xuanwo Mar 31, 2026
2f071f6
python: document cuvs installation requirements
Xuanwo Mar 31, 2026
1a6c44b
python: fix cuvs training on real datasets
Xuanwo Apr 1, 2026
76995c0
python: format cuvs helper
Xuanwo Apr 1, 2026
fbe0f50
python: clarify accelerator hardware requirements
Xuanwo Apr 1, 2026
f00d078
python: add cuvs one-pass ivfpq assignment
Xuanwo Apr 1, 2026
7f7e6e2
python: use cuvs transform for full ivf pq build
Xuanwo Apr 1, 2026
f9c5d03
python: route cuvs precomputed shuffle to v3 files
Xuanwo Apr 2, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 16 additions & 0 deletions python/DEVELOPMENT.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,22 @@ uv sync --extra tests --extra dev

Add extras such as `benchmarks`, `torch`, or `geo` only when you need them. After the environment is initialized, either activate it or use `uv run ...` for commands.

`accelerator="cuvs"` does not have a normal project extra today. cuVS Python
packages are published per CUDA major version and are typically installed from
NVIDIA's package index, for example:

```shell
uv pip install --extra-index-url https://pypi.nvidia.com cuvs-cu12
```

or:

```shell
uv pip install --extra-index-url https://pypi.nvidia.com cuvs-cu13
```

Pick the package that matches the CUDA version in your environment.

`uv sync` is not just downloading Python packages here. It also builds the local `pylance` Rust extension as part of the editable environment, so the first run, cache misses, or Rust dependency changes can make it noticeably slow. This is expected; let the build finish instead of interrupting it and switching to a different environment setup.

## Building the project
Expand Down
Loading
Loading