Skip to content

Commit ed345b6

Browse files
authored
Merge branch 'main' into restricted_paths_dummy_change
2 parents 9913d8c + e154500 commit ed345b6

File tree

146 files changed

+6097
-2521
lines changed

Some content is hidden

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

146 files changed

+6097
-2521
lines changed

.coveragerc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2-
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
2+
# SPDX-License-Identifier: Apache-2.0
33

44
[paths]
55
source =

.github/actionlint.yaml

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
#
3+
# SPDX-License-Identifier: Apache-2.0
4+
5+
# Self-hosted runner labels used in CI workflows.
6+
# Without this config, actionlint rejects non-GitHub-hosted labels.
7+
self-hosted-runner:
8+
labels:
9+
- linux-amd64-cpu8

.github/labeler.yml

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
#
3+
# SPDX-License-Identifier: Apache-2.0
4+
5+
# Configuration for https://github.com/actions/labeler
6+
# Auto-applies labels based on which files a PR changes.
7+
8+
cuda.bindings:
9+
- changed-files:
10+
- any-glob-to-any-file: 'cuda_bindings/**'
11+
12+
cuda.core:
13+
- changed-files:
14+
- any-glob-to-any-file: 'cuda_core/**'
15+
16+
cuda.pathfinder:
17+
- changed-files:
18+
- any-glob-to-any-file: 'cuda_pathfinder/**'
19+
20+
CI/CD:
21+
- changed-files:
22+
- any-glob-to-any-file:
23+
- '.github/**'
24+
- 'ci/**'

.github/workflows/build-wheel.yml

Lines changed: 18 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -44,20 +44,24 @@ jobs:
4444
with:
4545
fetch-depth: 0
4646

