Update build scripts

Author: joecummings

.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,16 +31,16 @@ jobs:
             {
               "python_version": "3.10",
               "gpu_arch_type": "cpu",
-              "gpu_arch_version": "12.9",
-              "desired_cuda": "cu129",
+              "gpu_arch_version": "12.8",
+              "desired_cuda": "cu128",
               "container_image": "pytorch/manylinux2_28-builder:cuda12.9",
               "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",
+              "stable_version": "2.9.0",
               "use_split_build": false
             }
           ]

.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,16 +31,16 @@ 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",
+              "stable_version": "2.9.0",
               "use_split_build": false
             }
           ]

.github/workflows/gpu_test.yaml

@@ -41,14 +41,14 @@ jobs:
       - name: Update pip
         run: python -m pip install --upgrade pip
       - name: Install pinned torch nightly
-        run: python -m pip install --pre torch==2.9.0.dev20250905 --no-cache-dir --index-url https://download.pytorch.org/whl/nightly/cu129
+        run: python -m pip install torch==2.9.0 --no-cache-dir --index-url https://download.pytorch.org/whl/cu128
       - name: Download and install vLLM and its dependencies
         # TODO: this honestly could not be hackier if I tried
         run: |
           python -m pip install -r .github/packaging/vllm_reqs.txt
-          python -m pip install vllm==0.10.1.dev0+g6d8d0a24c.d20251009.cu129 --no-cache-dir --index-url https://download.pytorch.org/whl/preview/forge
+          python -m pip install vllm==0.10.1.dev0+g6d8d0a24c.d20251009.cu128 --no-cache-dir --index-url https://download.pytorch.org/whl/preview/forge
       - name: Install Monarch
-        run: pip install torchmonarch==0.1.0rc1
+        run: pip install torchmonarch==0.1.0rc4
       - name: Install torchtitan and torchstore
         run: |
           python -m pip install git+https://github.com/pytorch/torchtitan.git