From d9e9aa851b98852e6ca917833e00fb537aab4c31 Mon Sep 17 00:00:00 2001 From: Evan Smothers Date: Tue, 7 Oct 2025 15:33:42 -0700 Subject: [PATCH 1/2] Build Monarch with CUDA 12.8 instead of 12.9 --- .github/workflows/build_wheels.yaml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/build_wheels.yaml b/.github/workflows/build_wheels.yaml index a384991cd..4088cc76d 100644 --- a/.github/workflows/build_wheels.yaml +++ b/.github/workflows/build_wheels.yaml @@ -31,11 +31,11 @@ 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", "channel": "nightly", From a618fd30efd373d4134470991716c9df6c532529 Mon Sep 17 00:00:00 2001 From: Evan Smothers Date: Tue, 7 Oct 2025 15:55:31 -0700 Subject: [PATCH 2/2] remaining 12.9 -> 12.8 updates --- .github/workflows/build_vllm.yaml | 12 ++++++------ .github/workflows/build_wheels.yaml | 4 ++-- 2 files changed, 8 insertions(+), 8 deletions(-) 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 4088cc76d..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 @@ -37,7 +37,7 @@ jobs: "package_type": "manywheel", "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",