1# @generated DO NOT EDIT MANUALLY 2 3# Template is at: .github/templates/linux_binary_build_workflow.yml.j2 4# Generation script: .github/scripts/generate_ci_workflows.py 5name: linux-binary-manywheel 6 7 8on: 9 push: 10 branches: 11 - main 12 tags: 13 - 'ciflow/trunk/*' 14 workflow_dispatch: 15 16env: 17 # Needed for conda builds 18 ALPINE_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/alpine" 19 ANACONDA_USER: pytorch 20 AWS_DEFAULT_REGION: us-east-1 21 BINARY_ENV_FILE: /tmp/env 22 BUILD_ENVIRONMENT: linux-binary-manywheel 23 BUILDER_ROOT: /builder 24 GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} 25 PR_NUMBER: ${{ github.event.pull_request.number }} 26 PYTORCH_FINAL_PACKAGE_DIR: /artifacts 27 PYTORCH_ROOT: /pytorch 28 SHA1: ${{ github.event.pull_request.head.sha || github.sha }} 29 SKIP_ALL_TESTS: 0 30concurrency: 31 group: linux-binary-manywheel-${{ github.event.pull_request.number || github.ref_name }}-${{ github.ref_type == 'branch' && github.sha }}-${{ github.event_name == 'workflow_dispatch' }} 32 cancel-in-progress: true 33 34jobs: 35 manywheel-py3_8-cuda11_8-build: 36 if: ${{ github.repository_owner == 'pytorch' }} 37 uses: ./.github/workflows/_binary-build-linux.yml 38 with: 39 PYTORCH_ROOT: /pytorch 40 BUILDER_ROOT: /builder 41 PACKAGE_TYPE: manywheel 42 # TODO: This is a legacy variable that we eventually want to get rid of in 43 # favor of GPU_ARCH_VERSION 44 DESIRED_CUDA: cu118 45 GPU_ARCH_VERSION: 11.8 46 GPU_ARCH_TYPE: cuda 47 DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4 48 DESIRED_PYTHON: "3.8" 49 build_name: manywheel-py3_8-cuda11_8 50 build_environment: linux-binary-manywheel 51 PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu11==11.8.89; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-runtime-cu11==11.8.89; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-cupti-cu11==11.8.87; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cudnn-cu11==9.1.0.70; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cublas-cu11==11.11.3.6; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufft-cu11==10.9.0.58; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-curand-cu11==10.3.0.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusolver-cu11==11.4.1.48; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparse-cu11==11.7.5.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nccl-cu11==2.20.5; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvtx-cu11==11.8.86; platform_system == 'Linux' and platform_machine == 'x86_64' 52 secrets: 53 github-token: ${{ secrets.GITHUB_TOKEN }} 54 manywheel-py3_8-cuda11_8-test: # Testing 55 if: ${{ github.repository_owner == 'pytorch' }} 56 needs: manywheel-py3_8-cuda11_8-build 57 uses: ./.github/workflows/_binary-test-linux.yml 58 with: 59 PYTORCH_ROOT: /pytorch 60 BUILDER_ROOT: /builder 61 PACKAGE_TYPE: manywheel 62 # TODO: This is a legacy variable that we eventually want to get rid of in 63 # favor of GPU_ARCH_VERSION 64 DESIRED_CUDA: cu118 65 GPU_ARCH_VERSION: 11.8 66 GPU_ARCH_TYPE: cuda 67 DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4 68 DESIRED_PYTHON: "3.8" 69 build_name: manywheel-py3_8-cuda11_8 70 build_environment: linux-binary-manywheel 71 runs_on: linux.4xlarge.nvidia.gpu 72 secrets: 73 github-token: ${{ secrets.GITHUB_TOKEN }} 74 75 manywheel-py3_8-cuda12_1-build: 76 if: ${{ github.repository_owner == 'pytorch' }} 77 uses: ./.github/workflows/_binary-build-linux.yml 78 with: 79 PYTORCH_ROOT: /pytorch 80 BUILDER_ROOT: /builder 81 PACKAGE_TYPE: manywheel 82 # TODO: This is a legacy variable that we eventually want to get rid of in 83 # favor of GPU_ARCH_VERSION 84 DESIRED_CUDA: cu121 85 GPU_ARCH_VERSION: 12.1 86 GPU_ARCH_TYPE: cuda 87 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4 88 DESIRED_PYTHON: "3.8" 89 build_name: manywheel-py3_8-cuda12_1 90 build_environment: linux-binary-manywheel 91 PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-runtime-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-cupti-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cudnn-cu12==9.1.0.70; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cublas-cu12==12.1.3.1; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufft-cu12==11.0.2.54; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-curand-cu12==10.3.2.106; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusolver-cu12==11.4.5.107; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparse-cu12==12.1.0.106; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nccl-cu12==2.20.5; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvtx-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' 92 secrets: 93 github-token: ${{ secrets.GITHUB_TOKEN }} 94 manywheel-py3_8-cuda12_1-test: # Testing 95 if: ${{ github.repository_owner == 'pytorch' }} 96 needs: manywheel-py3_8-cuda12_1-build 97 uses: ./.github/workflows/_binary-test-linux.yml 98 with: 99 PYTORCH_ROOT: /pytorch 100 BUILDER_ROOT: /builder 101 PACKAGE_TYPE: manywheel 102 # TODO: This is a legacy variable that we eventually want to get rid of in 103 # favor of GPU_ARCH_VERSION 104 DESIRED_CUDA: cu121 105 GPU_ARCH_VERSION: 12.1 106 GPU_ARCH_TYPE: cuda 107 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4 108 DESIRED_PYTHON: "3.8" 109 build_name: manywheel-py3_8-cuda12_1 110 build_environment: linux-binary-manywheel 111 runs_on: linux.4xlarge.nvidia.gpu 112 secrets: 113 github-token: ${{ secrets.GITHUB_TOKEN }} 114 115 manywheel-py3_8-cuda12_4-build: 116 if: ${{ github.repository_owner == 'pytorch' }} 117 uses: ./.github/workflows/_binary-build-linux.yml 118 with: 119 PYTORCH_ROOT: /pytorch 120 BUILDER_ROOT: /builder 121 PACKAGE_TYPE: manywheel 122 # TODO: This is a legacy variable that we eventually want to get rid of in 123 # favor of GPU_ARCH_VERSION 124 DESIRED_CUDA: cu124 125 GPU_ARCH_VERSION: 12.4 126 GPU_ARCH_TYPE: cuda 127 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4 128 DESIRED_PYTHON: "3.8" 129 build_name: manywheel-py3_8-cuda12_4 130 build_environment: linux-binary-manywheel 131 PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-runtime-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-cupti-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cudnn-cu12==9.1.0.70; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cublas-cu12==12.4.2.65; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufft-cu12==11.2.0.44; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-curand-cu12==10.3.5.119; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusolver-cu12==11.6.0.99; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparse-cu12==12.3.0.142; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nccl-cu12==2.20.5; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvtx-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvjitlink-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' 132 secrets: 133 github-token: ${{ secrets.GITHUB_TOKEN }} 134 manywheel-py3_8-cuda12_4-test: # Testing 135 if: ${{ github.repository_owner == 'pytorch' }} 136 needs: manywheel-py3_8-cuda12_4-build 137 uses: ./.github/workflows/_binary-test-linux.yml 138 with: 139 PYTORCH_ROOT: /pytorch 140 BUILDER_ROOT: /builder 141 PACKAGE_TYPE: manywheel 142 # TODO: This is a legacy variable that we eventually want to get rid of in 143 # favor of GPU_ARCH_VERSION 144 DESIRED_CUDA: cu124 145 GPU_ARCH_VERSION: 12.4 146 GPU_ARCH_TYPE: cuda 147 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4 148 DESIRED_PYTHON: "3.8" 149 build_name: manywheel-py3_8-cuda12_4 150 build_environment: linux-binary-manywheel 151 runs_on: linux.4xlarge.nvidia.gpu 152 secrets: 153 github-token: ${{ secrets.GITHUB_TOKEN }} 154