Unverified Commit 676ce434 authored by Dmitry Tokarev's avatar Dmitry Tokarev Committed by GitHub
Browse files

fix: Fixed CUDA VERSION value format for VLLM ARM CI job (#5018)


Signed-off-by: default avatarDmitry Tokarev <dtokarev@nvidia.com>
parent da59c6ae
...@@ -163,7 +163,7 @@ jobs: ...@@ -163,7 +163,7 @@ jobs:
platform: 'linux/${{ matrix.platform.arch }}' platform: 'linux/${{ matrix.platform.arch }}'
base_image_tag: ${{ matrix.platform.arch == 'arm64' && '25.06-cuda12.9-devel-ubuntu24.04' || '' }} base_image_tag: ${{ matrix.platform.arch == 'arm64' && '25.06-cuda12.9-devel-ubuntu24.04' || '' }}
runtime_image_tag: ${{ matrix.platform.arch == 'arm64' && '12.9.0-runtime-ubuntu24.04' || '' }} runtime_image_tag: ${{ matrix.platform.arch == 'arm64' && '12.9.0-runtime-ubuntu24.04' || '' }}
cuda_version: ${{ matrix.platform.arch == 'arm64' && '129' || '' }} cuda_version: ${{ matrix.platform.arch == 'arm64' && '12.9' || '' }}
torch_backend: ${{ matrix.platform.arch == 'arm64' && 'cu129' || '' }} torch_backend: ${{ matrix.platform.arch == 'arm64' && 'cu129' || '' }}
ci_token: ${{ secrets.CI_TOKEN }} ci_token: ${{ secrets.CI_TOKEN }}
aws_default_region: ${{ secrets.AWS_DEFAULT_REGION }} aws_default_region: ${{ secrets.AWS_DEFAULT_REGION }}
......
...@@ -126,12 +126,12 @@ jobs: ...@@ -126,12 +126,12 @@ jobs:
- framework: vllm - framework: vllm
base_image_tag: '25.06-cuda12.9-devel-ubuntu24.04' base_image_tag: '25.06-cuda12.9-devel-ubuntu24.04'
runtime_image_tag: '12.9.0-runtime-ubuntu24.04' runtime_image_tag: '12.9.0-runtime-ubuntu24.04'
cuda_version: '129' cuda_version: '12.9'
torch_backend: 'cu129' torch_backend: 'cu129'
- framework: trtllm - framework: trtllm
base_image_tag: '25.06-py3' base_image_tag: '25.06-py3'
runtime_image_tag: '' runtime_image_tag: ''
cuda_version: '129' cuda_version: '12.9'
torch_backend: 'cu129' torch_backend: 'cu129'
- framework: sglang - framework: sglang
base_image_tag: '' base_image_tag: ''
......
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