diff --git a/.github/workflows/build_vllm.yaml b/.github/workflows/build_vllm.yaml index beb90730c..442a3739d 100644 --- a/.github/workflows/build_vllm.yaml +++ b/.github/workflows/build_vllm.yaml @@ -36,14 +36,44 @@ jobs: "desired_cuda": "cu128", "container_image": "pytorch/manylinux2_28-builder:cuda12.8", "package_type": "manywheel", - "build_name": "manywheel-cuda12_8", + "build_name": "manywheel-py3_10-cuda12_8", "validation_runner": "linux.12xlarge.memory", "installation": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128", - "channel": "nightly", + "channel": "test", "upload_to_base_bucket": "no", "stable_version": "2.9.0", "use_split_build": false - } + }, + { + "python_version": "3.11", + "gpu_arch_type": "cpu", + "gpu_arch_version": "12.8", + "desired_cuda": "cu128", + "container_image": "pytorch/manylinux2_28-builder:cuda12.8", + "package_type": "manywheel", + "build_name": "manywheel-py3_11-cuda12_8", + "validation_runner": "linux.12xlarge.memory", + "installation": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128", + "channel": "test", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0", + "use_split_build": false + }, + { + "python_version": "3.12", + "gpu_arch_type": "cpu", + "gpu_arch_version": "12.8", + "desired_cuda": "cu128", + "container_image": "pytorch/manylinux2_28-builder:cuda12.8", + "package_type": "manywheel", + "build_name": "manywheel-py3_12-cuda12_8", + "validation_runner": "linux.12xlarge.memory", + "installation": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128", + "channel": "test", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0", + "use_split_build": false + }, ] } pre-script: .github/packaging/pre_build_cpu.sh