Skip to content

GitLab

Unverified Commit 7f9fdb6e authored by Anant Sharma's avatar Anant Sharma Committed by GitHub
Browse files

ci: enable GPU access for rust-checks in dynamo validation (#6545) 


Signed-off-by: default avatarAnant Sharma <anants@nvidia.com>
parent 8e236375
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
.github/workflows/container-validation-dynamo.yml
View file @ 7f9fdb6e
...@@ -121,7 +121,7 @@ jobs: ...@@ -121,7 +121,7 @@ jobs:
rust-checks: rust-checks:
needs: [changed-files, build] needs: [changed-files, build]
if: needs.changed-files.outputs.core == 'true' if: needs.changed-files.outputs.core == 'true'
runs-on: prod-builder-amd-v1 runs-on: prod-tester-amd-gpu-v1
name: Rust Checks name: Rust Checks
timeout-minutes: 30 timeout-minutes: 30
env: env:
...@@ -139,7 +139,7 @@ jobs: ...@@ -139,7 +139,7 @@ jobs:
run: docker pull ${{ env.IMAGE_TAG }} run: docker pull ${{ env.IMAGE_TAG }}
- name: Run Rust checks (block-manager + media-ffmpeg + integration tests) - name: Run Rust checks (block-manager + media-ffmpeg + integration tests)
run: | run: |
docker run --rm --user root -w /workspace/lib/llm \ docker run --rm --runtime=nvidia --gpus all --user root -w /workspace/lib/llm \
--name ${{ env.CONTAINER_ID }}_rust_checks \ --name ${{ env.CONTAINER_ID }}_rust_checks \
-e SCCACHE_BUCKET=${{ secrets.SCCACHE_S3_BUCKET }} \ -e SCCACHE_BUCKET=${{ secrets.SCCACHE_S3_BUCKET }} \
-e SCCACHE_REGION=${{ secrets.AWS_DEFAULT_REGION }} \ -e SCCACHE_REGION=${{ secrets.AWS_DEFAULT_REGION }} \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment