Skip to content
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .github/packaging/pre_build_gpu.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down
13 changes: 7 additions & 6 deletions .github/workflows/build_vllm.yaml
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
name: Build pinned vLLM against PyTorch nightly and upload

on:
pull_request:
push:
branches:
- nightly
Expand All @@ -12,7 +13,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
Expand All @@ -31,13 +32,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",
Expand Down
13 changes: 7 additions & 6 deletions .github/workflows/build_wheels.yaml
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
name: Build nightly wheels and publish to PyTorch Index

on:
pull_request:
push:
branches:
- nightly
Expand All @@ -12,7 +13,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
Expand All @@ -31,13 +32,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",
Expand Down
Loading