Skip to content

Commit ce1d7ea

Browse files
aamijaraamijar
authored
Merge branch 'main' into ivf-rabitq-ref
2 parents 1f1bedf + c95fc4e commit ce1d7ea

1,377 files changed

Lines changed: 129116 additions & 32263 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

‎.coderabbit.yaml‎

Lines changed: 77 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,77 @@
1+
# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# SPDX-License-Identifier: Apache-2.0
3+
4+
reviews:
5+
profile: chill
6+
high_level_summary: true
7+
high_level_summary_in_walkthrough: true
8+
poem: false
9+
auto_review:
10+
enabled: true
11+
drafts: false
12+
base_branches:
13+
- "^main$"
14+
- "^release/.*"
15+
- "^hotfix/.*"
16+
ignore_usernames: ["rapids-bot", "GPUtester", "nv-automation-bot", "copy-pr-bot"]
17+
tools:
18+
markdownlint:
19+
enabled: true
20+
shellcheck:
21+
enabled: true
22+
gitleaks:
23+
enabled: true
24+
sequence_diagrams: false
25+
collapse_walkthrough: true
26+
27+
# Reduce noise from status messages
28+
request_changes_workflow: false
29+
review_status: false
30+
31+
# Path-specific review instructions
32+
# Note: Detailed C++/CUDA and Python review guidelines are in cpp/REVIEW_GUIDELINES.md and python/REVIEW_GUIDELINES.md
33+
path_instructions:
34+
- path: "docs/**/*"
35+
instructions: |
36+
For documentation changes, focus on:
37+
- Accuracy: Verify code examples compile and run correctly
38+
- Completeness: Check if API changes (parameters, return values, errors) are documented
39+
- Clarity: Flag confusing explanations, missing prerequisites, or unclear examples
40+
- Consistency: Version numbers, parameter types, and terminology match code
41+
- Missing docs: If PR changes public APIs without updating docs, flag as HIGH priority
42+
43+
- path: "c/include/cuvs/**/*"
44+
instructions: |
45+
For public C API headers, additionally check:
46+
- Doxygen documentation for all public functions/classes
47+
- API changes flagged for docs/ updates
48+
- Breaking changes require deprecation warnings and migration guide updates
49+
50+
- path: "cpp/include/cuvs/**/*"
51+
instructions: |
52+
For public C++ API headers, additionally check:
53+
- Doxygen documentation for all public functions/classes
54+
- API changes flagged for docs/ updates
55+
- Breaking changes require deprecation warnings and migration guide updates
56+
57+
- path: "notebooks/**/*"
58+
instructions: |
59+
For example notebooks:
60+
- Verify code examples match current API (parameter names, return types)
61+
- Flag outdated imports or deprecated API usage
62+
63+
- path: "ci/**/*"
64+
instructions: |
65+
For CI/build scripts:
66+
- Check for proper conda environment handling
67+
- Verify GPU availability checks before tests
68+
- Check for proper error handling and meaningful error messages
69+
70+
knowledge_base:
71+
opt_out: false
72+
code_guidelines:
73+
filePatterns:
74+
- "cpp/REVIEW_GUIDELINES.md"
75+
- "python/REVIEW_GUIDELINES.md"
76+
- "docs/source/contributing.md"
77+
- "docs/source/developer_guide.md"

‎.devcontainer/Dockerfile‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# SPDX-FileCopyrightText: Copyright (c) 2023-2025, NVIDIA CORPORATION. All rights reserved.
1+
# SPDX-FileCopyrightText: Copyright (c) 2023-2026, NVIDIA CORPORATION. All rights reserved.
22
# SPDX-License-Identifier: Apache-2.0
33