47-
# The env vars ACTIONS_CACHE_SERVICE_V2, ACTIONS_RESULTS_URL, and ACTIONS_RUNTIME_TOKEN
48-
# are exposed by this action.
49-
- name: Enable sccache
50-
uses: mozilla-actions/sccache-action@7d986dd989559c6ecdb630a3fd2557667be217ad # 0.0.9
51-
with:
52-
disable_annotations: 'true'
47+
- name: Install latest rapidsai/sccache
48+
if: ${{ startsWith(inputs.host-platform, 'linux') }}
49+
run: |
50+
curl -fsSL "https://github.com/rapidsai/sccache/releases/latest/download/sccache-$(uname -m)-unknown-linux-musl.tar.gz" \
51+
| sudo tar -C /usr/local/bin -xvzf - --wildcards --strip-components=1 -x '*/sccache'
52+
echo "SCCACHE_PATH=/usr/local/bin/sccache" >> "$GITHUB_ENV"
53+
echo "SCCACHE_GHA_USE_PREPROCESSOR_CACHE_MODE=true" >> "$GITHUB_ENV"
5354
5455
# xref: https://github.com/orgs/community/discussions/42856#discussioncomment-7678867
5556
- name: Adding addtional GHA cache-related env vars
5657
uses: actions/github-script@v8
5758
with:
5859
script: |
59-
core.exportVariable('ACTIONS_CACHE_URL', process.env['ACTIONS_CACHE_URL'])
60-
core.exportVariable('ACTIONS_RUNTIME_URL', process.env['ACTIONS_RUNTIME_URL'])
60+
core.exportVariable('ACTIONS_CACHE_SERVICE_V2', 'on');
61+
core.exportVariable('ACTIONS_CACHE_URL', process.env['ACTIONS_CACHE_URL'] || '');
62+
core.exportVariable('ACTIONS_RESULTS_URL', process.env['ACTIONS_RESULTS_URL'] || '');
63+
core.exportVariable('ACTIONS_RUNTIME_URL', process.env['ACTIONS_RUNTIME_URL'] || '');
64+
core.exportVariable('ACTIONS_RUNTIME_TOKEN', process.env['ACTIONS_RUNTIME_TOKEN'] || '');
6165
6266
- name: Setup proxy cache
6367
uses: nv-gha-runners/setup-proxy-cache@main
@@ -176,14 +180,15 @@ jobs:
176180
ACTIONS_RESULTS_URL=${{ env.ACTIONS_RESULTS_URL }}
177181
ACTIONS_CACHE_URL=${{ env.ACTIONS_CACHE_URL }}
178182
ACTIONS_CACHE_SERVICE_V2=${{ env.ACTIONS_CACHE_SERVICE_V2 }}
183+
SCCACHE_GHA_USE_PREPROCESSOR_CACHE_MODE=${{ env.SCCACHE_GHA_USE_PREPROCESSOR_CACHE_MODE }}
179184
SCCACHE_DIR=/host/${{ env.SCCACHE_DIR }}
180185
SCCACHE_CACHE_SIZE=${{ env.SCCACHE_CACHE_SIZE }}
181186
CIBW_ENVIRONMENT_WINDOWS: >
182187
CUDA_PATH="$(cygpath -w ${{ env.CUDA_PATH }})"
183188
CUDA_PYTHON_PARALLEL_LEVEL=${{ env.CUDA_PYTHON_PARALLEL_LEVEL }}
184189
# check cache stats before leaving cibuildwheel
185190
CIBW_BEFORE_TEST_LINUX: >
186-
"/host/${{ env.SCCACHE_PATH }}" --show-stats &&
191+
"/host/${{ env.SCCACHE_PATH }}" --show-adv-stats &&
187192
"/host/${{ env.SCCACHE_PATH }}" --show-stats --stats-format=json > /host/${{ github.workspace }}/sccache_bindings.json
188193
# force the test stage to be run (so that before-test is not skipped)
189194
# TODO: we might want to think twice on adding this, it does a lot of
@@ -241,6 +246,7 @@ jobs:
241246
ACTIONS_RESULTS_URL=${{ env.ACTIONS_RESULTS_URL }}
242247
ACTIONS_CACHE_URL=${{ env.ACTIONS_CACHE_URL }}
243248
ACTIONS_CACHE_SERVICE_V2=${{ env.ACTIONS_CACHE_SERVICE_V2 }}
249+
SCCACHE_GHA_USE_PREPROCESSOR_CACHE_MODE=${{ env.SCCACHE_GHA_USE_PREPROCESSOR_CACHE_MODE }}
244250
SCCACHE_DIR=/host/${{ env.SCCACHE_DIR }}
245251
SCCACHE_CACHE_SIZE=${{ env.SCCACHE_CACHE_SIZE }}
246252
CIBW_ENVIRONMENT_WINDOWS: >
@@ -250,7 +256,7 @@ jobs:
250256
PIP_FIND_LINKS="$(cygpath -w ${{ env.CUDA_BINDINGS_ARTIFACTS_DIR }})"
251257
# check cache stats before leaving cibuildwheel
252258
CIBW_BEFORE_TEST_LINUX: >
253-
"/host${{ env.SCCACHE_PATH }}" --show-stats &&
259+
"/host${{ env.SCCACHE_PATH }}" --show-adv-stats &&
254260
"/host${{ env.SCCACHE_PATH }}" --show-stats --stats-format=json > /host/${{ github.workspace }}/sccache_core.json
255261
# force the test stage to be run (so that before-test is not skipped)
256262
# TODO: we might want to think twice on adding this, it does a lot of
@@ -429,6 +435,7 @@ jobs:
429435
ACTIONS_RESULTS_URL=${{ env.ACTIONS_RESULTS_URL }}
430436
ACTIONS_CACHE_URL=${{ env.ACTIONS_CACHE_URL }}
431437
ACTIONS_CACHE_SERVICE_V2=${{ env.ACTIONS_CACHE_SERVICE_V2 }}
438+
SCCACHE_GHA_USE_PREPROCESSOR_CACHE_MODE=${{ env.SCCACHE_GHA_USE_PREPROCESSOR_CACHE_MODE }}
432439
SCCACHE_DIR=/host/${{ env.SCCACHE_DIR }}
433440
SCCACHE_CACHE_SIZE=${{ env.SCCACHE_CACHE_SIZE }}
434441
CIBW_ENVIRONMENT_WINDOWS: >
@@ -438,7 +445,7 @@ jobs:
438445
PIP_FIND_LINKS="$(cygpath -w ${{ env.CUDA_BINDINGS_ARTIFACTS_DIR }})"
439446
# check cache stats before leaving cibuildwheel
440447
CIBW_BEFORE_TEST_LINUX: >
441-
"/host${{ env.SCCACHE_PATH }}" --show-stats &&
448+
"/host${{ env.SCCACHE_PATH }}" --show-adv-stats &&
442449
"/host${{ env.SCCACHE_PATH }}" --show-stats --stats-format=json > /host/${{ github.workspace }}/sccache_core_prev.json
443450
# force the test stage to be run (so that before-test is not skipped)
444451
# TODO: we might want to think twice on adding this, it does a lot of

0 commit comments

Comments
 (0)