Skip to content

Commit 3abb3e8

Browse files
committed
enable vLLM upload with CUDA 12.8 build
1 parent 2559be6 commit 3abb3e8

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

.github/workflows/build_vllm.yaml

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -12,15 +12,15 @@ permissions:
1212

1313
jobs:
1414
build:
15-
name: forge-cu129-nightly
16-
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
15+
name: forge-cu128-nightly
16+
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@vllm-push
1717
strategy:
1818
fail-fast: false
1919
with:
2020
repository: meta-pytorch/forge
2121
ref: ""
2222
test-infra-repository: pytorch/test-infra
23-
test-infra-ref: main
23+
test-infra-ref: vllm-push
2424
run-smoke-test: false
2525
wheel-nightly-policy: gha_workflow_preview_build_wheels
2626
wheel-upload-path: whl/preview/forge/
@@ -31,13 +31,13 @@ jobs:
3131
{
3232
"python_version": "3.10",
3333
"gpu_arch_type": "cpu",
34-
"gpu_arch_version": "12.9",
35-
"desired_cuda": "cu129",
36-
"container_image": "pytorch/manylinux2_28-builder:cuda12.9",
34+
"gpu_arch_version": "12.8",
35+
"desired_cuda": "cu128",
36+
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
3737
"package_type": "manywheel",
38-
"build_name": "manywheel-py3_10-cuda12_9",
38+
"build_name": "manywheel-py3_10-cuda12_8",
3939
"validation_runner": "linux.12xlarge.memory",
40-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129",
40+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128",
4141
"channel": "nightly",
4242
"upload_to_base_bucket": "no",
4343
"stable_version": "2.8.0",

0 commit comments

Comments
 (0)