Skip to content

Commit 202ef09

Browse files
committed
feat(vllm-tensorizer): Build matrix for CUDA 12.9 and CUDA 13.2
1 parent cc65ad3 commit 202ef09

2 files changed

Lines changed: 14 additions & 11 deletions

File tree

Lines changed: 13 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,13 @@
1-
vllm-commit:
2-
- 'v0.20.2'
3-
flashinfer-commit:
4-
- 'v0.6.8'
5-
lmcache-commit:
6-
- 'v0.4.2'
7-
builder-base-image:
8-
- 'ghcr.io/coreweave/ml-containers/torch:bc8c66e-nccl-cuda13.2.1-ubuntu24.04-nccl2.30.4-1-torch2.11.0-vision0.26.0-audio2.11.0-abi1'
9-
final-base-image:
10-
- 'ghcr.io/coreweave/ml-containers/torch:bc8c66e-nccl-cuda13.2.1-ubuntu24.04-nccl2.30.4-1-torch2.11.0-vision0.26.0-audio2.11.0-abi1'
1+
include:
2+
- vllm-commit: 'v0.20.2'
3+
flashinfer-commit: 'v0.6.8'
4+
lmcache-commit: 'v0.4.2'
5+
builder-base-image: 'ghcr.io/coreweave/ml-containers/torch:bc8c66e-nccl-cuda13.2.1-ubuntu24.04-nccl2.30.4-1-torch2.11.0-vision0.26.0-audio2.11.0-abi1'
6+
final-base-image: 'ghcr.io/coreweave/ml-containers/torch:bc8c66e-nccl-cuda13.2.1-ubuntu24.04-nccl2.30.4-1-torch2.11.0-vision0.26.0-audio2.11.0-abi1'
7+
tag-suffix: 'v0.20.2-cuda13.2.1-ubuntu24.04'
8+
- vllm-commit: 'v0.20.2'
9+
flashinfer-commit: 'v0.6.8'
10+
lmcache-commit: 'v0.4.2'
11+
builder-base-image: 'ghcr.io/coreweave/ml-containers/torch:bc8c66e-nccl-cuda12.9.1-ubuntu24.04-nccl2.30.4-1-torch2.11.0-vision0.26.0-audio2.11.0-abi1'
12+
final-base-image: 'ghcr.io/coreweave/ml-containers/torch:bc8c66e-nccl-cuda12.9.1-ubuntu24.04-nccl2.30.4-1-torch2.11.0-vision0.26.0-audio2.11.0-abi1'
13+
tag-suffix: 'v0.20.2-cuda12.9.1-ubuntu24.04'

.github/workflows/vllm-tensorizer.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ jobs:
2121
with:
2222
image-name: vllm-tensorizer
2323
folder: vllm-tensorizer
24-
tag-suffix: ${{ matrix.vllm-commit }}
24+
tag-suffix: ${{ matrix.tag-suffix }}
2525
build-contexts: |
2626
common=common
2727
object-storage-secrets: true

0 commit comments

Comments
 (0)