@@ -12,15 +12,15 @@ permissions:
1212
1313jobs :
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