_calculate-docker-image.yml

name: calculate-docker-image

on:
  workflow_call:
    inputs:
      docker-image-name:
        required: true
        type: string
        description: Name of the base docker image to build with.

    outputs:
      docker-image:
        value: ${{ jobs.calculate-docker-image.outputs.docker-image }}
        description: The docker image containing the built PyTorch.

jobs:
  calculate-docker-image:
    if: github.repository_owner == 'pytorch'
    runs-on: [self-hosted, linux.large]
    timeout-minutes: 30
    outputs:
      docker-image: ${{ steps.calculate-docker-image.outputs.docker-image }}
    steps:
      - name: Checkout PyTorch
        uses: pytorch/pytorch/.github/actions/checkout-pytorch@master
        with:
          submodules: false

      - name: Setup Linux
        uses: ./.github/actions/setup-linux

      - name: Calculate docker image
        id: calculate-docker-image
        uses: ./.github/actions/calculate-docker-image
        with:
          docker-image-name: ${{ inputs.docker-image-name }}
          # TODO: This flag is needed to make the Docker image available on ECR
          # to avoid "Requested image not found" error for linter workflow. The
          # longer term fix would be to refactor linter so that building the new
          # image and running lint are done in the same runner like _linux-build
          skip_push: false