| # @generated DO NOT EDIT MANUALLY |
| |
| # Template is at: .github/templates/linux_binary_build_workflow.yml.j2 |
| # Generation script: .github/scripts/generate_ci_workflows.py |
| name: linux-binary-libtorch-pre-cxx11 |
| |
| |
| on: |
| push: |
| # NOTE: Meta Employees can trigger new nightlies using: https://fburl.com/trigger_pytorch_nightly_build |
| branches: |
| - nightly |
| tags: |
| # NOTE: Binary build pipelines should only get triggered on release candidate builds |
| # Release candidate tags look like: v1.11.0-rc1 |
| - v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+ |
| - 'ciflow/binaries/*' |
| - 'ciflow/binaries_libtorch/*' |
| workflow_dispatch: |
| |
| env: |
| # Needed for conda builds |
| ALPINE_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/alpine" |
| ANACONDA_USER: pytorch |
| AWS_DEFAULT_REGION: us-east-1 |
| BINARY_ENV_FILE: /tmp/env |
| BUILD_ENVIRONMENT: linux-binary-libtorch-pre-cxx11 |
| BUILDER_ROOT: /builder |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| PR_NUMBER: ${{ github.event.pull_request.number }} |
| PYTORCH_FINAL_PACKAGE_DIR: /artifacts |
| PYTORCH_ROOT: /pytorch |
| SHA1: ${{ github.event.pull_request.head.sha || github.sha }} |
| SKIP_ALL_TESTS: 0 |
| concurrency: |
| group: linux-binary-libtorch-pre-cxx11-${{ github.event.pull_request.number || github.ref_name }}-${{ github.ref_type == 'branch' && github.sha }}-${{ github.event_name == 'workflow_dispatch' }} |
| cancel-in-progress: true |
| |
| jobs: |
| get-label-type: |
| name: get-label-type |
| uses: ./.github/workflows/_runner-determinator.yml |
| with: |
| triggering_actor: ${{ github.triggering_actor }} |
| issue_owner: ${{ github.event.pull_request.user.login || github.event.issue.user.login }} |
| curr_branch: ${{ github.head_ref || github.ref_name }} |
| curr_ref_type: ${{ github.ref_type }} |
| libtorch-cpu-shared-with-deps-pre-cxx11-build: |
| if: ${{ github.repository_owner == 'pytorch' }} |
| uses: ./.github/workflows/_binary-build-linux.yml |
| needs: get-label-type |
| with: |
| PYTORCH_ROOT: /pytorch |
| BUILDER_ROOT: /builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cpu |
| GPU_ARCH_TYPE: cpu |
| DOCKER_IMAGE: pytorch/manylinux-builder:cpu-main |
| LIBTORCH_VARIANT: shared-with-deps |
| DESIRED_DEVTOOLSET: pre-cxx11 |
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" |
| build_name: libtorch-cpu-shared-with-deps-pre-cxx11 |
| build_environment: linux-binary-libtorch-pre-cxx11 |
| secrets: |
| github-token: ${{ secrets.GITHUB_TOKEN }} |
| libtorch-cpu-shared-with-deps-pre-cxx11-test: # Testing |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: |
| - libtorch-cpu-shared-with-deps-pre-cxx11-build |
| - get-label-type |
| uses: ./.github/workflows/_binary-test-linux.yml |
| with: |
| PYTORCH_ROOT: /pytorch |
| BUILDER_ROOT: /builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cpu |
| GPU_ARCH_TYPE: cpu |
| DOCKER_IMAGE: pytorch/manylinux-builder:cpu-main |
| LIBTORCH_VARIANT: shared-with-deps |
| DESIRED_DEVTOOLSET: pre-cxx11 |
| build_name: libtorch-cpu-shared-with-deps-pre-cxx11 |
| build_environment: linux-binary-libtorch-pre-cxx11 |
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" |
| runs_on: linux.4xlarge |
| secrets: |
| github-token: ${{ secrets.GITHUB_TOKEN }} |
| libtorch-cpu-shared-with-deps-pre-cxx11-upload: # Uploading |
| if: ${{ github.repository_owner == 'pytorch' }} |
| permissions: |
| id-token: write |
| contents: read |
| needs: libtorch-cpu-shared-with-deps-pre-cxx11-test |
| with: |
| PYTORCH_ROOT: /pytorch |
| BUILDER_ROOT: /builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cpu |
| GPU_ARCH_TYPE: cpu |
| DOCKER_IMAGE: pytorch/manylinux-builder:cpu-main |
| LIBTORCH_VARIANT: shared-with-deps |
| DESIRED_DEVTOOLSET: pre-cxx11 |
| build_name: libtorch-cpu-shared-with-deps-pre-cxx11 |
| secrets: |
| github-token: ${{ secrets.GITHUB_TOKEN }} |
| conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} |
| conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} |
| uses: ./.github/workflows/_binary-upload.yml |
| |
| libtorch-cuda11_8-shared-with-deps-pre-cxx11-build: |
| if: ${{ github.repository_owner == 'pytorch' }} |
| uses: ./.github/workflows/_binary-build-linux.yml |
| needs: get-label-type |
| with: |
| PYTORCH_ROOT: /pytorch |
| BUILDER_ROOT: /builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu118 |
| GPU_ARCH_VERSION: 11.8 |
| GPU_ARCH_TYPE: cuda |
| DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-main |
| LIBTORCH_VARIANT: shared-with-deps |
| DESIRED_DEVTOOLSET: pre-cxx11 |
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" |
| build_name: libtorch-cuda11_8-shared-with-deps-pre-cxx11 |
| build_environment: linux-binary-libtorch-pre-cxx11 |
| secrets: |
| github-token: ${{ secrets.GITHUB_TOKEN }} |
| libtorch-cuda11_8-shared-with-deps-pre-cxx11-test: # Testing |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: |
| - libtorch-cuda11_8-shared-with-deps-pre-cxx11-build |
| - get-label-type |
| uses: ./.github/workflows/_binary-test-linux.yml |
| with: |
| PYTORCH_ROOT: /pytorch |
| BUILDER_ROOT: /builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu118 |
| GPU_ARCH_VERSION: 11.8 |
| GPU_ARCH_TYPE: cuda |
| DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-main |
| LIBTORCH_VARIANT: shared-with-deps |
| DESIRED_DEVTOOLSET: pre-cxx11 |
| build_name: libtorch-cuda11_8-shared-with-deps-pre-cxx11 |
| build_environment: linux-binary-libtorch-pre-cxx11 |
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" |
| runs_on: linux.4xlarge.nvidia.gpu |
| secrets: |
| github-token: ${{ secrets.GITHUB_TOKEN }} |
| libtorch-cuda11_8-shared-with-deps-pre-cxx11-upload: # Uploading |
| if: ${{ github.repository_owner == 'pytorch' }} |
| permissions: |
| id-token: write |
| contents: read |
| needs: libtorch-cuda11_8-shared-with-deps-pre-cxx11-test |
| with: |
| PYTORCH_ROOT: /pytorch |
| BUILDER_ROOT: /builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu118 |
| GPU_ARCH_VERSION: 11.8 |
| GPU_ARCH_TYPE: cuda |
| DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-main |
| LIBTORCH_VARIANT: shared-with-deps |
| DESIRED_DEVTOOLSET: pre-cxx11 |
| build_name: libtorch-cuda11_8-shared-with-deps-pre-cxx11 |
| secrets: |
| github-token: ${{ secrets.GITHUB_TOKEN }} |
| conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} |
| conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} |
| uses: ./.github/workflows/_binary-upload.yml |
| |
| libtorch-cuda12_1-shared-with-deps-pre-cxx11-build: |
| if: ${{ github.repository_owner == 'pytorch' }} |
| uses: ./.github/workflows/_binary-build-linux.yml |
| needs: get-label-type |
| with: |
| PYTORCH_ROOT: /pytorch |
| BUILDER_ROOT: /builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu121 |
| GPU_ARCH_VERSION: 12.1 |
| GPU_ARCH_TYPE: cuda |
| DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-main |
| LIBTORCH_VARIANT: shared-with-deps |
| DESIRED_DEVTOOLSET: pre-cxx11 |
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" |
| build_name: libtorch-cuda12_1-shared-with-deps-pre-cxx11 |
| build_environment: linux-binary-libtorch-pre-cxx11 |
| secrets: |
| github-token: ${{ secrets.GITHUB_TOKEN }} |
| libtorch-cuda12_1-shared-with-deps-pre-cxx11-test: # Testing |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: |
| - libtorch-cuda12_1-shared-with-deps-pre-cxx11-build |
| - get-label-type |
| uses: ./.github/workflows/_binary-test-linux.yml |
| with: |
| PYTORCH_ROOT: /pytorch |
| BUILDER_ROOT: /builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu121 |
| GPU_ARCH_VERSION: 12.1 |
| GPU_ARCH_TYPE: cuda |
| DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-main |
| LIBTORCH_VARIANT: shared-with-deps |
| DESIRED_DEVTOOLSET: pre-cxx11 |
| build_name: libtorch-cuda12_1-shared-with-deps-pre-cxx11 |
| build_environment: linux-binary-libtorch-pre-cxx11 |
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" |
| runs_on: linux.4xlarge.nvidia.gpu |
| secrets: |
| github-token: ${{ secrets.GITHUB_TOKEN }} |
| libtorch-cuda12_1-shared-with-deps-pre-cxx11-upload: # Uploading |
| if: ${{ github.repository_owner == 'pytorch' }} |
| permissions: |
| id-token: write |
| contents: read |
| needs: libtorch-cuda12_1-shared-with-deps-pre-cxx11-test |
| with: |
| PYTORCH_ROOT: /pytorch |
| BUILDER_ROOT: /builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu121 |
| GPU_ARCH_VERSION: 12.1 |
| GPU_ARCH_TYPE: cuda |
| DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-main |
| LIBTORCH_VARIANT: shared-with-deps |
| DESIRED_DEVTOOLSET: pre-cxx11 |
| build_name: libtorch-cuda12_1-shared-with-deps-pre-cxx11 |
| secrets: |
| github-token: ${{ secrets.GITHUB_TOKEN }} |
| conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} |
| conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} |
| uses: ./.github/workflows/_binary-upload.yml |
| |
| libtorch-cuda12_4-shared-with-deps-pre-cxx11-build: |
| if: ${{ github.repository_owner == 'pytorch' }} |
| uses: ./.github/workflows/_binary-build-linux.yml |
| needs: get-label-type |
| with: |
| PYTORCH_ROOT: /pytorch |
| BUILDER_ROOT: /builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu124 |
| GPU_ARCH_VERSION: 12.4 |
| GPU_ARCH_TYPE: cuda |
| DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-main |
| LIBTORCH_VARIANT: shared-with-deps |
| DESIRED_DEVTOOLSET: pre-cxx11 |
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" |
| build_name: libtorch-cuda12_4-shared-with-deps-pre-cxx11 |
| build_environment: linux-binary-libtorch-pre-cxx11 |
| secrets: |
| github-token: ${{ secrets.GITHUB_TOKEN }} |
| libtorch-cuda12_4-shared-with-deps-pre-cxx11-test: # Testing |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: |
| - libtorch-cuda12_4-shared-with-deps-pre-cxx11-build |
| - get-label-type |
| uses: ./.github/workflows/_binary-test-linux.yml |
| with: |
| PYTORCH_ROOT: /pytorch |
| BUILDER_ROOT: /builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu124 |
| GPU_ARCH_VERSION: 12.4 |
| GPU_ARCH_TYPE: cuda |
| DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-main |
| LIBTORCH_VARIANT: shared-with-deps |
| DESIRED_DEVTOOLSET: pre-cxx11 |
| build_name: libtorch-cuda12_4-shared-with-deps-pre-cxx11 |
| build_environment: linux-binary-libtorch-pre-cxx11 |
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" |
| runs_on: linux.4xlarge.nvidia.gpu |
| secrets: |
| github-token: ${{ secrets.GITHUB_TOKEN }} |
| libtorch-cuda12_4-shared-with-deps-pre-cxx11-upload: # Uploading |
| if: ${{ github.repository_owner == 'pytorch' }} |
| permissions: |
| id-token: write |
| contents: read |
| needs: libtorch-cuda12_4-shared-with-deps-pre-cxx11-test |
| with: |
| PYTORCH_ROOT: /pytorch |
| BUILDER_ROOT: /builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu124 |
| GPU_ARCH_VERSION: 12.4 |
| GPU_ARCH_TYPE: cuda |
| DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-main |
| LIBTORCH_VARIANT: shared-with-deps |
| DESIRED_DEVTOOLSET: pre-cxx11 |
| build_name: libtorch-cuda12_4-shared-with-deps-pre-cxx11 |
| secrets: |
| github-token: ${{ secrets.GITHUB_TOKEN }} |
| conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} |
| conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} |
| uses: ./.github/workflows/_binary-upload.yml |
| |
| libtorch-rocm6_1-shared-with-deps-pre-cxx11-build: |
| if: ${{ github.repository_owner == 'pytorch' }} |
| uses: ./.github/workflows/_binary-build-linux.yml |
| needs: get-label-type |
| with: |
| PYTORCH_ROOT: /pytorch |
| BUILDER_ROOT: /builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: rocm6.1 |
| GPU_ARCH_VERSION: 6.1 |
| GPU_ARCH_TYPE: rocm |
| DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-main |
| LIBTORCH_VARIANT: shared-with-deps |
| DESIRED_DEVTOOLSET: pre-cxx11 |
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" |
| build_name: libtorch-rocm6_1-shared-with-deps-pre-cxx11 |
| build_environment: linux-binary-libtorch-pre-cxx11 |
| secrets: |
| github-token: ${{ secrets.GITHUB_TOKEN }} |
| libtorch-rocm6_1-shared-with-deps-pre-cxx11-test: # Testing |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: |
| - libtorch-rocm6_1-shared-with-deps-pre-cxx11-build |
| - get-label-type |
| runs-on: linux.rocm.gpu |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: /pytorch |
| BUILDER_ROOT: /builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: rocm6.1 |
| GPU_ARCH_VERSION: 6.1 |
| GPU_ARCH_TYPE: rocm |
| SKIP_ALL_TESTS: 1 |
| DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-main |
| LIBTORCH_VARIANT: shared-with-deps |
| DESIRED_DEVTOOLSET: pre-cxx11 |
| steps: |
| - name: Setup ROCm |
| uses: ./.github/actions/setup-rocm |
| - uses: actions/download-artifact@v3 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-rocm6_1-shared-with-deps-pre-cxx11 |
| path: "${{ runner.temp }}/artifacts/" |
| - name: Checkout PyTorch |
| uses: malfet/checkout@silent-checkout |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| quiet-checkout: true |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: malfet/checkout@silent-checkout |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| quiet-checkout: true |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: ROCm set GPU_FLAG |
| run: | |
| echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add daemon" >> "${GITHUB_ENV}" |
| - name: Pull Docker image |
| uses: pytorch/test-infra/.github/actions/pull-docker-image@main |
| with: |
| docker-image: pytorch/manylinux-builder:rocm6.1-main |
| - name: Test Pytorch binary |
| uses: ./pytorch/.github/actions/test-pytorch-binary |
| - name: Teardown ROCm |
| uses: ./.github/actions/teardown-rocm |
| libtorch-rocm6_1-shared-with-deps-pre-cxx11-upload: # Uploading |
| if: ${{ github.repository_owner == 'pytorch' }} |
| permissions: |
| id-token: write |
| contents: read |
| needs: libtorch-rocm6_1-shared-with-deps-pre-cxx11-test |
| with: |
| PYTORCH_ROOT: /pytorch |
| BUILDER_ROOT: /builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: rocm6.1 |
| GPU_ARCH_VERSION: 6.1 |
| GPU_ARCH_TYPE: rocm |
| DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-main |
| LIBTORCH_VARIANT: shared-with-deps |
| DESIRED_DEVTOOLSET: pre-cxx11 |
| build_name: libtorch-rocm6_1-shared-with-deps-pre-cxx11 |
| secrets: |
| github-token: ${{ secrets.GITHUB_TOKEN }} |
| conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} |
| conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} |
| uses: ./.github/workflows/_binary-upload.yml |
| |
| libtorch-rocm6_2-shared-with-deps-pre-cxx11-build: |
| if: ${{ github.repository_owner == 'pytorch' }} |
| uses: ./.github/workflows/_binary-build-linux.yml |
| needs: get-label-type |
| with: |
| PYTORCH_ROOT: /pytorch |
| BUILDER_ROOT: /builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: rocm6.2 |
| GPU_ARCH_VERSION: 6.2 |
| GPU_ARCH_TYPE: rocm |
| DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.2-main |
| LIBTORCH_VARIANT: shared-with-deps |
| DESIRED_DEVTOOLSET: pre-cxx11 |
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" |
| build_name: libtorch-rocm6_2-shared-with-deps-pre-cxx11 |
| build_environment: linux-binary-libtorch-pre-cxx11 |
| secrets: |
| github-token: ${{ secrets.GITHUB_TOKEN }} |
| libtorch-rocm6_2-shared-with-deps-pre-cxx11-test: # Testing |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: |
| - libtorch-rocm6_2-shared-with-deps-pre-cxx11-build |
| - get-label-type |
| runs-on: linux.rocm.gpu |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: /pytorch |
| BUILDER_ROOT: /builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: rocm6.2 |
| GPU_ARCH_VERSION: 6.2 |
| GPU_ARCH_TYPE: rocm |
| SKIP_ALL_TESTS: 1 |
| DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.2-main |
| LIBTORCH_VARIANT: shared-with-deps |
| DESIRED_DEVTOOLSET: pre-cxx11 |
| steps: |
| - name: Setup ROCm |
| uses: ./.github/actions/setup-rocm |
| - uses: actions/download-artifact@v3 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-rocm6_2-shared-with-deps-pre-cxx11 |
| path: "${{ runner.temp }}/artifacts/" |
| - name: Checkout PyTorch |
| uses: malfet/checkout@silent-checkout |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| quiet-checkout: true |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: malfet/checkout@silent-checkout |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| quiet-checkout: true |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: ROCm set GPU_FLAG |
| run: | |
| echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add daemon" >> "${GITHUB_ENV}" |
| - name: Pull Docker image |
| uses: pytorch/test-infra/.github/actions/pull-docker-image@main |
| with: |
| docker-image: pytorch/manylinux-builder:rocm6.2-main |
| - name: Test Pytorch binary |
| uses: ./pytorch/.github/actions/test-pytorch-binary |
| - name: Teardown ROCm |
| uses: ./.github/actions/teardown-rocm |
| libtorch-rocm6_2-shared-with-deps-pre-cxx11-upload: # Uploading |
| if: ${{ github.repository_owner == 'pytorch' }} |
| permissions: |
| id-token: write |
| contents: read |
| needs: libtorch-rocm6_2-shared-with-deps-pre-cxx11-test |
| with: |
| PYTORCH_ROOT: /pytorch |
| BUILDER_ROOT: /builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: rocm6.2 |
| GPU_ARCH_VERSION: 6.2 |
| GPU_ARCH_TYPE: rocm |
| DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.2-main |
| LIBTORCH_VARIANT: shared-with-deps |
| DESIRED_DEVTOOLSET: pre-cxx11 |
| build_name: libtorch-rocm6_2-shared-with-deps-pre-cxx11 |
| secrets: |
| github-token: ${{ secrets.GITHUB_TOKEN }} |
| conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} |
| conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} |
| uses: ./.github/workflows/_binary-upload.yml |
