increase job time to 1000sec

Author: sandeepgupta12

.github/workflows/_binary-build-linux.yml

@@ -62,10 +62,6 @@ on:
         required: true
         type: string
         description: Docker image to use
-      DOCKER_IMAGE_TAG_PREFIX:
-        required: true
-        type: string
-        description: Docker image tag to use
       LIBTORCH_CONFIG:
         required: false
         type: string
@@ -88,9 +84,6 @@ on:
         required: true
         description: Github Token
 
-permissions:
-  id-token: write
-
 jobs:
   build:
     runs-on: ${{ inputs.runner_prefix}}${{ inputs.runs_on }}
@@ -185,7 +178,7 @@ jobs:
           fi
 
       - name: Checkout PyTorch to pytorch dir
-        uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
+        uses: actions/checkout@v4
         with:
           ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
           submodules: recursive
@@ -210,43 +203,17 @@ jobs:
               { config: "default" },
             ]}
 
-      - name: configure aws credentials
-        id: aws_creds
-        if: ${{ steps.filter.outputs.is-test-matrix-empty == 'False' && inputs.build_environment != 'linux-s390x-binary-manywheel' && inputs.build_environment != 'linux-ppc64le-binary-manywheel' && startsWith(github.event.ref, 'refs/tags/ciflow/') }}
-        uses: aws-actions/configure-aws-credentials@ececac1a45f3b08a01d2dd070d28d111c5fe6722 # v4.1.0
-        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: Build Docker image for ppc64le
-        run: |
-          docker build -f ./pytorch/.ci/docker/manywheel/Dockerfile_ppc64le -t pytorch/manylinuxppc64le-builder:cpu-ppc64le-main .
-      - name: Calculate docker image
-        id: calculate-docker-image
-        if: ${{ steps.filter.outputs.is-test-matrix-empty == 'False' && inputs.build_environment != 'linux-s390x-binary-manywheel' && inputs.build_environment != 'linux-ppc64le-binary-manywheel' }}
-        uses: pytorch/test-infra/.github/actions/calculate-docker-image@main
-        with:
-          # If doing this in main or release branch, use docker.io. Otherwise
-          # use ECR
-          docker-registry: ${{ startsWith(github.event.ref, 'refs/tags/ciflow/') && '308535385114.dkr.ecr.us-east-1.amazonaws.com' || 'docker.io' }}
-          docker-image-name: ${{ inputs.DOCKER_IMAGE }}
-          custom-tag-prefix: ${{ inputs.DOCKER_IMAGE_TAG_PREFIX }}
-          # The build.sh script in this folder is not actually the correct one,
-          # this is just needed for sha calculation
-          docker-build-dir: .ci/docker
-          working-directory: pytorch
-
       - name: Pull Docker image
         if: ${{ steps.filter.outputs.is-test-matrix-empty == 'False' && inputs.build_environment != 'linux-s390x-binary-manywheel' && inputs.build_environment != 'linux-ppc64le-binary-manywheel' }}
         uses: pytorch/test-infra/.github/actions/pull-docker-image@main
         with:
-          docker-image: ${{ steps.calculate-docker-image.outputs.docker-image }}
+          docker-image: ${{ inputs.DOCKER_IMAGE }}
 
+      - name: Build Docker image for ppc64le
+        run: |
+          docker build -f ./pytorch/.ci/docker/manywheel/Dockerfile_ppc64le -t pytorch/manylinuxppc64le-builder:cpu-ppc64le-main .
       - name: Build PyTorch binary
         if: ${{ steps.filter.outputs.is-test-matrix-empty == 'False' }}
-        env:
-          DOCKER_IMAGE: ${{ steps.calculate-docker-image.outputs.docker-image || format('{0}:{1}', inputs.DOCKER_IMAGE, inputs.DOCKER_IMAGE_TAG_PREFIX) }}
         run: |
           set -x
           mkdir -p artifacts/
@@ -273,7 +240,11 @@ jobs:
             "${DOCKER_IMAGE}"
           )
           docker exec -t -w "${PYTORCH_ROOT}" "${container_name}" bash -c "bash .circleci/scripts/binary_populate_env.sh"
-         
+          if [[ ${BUILD_ENVIRONMENT} == *"aarch64"* ]]; then
+            docker exec -t "${container_name}" bash -c "source ${BINARY_ENV_FILE} && bash /pytorch/.ci/aarch64_linux/aarch64_ci_build.sh"
+          else
+            docker exec -t "${container_name}" bash -c "source ${BINARY_ENV_FILE} && bash /pytorch/.ci/${{ inputs.PACKAGE_TYPE }}/build.sh"
+          fi
 
       - name: Chown artifacts
         if: ${{ steps.filter.outputs.is-test-matrix-empty == 'False' && inputs.build_environment != 'linux-s390x-binary-manywheel' && inputs.build_environment != 'linux-ppc64le-binary-manywheel' }}
@@ -282,7 +253,7 @@ jobs:
           # Ensure the working directory gets chowned back to the current user
           docker run --rm -v "${RUNNER_TEMP}/artifacts:/v" -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" .
 
-      - uses: actions/upload-artifact@50769540e7f4bd5e21e526ee35c689e35e0d6874 # v4.4.0
+      - uses: actions/[email protected]
         if: ${{ steps.filter.outputs.is-test-matrix-empty == 'False' }}
         with:
           name: ${{ inputs.build_name }}
@@ -301,7 +272,7 @@ jobs:
           ALPINE_IMAGE: ${{ inputs.ALPINE_IMAGE }}
 
       - name: Cleanup docker
-        if: always() && ( inputs.build_environment == 'linux-s390x-binary-manywheel' || inputs.build_environment != 'linux-ppc64le-binary-manywheel' )
+        if: always() && (inputs.build_environment == 'linux-s390x-binary-manywheel' || inputs.build_environment == 'linux-ppc64le-binary-manywheel')
         shell: bash
         run: |
           # on s390x and ppc64le stop the container for clean worker stop

.github/workflows/_linux-build.yml

@@ -126,7 +126,7 @@ jobs:
     # Don't run on forked repos
 
     runs-on: ${{ inputs.runner_prefix}}${{ inputs.runner }}
-    timeout-minutes: 240
+    timeout-minutes: 1000
     outputs:
       docker-image: ${{ steps.calculate-docker-image.outputs.docker-image }}
       test-matrix: ${{ steps.filter.outputs.test-matrix }}