diff --git a/.github/workflows/build_vllm.yaml b/.github/workflows/build_vllm.yaml index 0e8279ac4..ca5246418 100644 --- a/.github/workflows/build_vllm.yaml +++ b/.github/workflows/build_vllm.yaml @@ -12,7 +12,7 @@ permissions: jobs: build: - name: forge-cu129-nightly + name: forge-cu128-nightly uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main strategy: fail-fast: false @@ -31,13 +31,13 @@ jobs: { "python_version": "3.10", "gpu_arch_type": "cpu", - "gpu_arch_version": "12.9", - "desired_cuda": "cu129", - "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "gpu_arch_version": "12.8", + "desired_cuda": "cu128", + "container_image": "pytorch/manylinux2_28-builder:cuda12.8", "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda12_9", + "build_name": "manywheel-py3_10-cuda12_8", "validation_runner": "linux.12xlarge.memory", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.8.0", diff --git a/.github/workflows/build_wheels.yaml b/.github/workflows/build_wheels.yaml index a384991cd..3f337e9f4 100644 --- a/.github/workflows/build_wheels.yaml +++ b/.github/workflows/build_wheels.yaml @@ -12,7 +12,7 @@ permissions: jobs: build: - name: forge-cu129-nightly + name: forge-cu128-nightly uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main strategy: fail-fast: false @@ -31,13 +31,13 @@ jobs: { "python_version": "3.10", "gpu_arch_type": "cuda", - "gpu_arch_version": "12.9", - "desired_cuda": "cu129", - "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "gpu_arch_version": "12.8", + "desired_cuda": "cu128", + "container_image": "pytorch/manylinux2_28-builder:cuda12.8", "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda12_9", + "build_name": "manywheel-py3_10-cuda12_8", "validation_runner": "linux.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.8.0",