44
# syntax=docker/dockerfile:1.5
@@ -42,7 +42,8 @@ ENV HISTFILE="/home/coder/.cache/._bash_history"
4242
###
4343
# sccache configuration
4444
###
45-
ENV AWS_ROLE_ARN="arn:aws:iam::279114543810:role/nv-gha-token-sccache-devs"
45+
ENV AWS_IDP_URL="https://token.rapids.nvidia.com"
46+
ENV AWS_ROLE_ARN="arn:aws:iam::279114543810:role/rapids-token-sccache-devs"
4647
ENV SCCACHE_REGION="us-east-2"
4748
ENV SCCACHE_BUCKET="rapids-sccache-devs"
4849
ENV SCCACHE_S3_USE_PREPROCESSOR_CACHE_MODE=true

‎.devcontainer/README.md‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
1-
# RAFT Development Containers
1+
# cuVS Development Containers
22

33
This directory contains [devcontainer configurations](https://containers.dev/implementors/json_reference/) for using VSCode to [develop in a container](https://code.visualstudio.com/docs/devcontainers/containers) via the `Remote Containers` [extension](https://marketplace.visualstudio.com/items?itemName=ms-vscode-remote.remote-containers) or [GitHub Codespaces](https://github.com/codespaces).
44

5-
This container is a turnkey development environment for building and testing the RAFT C++ and Python libraries.
5+
This container is a turnkey development environment for building and testing the cuVS C++ and Python libraries.
66

77
## Table of Contents
88

@@ -30,7 +30,7 @@ This ensures caches, configurations, dependencies, and your commits are persiste
3030

3131
## Launch a Dev Container
3232

33-
To launch a devcontainer from VSCode, open the RAFT repo and select the "Reopen in Container" button in the bottom right:<br/><img src="https://user-images.githubusercontent.com/178183/221771999-97ab29d5-e718-4e5f-b32f-2cdd51bba25c.png"/>
33+
To launch a devcontainer from VSCode, open the cuVS repo and select the "Reopen in Container" button in the bottom right:<br/><img src="https://user-images.githubusercontent.com/178183/221771999-97ab29d5-e718-4e5f-b32f-2cdd51bba25c.png"/>
3434

3535
Alternatively, open the VSCode command palette (typically `cmd/ctrl + shift + P`) and run the "Rebuild and Reopen in Container" command.
3636

‎.devcontainer/cuda12.9-conda/devcontainer.json‎

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,19 +5,22 @@
55
"args": {
66
"CUDA": "12.9",
77
"PYTHON_PACKAGE_MANAGER": "conda",
8-
"BASE": "rapidsai/devcontainers:26.06-cpp-mambaforge"
9-
}
8+
"BASE": "rapidsai/devcontainers:26.08-cpp-mambaforge"
9+
},
10+
"cacheFrom": [
11+
"ghcr.io/rapidsai/cuvs/devcontainer:26.08-cuda12.9-conda"
12+
]
1013
},
1114
"runArgs": [
1215
"--rm",
1316
"--name",
14-
"${localEnv:USER:anon}-rapids-${localWorkspaceFolderBasename}-26.06-cuda12.9-conda",
17+
"${localEnv:USER:anon}-rapids-${localWorkspaceFolderBasename}-26.08-cuda12.9-conda",
1518
"--ulimit",
1619
"nofile=500000"
1720
],
1821
"hostRequirements": {"gpu": "optional"},
1922
"features": {
20-
"ghcr.io/rapidsai/devcontainers/features/rapids-build-utils:26.6": {}
23+
"ghcr.io/rapidsai/devcontainers/features/rapids-build-utils:26.8": {}
2124
},
2225
"overrideFeatureInstallOrder": [
2326
"ghcr.io/rapidsai/devcontainers/features/rapids-build-utils"

‎.devcontainer/cuda12.9-pip/devcontainer.json‎

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -5,26 +5,29 @@
55
"args": {
66
"CUDA": "12.9",
77
"PYTHON_PACKAGE_MANAGER": "pip",
8-
"BASE": "rapidsai/devcontainers:26.06-cpp-cuda12.9-ucx1.19.0-openmpi5.0.7"
9-
}
8+
"BASE": "rapidsai/devcontainers:26.08-cpp-cuda12.9-ucx1.19.0-openmpi5.0.10"
9+
},
10+
"cacheFrom": [
11+
"ghcr.io/rapidsai/cuvs/devcontainer:26.08-cuda12.9-pip"
12+
]
1013
},
1114
"runArgs": [
1215
"--rm",
1316
"--name",
14-
"${localEnv:USER:anon}-rapids-${localWorkspaceFolderBasename}-26.06-cuda12.9-pip",
17+
"${localEnv:USER:anon}-rapids-${localWorkspaceFolderBasename}-26.08-cuda12.9-pip",
1518
"--ulimit",
1619
"nofile=500000"
1720
],
1821
"hostRequirements": {"gpu": "optional"},
1922
"features": {
20-
"ghcr.io/rapidsai/devcontainers/features/cuda:26.6": {
23+
"ghcr.io/rapidsai/devcontainers/features/cuda:26.8": {
2124
"version": "12.9",
2225
"installcuBLAS": true,
2326
"installcuSOLVER": true,
2427
"installcuRAND": true,
2528
"installcuSPARSE": true
2629
},
27-
"ghcr.io/rapidsai/devcontainers/features/rapids-build-utils:26.6": {}
30+
"ghcr.io/rapidsai/devcontainers/features/rapids-build-utils:26.8": {}
2831
},
2932
"overrideFeatureInstallOrder": [
3033
"ghcr.io/rapidsai/devcontainers/features/ucx",

‎.devcontainer/cuda13.1-conda/devcontainer.json‎

Lines changed: 0 additions & 44 deletions
This file was deleted.

‎.devcontainer/cuda13.1-pip/devcontainer.json‎

Lines changed: 0 additions & 52 deletions
This file was deleted.

‎.devcontainer/cuda13.3-conda/devcontainer.json‎

Lines changed: 47 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,47 @@
1+
{
2+
"build": {
3+
"context": "${localWorkspaceFolder}/.devcontainer",
4+
"dockerfile": "${localWorkspaceFolder}/.devcontainer/Dockerfile",
5+
"args": {
6+
"CUDA": "13.3",
7+
"PYTHON_PACKAGE_MANAGER": "conda",
8+
"BASE": "rapidsai/devcontainers:26.08-cpp-mambaforge"
9+
},
10+
"cacheFrom": [
11+
"ghcr.io/rapidsai/cuvs/devcontainer:26.08-cuda13.3-conda"
12+
]
13+
},
14+
"runArgs": [
15+
"--rm",
16+
"--name",
17+
"${localEnv:USER:anon}-rapids-${localWorkspaceFolderBasename}-26.08-cuda13.3-conda",
18+
"--ulimit",
19+
"nofile=500000"
20+
],
21+
"hostRequirements": {"gpu": "optional"},
22+
"features": {
23+
"ghcr.io/rapidsai/devcontainers/features/rapids-build-utils:26.8": {}
24+
},
25+
"overrideFeatureInstallOrder": [
26+
"ghcr.io/rapidsai/devcontainers/features/rapids-build-utils"
27+
],
28+
"initializeCommand": ["/bin/bash", "-c", "mkdir -m 0755 -p ${localWorkspaceFolder}/../.{aws,cache,config,conda/pkgs,conda/${localWorkspaceFolderBasename}-cuda13.3-envs}"],
29+
"postAttachCommand": ["/bin/bash", "-c", "if [ ${CODESPACES:-false} = 'true' ]; then . devcontainer-utils-post-attach-command; . rapids-post-attach-command; fi"],
30+
"workspaceFolder": "/home/coder",
31+
"workspaceMount": "source=${localWorkspaceFolder},target=/home/coder/cuvs,type=bind,consistency=consistent",
32+
"mounts": [
33+
"source=${localWorkspaceFolder}/../.aws,target=/home/coder/.aws,type=bind,consistency=consistent",
34+
"source=${localWorkspaceFolder}/../.cache,target=/home/coder/.cache,type=bind,consistency=consistent",
35+
"source=${localWorkspaceFolder}/../.config,target=/home/coder/.config,type=bind,consistency=consistent",
36+
"source=${localWorkspaceFolder}/../.conda/pkgs,target=/home/coder/.conda/pkgs,type=bind,consistency=consistent",
37+
"source=${localWorkspaceFolder}/../.conda/${localWorkspaceFolderBasename}-cuda13.3-envs,target=/home/coder/.conda/envs,type=bind,consistency=consistent"
38+
],
39+
"customizations": {
40+
"vscode": {
41+
"extensions": [
42+
"ms-python.flake8",
43+
"nvidia.nsight-vscode-edition"
44+
]
45+
}
46+
}
47+
}

‎.devcontainer/cuda13.3-pip/devcontainer.json‎

Lines changed: 55 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,55 @@
1+
{
2+
"build": {
3+
"context": "${localWorkspaceFolder}/.devcontainer",
4+
"dockerfile": "${localWorkspaceFolder}/.devcontainer/Dockerfile",
5+
"args": {
6+
"CUDA": "13.3",
7+
"PYTHON_PACKAGE_MANAGER": "pip",
8+
"BASE": "rapidsai/devcontainers:26.08-cpp-cuda13.3-ucx1.19.0-openmpi5.0.10"
9+
},
10+
"cacheFrom": [
11+
"ghcr.io/rapidsai/cuvs/devcontainer:26.08-cuda13.3-pip"
12+
]
13+
},
14+
"runArgs": [
15+
"--rm",
16+
"--name",
17+
"${localEnv:USER:anon}-rapids-${localWorkspaceFolderBasename}-26.08-cuda13.3-pip",
18+
"--ulimit",
19+
"nofile=500000"
20+
],
21+
"hostRequirements": {"gpu": "optional"},
22+
"features": {
23+
"ghcr.io/rapidsai/devcontainers/features/cuda:26.8": {
24+
"version": "13.3",
25+
"installcuBLAS": true,
26+
"installcuSOLVER": true,
27+
"installcuRAND": true,
28+
"installcuSPARSE": true
29+
},
30+
"ghcr.io/rapidsai/devcontainers/features/rapids-build-utils:26.8": {}
31+
},
32+
"overrideFeatureInstallOrder": [
33+
"ghcr.io/rapidsai/devcontainers/features/ucx",
34+
"ghcr.io/rapidsai/devcontainers/features/cuda",
35+
"ghcr.io/rapidsai/devcontainers/features/rapids-build-utils"
36+
],
37+
"initializeCommand": ["/bin/bash", "-c", "mkdir -m 0755 -p ${localWorkspaceFolder}/../.{aws,cache,config/pip,local/share/${localWorkspaceFolderBasename}-cuda13.3-venvs}"],
38+
"postAttachCommand": ["/bin/bash", "-c", "if [ ${CODESPACES:-false} = 'true' ]; then . devcontainer-utils-post-attach-command; . rapids-post-attach-command; fi"],
39+
"workspaceFolder": "/home/coder",
40+
"workspaceMount": "source=${localWorkspaceFolder},target=/home/coder/cuvs,type=bind,consistency=consistent",
41+
"mounts": [
42+
"source=${localWorkspaceFolder}/../.aws,target=/home/coder/.aws,type=bind,consistency=consistent",
43+
"source=${localWorkspaceFolder}/../.cache,target=/home/coder/.cache,type=bind,consistency=consistent",
44+
"source=${localWorkspaceFolder}/../.config,target=/home/coder/.config,type=bind,consistency=consistent",
45+
"source=${localWorkspaceFolder}/../.local/share/${localWorkspaceFolderBasename}-cuda13.3-venvs,target=/home/coder/.local/share/venvs,type=bind,consistency=consistent"
46+
],
47+
"customizations": {
48+
"vscode": {
49+
"extensions": [
50+
"ms-python.flake8",
51+
"nvidia.nsight-vscode-edition"
52+
]
53+
}
54+
}
55+
}

0 commit comments

Comments
 (0)