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",