docker mage job

Author: sandeepgupta12

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

@@ -240,10 +240,20 @@ jobs:
         with:
           docker-image: ${{ steps.calculate-docker-image.outputs.docker-image }}
 
-      - name: Build Docker image for ppc64le
-        run: |
-          docker build --network=host -f ./pytorch/.ci/docker/manywheel/Dockerfile_ppc64le -t pytorch/manylinuxppc64le-builder:cpu-ppc64le-main .
+      # - name: Build Docker image for ppc64le
+      #   run: |
+      #     docker build --network=host -f ./pytorch/.ci/docker/manywheel/Dockerfile_ppc64le -t pytorch/manylinuxppc64le-builder:cpu-ppc64le-main .
+
+      - name: Download Docker image artifact
+        uses: actions/download-artifact@v4
+        with:
+          name: ppc64le-builder-image
+          path: /tmp
 
+      - name: Load Docker image
+        run: |
+          gunzip -c /tmp/ppc64le-builder.tar.gz | docker load
+          
       - name: Build PyTorch binary
         if: ${{ steps.filter.outputs.is-test-matrix-empty == 'False' }}
         env:

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

@@ -223,10 +223,20 @@ jobs:
         uses: pytorch/test-infra/.github/actions/pull-docker-image@main
         with:
           docker-image: ${{ steps.calculate-docker-image.outputs.docker-image }}
-      - name: Build Docker image for ppc64le
-        run: |
-          docker build --network=host -f ./pytorch/.ci/docker/manywheel/Dockerfile_ppc64le -t pytorch/manylinuxppc64le-builder:cpu-ppc64le-main .
+      # - name: Build Docker image for ppc64le
+      #   run: |
+      #     docker build --network=host -f ./pytorch/.ci/docker/manywheel/Dockerfile_ppc64le -t pytorch/manylinuxppc64le-builder:cpu-ppc64le-main .
+
+      - name: Download Docker image artifact
+        uses: actions/download-artifact@v4
+        with:
+          name: ppc64le-builder-image
+          path: /tmp
 
+      - name: Load Docker image
+        run: |
+          gunzip -c /tmp/ppc64le-builder.tar.gz | docker load
+          
       - name: Test Pytorch binary
         if: ${{ steps.filter.outputs.is-test-matrix-empty == 'False' }}
         uses: ./pytorch/.github/actions/test-pytorch-binary