From ee14e8c70ae93535782eb0310b6bd1faf3a87687 Mon Sep 17 00:00:00 2001 From: Evan Smothers Date: Wed, 8 Oct 2025 16:48:58 -0700 Subject: [PATCH 1/3] revert 12.6, rename monarch wheel --- .github/packaging/pre_build_gpu.sh | 1 + .github/workflows/build_vllm.yaml | 12 ++++++------ .github/workflows/build_wheels.yaml | 12 ++++++------ 3 files changed, 13 insertions(+), 12 deletions(-) diff --git a/.github/packaging/pre_build_gpu.sh b/.github/packaging/pre_build_gpu.sh index d81f52782..9a7dd653b 100644 --- a/.github/packaging/pre_build_gpu.sh +++ b/.github/packaging/pre_build_gpu.sh @@ -16,6 +16,7 @@ echo "build dir is $BUILD_DIR" echo "wheel dir is $WHL_DIR" build_monarch() { + export MONARCH_PACKAGE_NAME="torchmonarch" # Get Rust build related pieces if ! command -v rustup &> /dev/null; then echo "getting rustup" diff --git a/.github/workflows/build_vllm.yaml b/.github/workflows/build_vllm.yaml index cba29257d..0e8279ac4 100644 --- a/.github/workflows/build_vllm.yaml +++ b/.github/workflows/build_vllm.yaml @@ -12,7 +12,7 @@ permissions: jobs: build: - name: forge-cu126-nightly + name: forge-cu129-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.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda12_6", + "build_name": "manywheel-py3_10-cuda12_9", "validation_runner": "linux.12xlarge.memory", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "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 087d65da8..a384991cd 100644 --- a/.github/workflows/build_wheels.yaml +++ b/.github/workflows/build_wheels.yaml @@ -12,7 +12,7 @@ permissions: jobs: build: - name: forge-cu126-nightly + name: forge-cu129-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.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda12_6", + "build_name": "manywheel-py3_10-cuda12_9", "validation_runner": "linux.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.8.0", From 76ebfd213697534c3b62ef6a42186986a9d9734b Mon Sep 17 00:00:00 2001 From: Evan Smothers Date: Wed, 8 Oct 2025 16:50:00 -0700 Subject: [PATCH 2/3] testing --- .github/workflows/build_vllm.yaml | 1 + .github/workflows/build_wheels.yaml | 1 + 2 files changed, 2 insertions(+) diff --git a/.github/workflows/build_vllm.yaml b/.github/workflows/build_vllm.yaml index 0e8279ac4..4b6c59d2b 100644 --- a/.github/workflows/build_vllm.yaml +++ b/.github/workflows/build_vllm.yaml @@ -1,6 +1,7 @@ name: Build pinned vLLM against PyTorch nightly and upload on: + pull_request: push: branches: - nightly diff --git a/.github/workflows/build_wheels.yaml b/.github/workflows/build_wheels.yaml index a384991cd..40a07ab1e 100644 --- a/.github/workflows/build_wheels.yaml +++ b/.github/workflows/build_wheels.yaml @@ -1,6 +1,7 @@ name: Build nightly wheels and publish to PyTorch Index on: + pull_request: push: branches: - nightly From 9fa8b4fc68988ad6a66fd8cc94141275f6df6156 Mon Sep 17 00:00:00 2001 From: Evan Smothers Date: Wed, 8 Oct 2025 19:52:39 -0700 Subject: [PATCH 3/3] revert testing changes --- .github/workflows/build_vllm.yaml | 1 - .github/workflows/build_wheels.yaml | 1 - 2 files changed, 2 deletions(-) diff --git a/.github/workflows/build_vllm.yaml b/.github/workflows/build_vllm.yaml index 4b6c59d2b..0e8279ac4 100644 --- a/.github/workflows/build_vllm.yaml +++ b/.github/workflows/build_vllm.yaml @@ -1,7 +1,6 @@ name: Build pinned vLLM against PyTorch nightly and upload on: - pull_request: push: branches: - nightly diff --git a/.github/workflows/build_wheels.yaml b/.github/workflows/build_wheels.yaml index 40a07ab1e..a384991cd 100644 --- a/.github/workflows/build_wheels.yaml +++ b/.github/workflows/build_wheels.yaml @@ -1,7 +1,6 @@ name: Build nightly wheels and publish to PyTorch Index on: - pull_request: push: branches: - nightly