NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-29 19:21:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change to GPU runner for testing

Browse Source
This commit is contained in:
Ashwin Srinath
2025-12-03 15:18:39 -05:00
parent b7eaf44ca3
commit 7ad064ea4f
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
.github/workflows/build-and-test-python-wheels.yml vendored
View File

@@ -48,7 +48,7 @@ jobs:
test-wheels: test-wheels:
name: Test wheel (CUDA ${{ matrix.cuda }}, Python ${{ matrix.python }}) name: Test wheel (CUDA ${{ matrix.cuda }}, Python ${{ matrix.python }})
needs: build-wheels needs: build-wheels
runs-on: ubuntu-latest runs-on: linux-amd64-gpu-l4-latest-1
permissions: permissions:
id-token: write id-token: write
contents: read contents: read
@@ -82,6 +82,7 @@ jobs:
docker run --rm \ docker run --rm \
--workdir /workspace \ --workdir /workspace \
--gpus all \
--mount type=bind,source=$(pwd),target=/workspace/ \ --mount type=bind,source=$(pwd),target=/workspace/ \
--env py_version=${{ matrix.python }} \ --env py_version=${{ matrix.python }} \
--env cuda_version=${{ matrix.cuda }} \ --env cuda_version=${{ matrix.cuda }} \