Cherry-picked commit with merge conflict #3
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # @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 | ||
| 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: | ||
| permissions: | ||
| id-token: write | ||
| env: | ||
| # Needed for conda builds | ||
| ALPINE_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/alpine" | ||
| AWS_DEFAULT_REGION: us-east-1 | ||
| BINARY_ENV_FILE: /tmp/env | ||
| BUILD_ENVIRONMENT: linux-binary-libtorch | ||
| 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-${{ 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: | ||
| if: github.repository_owner == 'pytorch' | ||
| name: get-label-type | ||
| <<<<<<< HEAD | ||
| uses: pytorch/pytorch/.github/workflows/_runner-determinator.yml@release/2.9 | ||
| ======= | ||
| uses: pytorch/pytorch/.github/workflows/_runner-determinator.yml@release/2.8 | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| 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-release-build: | ||
| if: ${{ github.repository_owner == 'pytorch' }} | ||
| uses: ./.github/workflows/_binary-build-linux.yml | ||
| needs: get-label-type | ||
| with: | ||
| PYTORCH_ROOT: /pytorch | ||
| 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: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: cpu | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" | ||
| build_name: libtorch-cpu-shared-with-deps-release | ||
| build_environment: linux-binary-libtorch | ||
| secrets: | ||
| github-token: ${{ secrets.GITHUB_TOKEN }} | ||
| libtorch-cpu-shared-with-deps-release-test: # Testing | ||
| if: ${{ github.repository_owner == 'pytorch' }} | ||
| needs: | ||
| - libtorch-cpu-shared-with-deps-release-build | ||
| - get-label-type | ||
| uses: ./.github/workflows/_binary-test-linux.yml | ||
| with: | ||
| PYTORCH_ROOT: /pytorch | ||
| 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: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: cpu | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| build_name: libtorch-cpu-shared-with-deps-release | ||
| build_environment: linux-binary-libtorch | ||
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" | ||
| runs_on: linux.4xlarge | ||
| secrets: | ||
| github-token: ${{ secrets.GITHUB_TOKEN }} | ||
| libtorch-cpu-shared-with-deps-release-upload: # Uploading | ||
| if: ${{ github.repository_owner == 'pytorch' }} | ||
| permissions: | ||
| id-token: write | ||
| contents: read | ||
| needs: libtorch-cpu-shared-with-deps-release-test | ||
| with: | ||
| PYTORCH_ROOT: /pytorch | ||
| 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: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: cpu | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| build_name: libtorch-cpu-shared-with-deps-release | ||
| secrets: | ||
| github-token: ${{ secrets.GITHUB_TOKEN }} | ||
| uses: ./.github/workflows/_binary-upload.yml | ||
| libtorch-cuda12_6-shared-with-deps-release-build: | ||
| if: ${{ github.repository_owner == 'pytorch' }} | ||
| uses: ./.github/workflows/_binary-build-linux.yml | ||
| needs: get-label-type | ||
| with: | ||
| PYTORCH_ROOT: /pytorch | ||
| 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: cu126 | ||
| <<<<<<< HEAD | ||
| GPU_ARCH_VERSION: "12.6" | ||
| ======= | ||
| GPU_ARCH_VERSION: 12.6 | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| GPU_ARCH_TYPE: cuda | ||
| DOCKER_IMAGE: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: cuda12.6 | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" | ||
| build_name: libtorch-cuda12_6-shared-with-deps-release | ||
| build_environment: linux-binary-libtorch | ||
| secrets: | ||
| github-token: ${{ secrets.GITHUB_TOKEN }} | ||
| libtorch-cuda12_6-shared-with-deps-release-test: # Testing | ||
| if: ${{ github.repository_owner == 'pytorch' }} | ||
| needs: | ||
| - libtorch-cuda12_6-shared-with-deps-release-build | ||
| - get-label-type | ||
| uses: ./.github/workflows/_binary-test-linux.yml | ||
| with: | ||
| PYTORCH_ROOT: /pytorch | ||
| 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: cu126 | ||
| <<<<<<< HEAD | ||
| GPU_ARCH_VERSION: "12.6" | ||
| ======= | ||
| GPU_ARCH_VERSION: 12.6 | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| GPU_ARCH_TYPE: cuda | ||
| DOCKER_IMAGE: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: cuda12.6 | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| build_name: libtorch-cuda12_6-shared-with-deps-release | ||
| build_environment: linux-binary-libtorch | ||
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" | ||
| <<<<<<< HEAD | ||
| runs_on: linux.4xlarge.nvidia.gpu # 12.6 build can use maxwell (sm_50) runner | ||
| ======= | ||
| runs_on: linux.4xlarge.nvidia.gpu # for other cuda versions, we use 4xlarge runner | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| secrets: | ||
| github-token: ${{ secrets.GITHUB_TOKEN }} | ||
| libtorch-cuda12_6-shared-with-deps-release-upload: # Uploading | ||
| if: ${{ github.repository_owner == 'pytorch' }} | ||
| permissions: | ||
| id-token: write | ||
| contents: read | ||
| needs: libtorch-cuda12_6-shared-with-deps-release-test | ||
| with: | ||
| PYTORCH_ROOT: /pytorch | ||
| 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: cu126 | ||
| <<<<<<< HEAD | ||
| GPU_ARCH_VERSION: "12.6" | ||
| ======= | ||
| GPU_ARCH_VERSION: 12.6 | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| GPU_ARCH_TYPE: cuda | ||
| DOCKER_IMAGE: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: cuda12.6 | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| build_name: libtorch-cuda12_6-shared-with-deps-release | ||
| secrets: | ||
| github-token: ${{ secrets.GITHUB_TOKEN }} | ||
| uses: ./.github/workflows/_binary-upload.yml | ||
| libtorch-cuda12_8-shared-with-deps-release-build: | ||
| if: ${{ github.repository_owner == 'pytorch' }} | ||
| uses: ./.github/workflows/_binary-build-linux.yml | ||
| needs: get-label-type | ||
| with: | ||
| PYTORCH_ROOT: /pytorch | ||
| 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: cu128 | ||
| <<<<<<< HEAD | ||
| GPU_ARCH_VERSION: "12.8" | ||
| ======= | ||
| GPU_ARCH_VERSION: 12.8 | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| GPU_ARCH_TYPE: cuda | ||
| DOCKER_IMAGE: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: cuda12.8 | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" | ||
| build_name: libtorch-cuda12_8-shared-with-deps-release | ||
| build_environment: linux-binary-libtorch | ||
| secrets: | ||
| github-token: ${{ secrets.GITHUB_TOKEN }} | ||
| libtorch-cuda12_8-shared-with-deps-release-test: # Testing | ||
| if: ${{ github.repository_owner == 'pytorch' }} | ||
| needs: | ||
| - libtorch-cuda12_8-shared-with-deps-release-build | ||
| - get-label-type | ||
| uses: ./.github/workflows/_binary-test-linux.yml | ||
| with: | ||
| PYTORCH_ROOT: /pytorch | ||
| 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: cu128 | ||
| <<<<<<< HEAD | ||
| GPU_ARCH_VERSION: "12.8" | ||
| ======= | ||
| GPU_ARCH_VERSION: 12.8 | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| GPU_ARCH_TYPE: cuda | ||
| DOCKER_IMAGE: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: cuda12.8 | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| build_name: libtorch-cuda12_8-shared-with-deps-release | ||
| build_environment: linux-binary-libtorch | ||
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" | ||
| <<<<<<< HEAD | ||
| runs_on: linux.g4dn.4xlarge.nvidia.gpu # 12.8+ builds need sm_70+ runner | ||
| ======= | ||
| runs_on: linux.g4dn.4xlarge.nvidia.gpu # 12.8 and 12.9 build need sm_70+ runner | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| secrets: | ||
| github-token: ${{ secrets.GITHUB_TOKEN }} | ||
| libtorch-cuda12_8-shared-with-deps-release-upload: # Uploading | ||
| if: ${{ github.repository_owner == 'pytorch' }} | ||
| permissions: | ||
| id-token: write | ||
| contents: read | ||
| needs: libtorch-cuda12_8-shared-with-deps-release-test | ||
| with: | ||
| PYTORCH_ROOT: /pytorch | ||
| 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: cu128 | ||
| <<<<<<< HEAD | ||
| GPU_ARCH_VERSION: "12.8" | ||
| ======= | ||
| GPU_ARCH_VERSION: 12.8 | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| GPU_ARCH_TYPE: cuda | ||
| DOCKER_IMAGE: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: cuda12.8 | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| build_name: libtorch-cuda12_8-shared-with-deps-release | ||
| secrets: | ||
| github-token: ${{ secrets.GITHUB_TOKEN }} | ||
| uses: ./.github/workflows/_binary-upload.yml | ||
| <<<<<<< HEAD | ||
| libtorch-cuda13_0-shared-with-deps-release-build: | ||
| ======= | ||
| libtorch-cuda12_9-shared-with-deps-release-build: | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| if: ${{ github.repository_owner == 'pytorch' }} | ||
| uses: ./.github/workflows/_binary-build-linux.yml | ||
| needs: get-label-type | ||
| with: | ||
| PYTORCH_ROOT: /pytorch | ||
| PACKAGE_TYPE: libtorch | ||
| # TODO: This is a legacy variable that we eventually want to get rid of in | ||
| # favor of GPU_ARCH_VERSION | ||
| <<<<<<< HEAD | ||
| DESIRED_CUDA: cu130 | ||
| GPU_ARCH_VERSION: "13.0" | ||
| GPU_ARCH_TYPE: cuda | ||
| DOCKER_IMAGE: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: cuda13.0 | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" | ||
| build_name: libtorch-cuda13_0-shared-with-deps-release | ||
| build_environment: linux-binary-libtorch | ||
| secrets: | ||
| github-token: ${{ secrets.GITHUB_TOKEN }} | ||
| libtorch-cuda13_0-shared-with-deps-release-test: # Testing | ||
| if: ${{ github.repository_owner == 'pytorch' }} | ||
| needs: | ||
| - libtorch-cuda13_0-shared-with-deps-release-build | ||
| ======= | ||
| DESIRED_CUDA: cu129 | ||
| GPU_ARCH_VERSION: 12.9 | ||
| GPU_ARCH_TYPE: cuda | ||
| DOCKER_IMAGE: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: cuda12.9 | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" | ||
| build_name: libtorch-cuda12_9-shared-with-deps-release | ||
| build_environment: linux-binary-libtorch | ||
| secrets: | ||
| github-token: ${{ secrets.GITHUB_TOKEN }} | ||
| libtorch-cuda12_9-shared-with-deps-release-test: # Testing | ||
| if: ${{ github.repository_owner == 'pytorch' }} | ||
| needs: | ||
| - libtorch-cuda12_9-shared-with-deps-release-build | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| - get-label-type | ||
| uses: ./.github/workflows/_binary-test-linux.yml | ||
| with: | ||
| PYTORCH_ROOT: /pytorch | ||
| PACKAGE_TYPE: libtorch | ||
| # TODO: This is a legacy variable that we eventually want to get rid of in | ||
| # favor of GPU_ARCH_VERSION | ||
| <<<<<<< HEAD | ||
| DESIRED_CUDA: cu130 | ||
| GPU_ARCH_VERSION: "13.0" | ||
| GPU_ARCH_TYPE: cuda | ||
| DOCKER_IMAGE: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: cuda13.0 | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| build_name: libtorch-cuda13_0-shared-with-deps-release | ||
| build_environment: linux-binary-libtorch | ||
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" | ||
| runs_on: linux.g4dn.4xlarge.nvidia.gpu # 12.8+ builds need sm_70+ runner | ||
| secrets: | ||
| github-token: ${{ secrets.GITHUB_TOKEN }} | ||
| libtorch-cuda13_0-shared-with-deps-release-upload: # Uploading | ||
| ======= | ||
| DESIRED_CUDA: cu129 | ||
| GPU_ARCH_VERSION: 12.9 | ||
| GPU_ARCH_TYPE: cuda | ||
| DOCKER_IMAGE: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: cuda12.9 | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| build_name: libtorch-cuda12_9-shared-with-deps-release | ||
| build_environment: linux-binary-libtorch | ||
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" | ||
| runs_on: linux.g4dn.4xlarge.nvidia.gpu # 12.8 and 12.9 build need sm_70+ runner | ||
| secrets: | ||
| github-token: ${{ secrets.GITHUB_TOKEN }} | ||
| libtorch-cuda12_9-shared-with-deps-release-upload: # Uploading | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| if: ${{ github.repository_owner == 'pytorch' }} | ||
| permissions: | ||
| id-token: write | ||
| contents: read | ||
| <<<<<<< HEAD | ||
| needs: libtorch-cuda13_0-shared-with-deps-release-test | ||
| ======= | ||
| needs: libtorch-cuda12_9-shared-with-deps-release-test | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| with: | ||
| PYTORCH_ROOT: /pytorch | ||
| PACKAGE_TYPE: libtorch | ||
| # TODO: This is a legacy variable that we eventually want to get rid of in | ||
| # favor of GPU_ARCH_VERSION | ||
| <<<<<<< HEAD | ||
| DESIRED_CUDA: cu130 | ||
| GPU_ARCH_VERSION: "13.0" | ||
| GPU_ARCH_TYPE: cuda | ||
| DOCKER_IMAGE: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: cuda13.0 | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| build_name: libtorch-cuda13_0-shared-with-deps-release | ||
| ======= | ||
| DESIRED_CUDA: cu129 | ||
| GPU_ARCH_VERSION: 12.9 | ||
| GPU_ARCH_TYPE: cuda | ||
| DOCKER_IMAGE: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: cuda12.9 | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| build_name: libtorch-cuda12_9-shared-with-deps-release | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| secrets: | ||
| github-token: ${{ secrets.GITHUB_TOKEN }} | ||
| uses: ./.github/workflows/_binary-upload.yml | ||
| libtorch-rocm6_3-shared-with-deps-release-build: | ||
| if: ${{ github.repository_owner == 'pytorch' }} | ||
| uses: ./.github/workflows/_binary-build-linux.yml | ||
| needs: get-label-type | ||
| with: | ||
| PYTORCH_ROOT: /pytorch | ||
| 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.3 | ||
| <<<<<<< HEAD | ||
| GPU_ARCH_VERSION: "6.3" | ||
| ======= | ||
| GPU_ARCH_VERSION: 6.3 | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| GPU_ARCH_TYPE: rocm | ||
| DOCKER_IMAGE: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: rocm6.3 | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" | ||
| <<<<<<< HEAD | ||
| timeout-minutes: 300 | ||
| ======= | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| build_name: libtorch-rocm6_3-shared-with-deps-release | ||
| build_environment: linux-binary-libtorch | ||
| secrets: | ||
| github-token: ${{ secrets.GITHUB_TOKEN }} | ||
| libtorch-rocm6_3-shared-with-deps-release-test: # Testing | ||
| if: ${{ github.repository_owner == 'pytorch' }} | ||
| needs: | ||
| - libtorch-rocm6_3-shared-with-deps-release-build | ||
| - get-label-type | ||
| runs-on: linux.rocm.gpu.mi250 | ||
| timeout-minutes: 240 | ||
| env: | ||
| PYTORCH_ROOT: /pytorch | ||
| 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.3 | ||
| <<<<<<< HEAD | ||
| GPU_ARCH_VERSION: "6.3" | ||
| ======= | ||
| GPU_ARCH_VERSION: 6.3 | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| GPU_ARCH_TYPE: rocm | ||
| SKIP_ALL_TESTS: 1 | ||
| DOCKER_IMAGE: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: rocm6.3 | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| steps: | ||
| - name: Setup ROCm | ||
| uses: ./.github/actions/setup-rocm | ||
| - uses: actions/[email protected] | ||
| name: Download Build Artifacts | ||
| with: | ||
| name: libtorch-rocm6_3-shared-with-deps-release | ||
| path: "${{ runner.temp }}/artifacts/" | ||
| - name: Checkout PyTorch | ||
| uses: actions/checkout@v4 | ||
| with: | ||
| submodules: recursive | ||
| path: pytorch | ||
| show-progress: false | ||
| - name: Clean PyTorch checkout | ||
| run: | | ||
| # Remove any artifacts from the previous checkouts | ||
| git clean -fxd | ||
| working-directory: pytorch | ||
| - 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: configure aws credentials | ||
| id: aws_creds | ||
| if: ${{ startsWith(github.event.ref, 'refs/tags/ciflow/') }} | ||
| uses: aws-actions/configure-aws-credentials@v4 | ||
| with: | ||
| role-to-assume: arn:aws:iam::308535385114:role/gha_workflow_s3_and_ecr_read_only | ||
| aws-region: us-east-1 | ||
| role-duration-seconds: 18000 | ||
| - name: Calculate docker image | ||
| id: calculate-docker-image | ||
| <<<<<<< HEAD | ||
| uses: pytorch/test-infra/.github/actions/calculate-docker-image@release/2.9 | ||
| ======= | ||
| uses: pytorch/test-infra/.github/actions/calculate-docker-image@release/2.8 | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| with: | ||
| docker-registry: ${{ startsWith(github.event.ref, 'refs/tags/ciflow/') && '308535385114.dkr.ecr.us-east-1.amazonaws.com' || 'docker.io' }} | ||
| docker-image-name: libtorch-cxx11-builder | ||
| custom-tag-prefix: rocm6.3 | ||
| docker-build-dir: .ci/docker | ||
| working-directory: pytorch | ||
| - name: Pull Docker image | ||
| <<<<<<< HEAD | ||
| uses: pytorch/test-infra/.github/actions/pull-docker-image@release/2.9 | ||
| ======= | ||
| uses: pytorch/test-infra/.github/actions/pull-docker-image@release/2.8 | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| with: | ||
| docker-image: ${{ steps.calculate-docker-image.outputs.docker-image }} | ||
| - name: Test Pytorch binary | ||
| uses: ./pytorch/.github/actions/test-pytorch-binary | ||
| env: | ||
| DOCKER_IMAGE: ${{ steps.calculate-docker-image.outputs.docker-image }} | ||
| - name: Teardown ROCm | ||
| uses: ./.github/actions/teardown-rocm | ||
| libtorch-rocm6_3-shared-with-deps-release-upload: # Uploading | ||
| if: ${{ github.repository_owner == 'pytorch' }} | ||
| permissions: | ||
| id-token: write | ||
| contents: read | ||
| needs: libtorch-rocm6_3-shared-with-deps-release-test | ||
| with: | ||
| PYTORCH_ROOT: /pytorch | ||
| 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.3 | ||
| <<<<<<< HEAD | ||
| GPU_ARCH_VERSION: "6.3" | ||
| ======= | ||
| GPU_ARCH_VERSION: 6.3 | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| GPU_ARCH_TYPE: rocm | ||
| DOCKER_IMAGE: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: rocm6.3 | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| build_name: libtorch-rocm6_3-shared-with-deps-release | ||
| secrets: | ||
| github-token: ${{ secrets.GITHUB_TOKEN }} | ||
| uses: ./.github/workflows/_binary-upload.yml | ||
| libtorch-rocm6_4-shared-with-deps-release-build: | ||
| if: ${{ github.repository_owner == 'pytorch' }} | ||
| uses: ./.github/workflows/_binary-build-linux.yml | ||
| needs: get-label-type | ||
| with: | ||
| PYTORCH_ROOT: /pytorch | ||
| 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.4 | ||
| <<<<<<< HEAD | ||
| GPU_ARCH_VERSION: "6.4" | ||
| ======= | ||
| GPU_ARCH_VERSION: 6.4 | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| GPU_ARCH_TYPE: rocm | ||
| DOCKER_IMAGE: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: rocm6.4 | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" | ||
| <<<<<<< HEAD | ||
| timeout-minutes: 300 | ||
| ======= | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| build_name: libtorch-rocm6_4-shared-with-deps-release | ||
| build_environment: linux-binary-libtorch | ||
| secrets: | ||
| github-token: ${{ secrets.GITHUB_TOKEN }} | ||
| libtorch-rocm6_4-shared-with-deps-release-test: # Testing | ||
| if: ${{ github.repository_owner == 'pytorch' }} | ||
| needs: | ||
| - libtorch-rocm6_4-shared-with-deps-release-build | ||
| - get-label-type | ||
| runs-on: linux.rocm.gpu.mi250 | ||
| timeout-minutes: 240 | ||
| env: | ||
| PYTORCH_ROOT: /pytorch | ||
| 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.4 | ||
| <<<<<<< HEAD | ||
| GPU_ARCH_VERSION: "6.4" | ||
| ======= | ||
| GPU_ARCH_VERSION: 6.4 | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| GPU_ARCH_TYPE: rocm | ||
| SKIP_ALL_TESTS: 1 | ||
| DOCKER_IMAGE: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: rocm6.4 | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| steps: | ||
| - name: Setup ROCm | ||
| uses: ./.github/actions/setup-rocm | ||
| - uses: actions/[email protected] | ||
| name: Download Build Artifacts | ||
| with: | ||
| name: libtorch-rocm6_4-shared-with-deps-release | ||
| path: "${{ runner.temp }}/artifacts/" | ||
| - name: Checkout PyTorch | ||
| uses: actions/checkout@v4 | ||
| with: | ||
| submodules: recursive | ||
| path: pytorch | ||
| show-progress: false | ||
| - name: Clean PyTorch checkout | ||
| run: | | ||
| # Remove any artifacts from the previous checkouts | ||
| git clean -fxd | ||
| working-directory: pytorch | ||
| - 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: configure aws credentials | ||
| id: aws_creds | ||
| if: ${{ startsWith(github.event.ref, 'refs/tags/ciflow/') }} | ||
| uses: aws-actions/configure-aws-credentials@v4 | ||
| with: | ||
| role-to-assume: arn:aws:iam::308535385114:role/gha_workflow_s3_and_ecr_read_only | ||
| aws-region: us-east-1 | ||
| role-duration-seconds: 18000 | ||
| - name: Calculate docker image | ||
| id: calculate-docker-image | ||
| <<<<<<< HEAD | ||
| uses: pytorch/test-infra/.github/actions/calculate-docker-image@release/2.9 | ||
| ======= | ||
| uses: pytorch/test-infra/.github/actions/calculate-docker-image@release/2.8 | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| with: | ||
| docker-registry: ${{ startsWith(github.event.ref, 'refs/tags/ciflow/') && '308535385114.dkr.ecr.us-east-1.amazonaws.com' || 'docker.io' }} | ||
| docker-image-name: libtorch-cxx11-builder | ||
| custom-tag-prefix: rocm6.4 | ||
| docker-build-dir: .ci/docker | ||
| working-directory: pytorch | ||
| - name: Pull Docker image | ||
| <<<<<<< HEAD | ||
| uses: pytorch/test-infra/.github/actions/pull-docker-image@release/2.9 | ||
| ======= | ||
| uses: pytorch/test-infra/.github/actions/pull-docker-image@release/2.8 | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| with: | ||
| docker-image: ${{ steps.calculate-docker-image.outputs.docker-image }} | ||
| - name: Test Pytorch binary | ||
| uses: ./pytorch/.github/actions/test-pytorch-binary | ||
| env: | ||
| DOCKER_IMAGE: ${{ steps.calculate-docker-image.outputs.docker-image }} | ||
| - name: Teardown ROCm | ||
| uses: ./.github/actions/teardown-rocm | ||
| libtorch-rocm6_4-shared-with-deps-release-upload: # Uploading | ||
| if: ${{ github.repository_owner == 'pytorch' }} | ||
| permissions: | ||
| id-token: write | ||
| contents: read | ||
| needs: libtorch-rocm6_4-shared-with-deps-release-test | ||
| with: | ||
| PYTORCH_ROOT: /pytorch | ||
| 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.4 | ||
| <<<<<<< HEAD | ||
| GPU_ARCH_VERSION: "6.4" | ||
| ======= | ||
| GPU_ARCH_VERSION: 6.4 | ||
| >>>>>>> 5729657180 ([ROCm] Specialized binary elementwise broadcast kernel for mixed dtypes with float/bfloat16/half (#2791)) | ||
| GPU_ARCH_TYPE: rocm | ||
| DOCKER_IMAGE: libtorch-cxx11-builder | ||
| DOCKER_IMAGE_TAG_PREFIX: rocm6.4 | ||
| LIBTORCH_CONFIG: release | ||
| LIBTORCH_VARIANT: shared-with-deps | ||
| build_name: libtorch-rocm6_4-shared-with-deps-release | ||
| secrets: | ||
| github-token: ${{ secrets.GITHUB_TOKEN }} | ||
| uses: ./.github/workflows/_binary-upload.yml | ||
