@@ -109,7 +109,7 @@ jobs:
109109 steps :
110110 - name : Setup SSH (Click me for login details)
111111 uses : pytorch/test-infra/.github/actions/setup-ssh@main
112- if : inputs.build-environment != 'linux-s390x-binary-manywheel'
112+ if : inputs.build-environment != 'linux-s390x-binary-manywheel' && inputs.build-environment != 'linux-ppc64le-binary-manywheel'
113113 with :
114114 github-secret : ${{ secrets.GITHUB_TOKEN }}
115115
@@ -120,15 +120,15 @@ jobs:
120120 - name : Checkout PyTorch
121121 uses : pytorch/pytorch/.github/actions/checkout-pytorch@main
122122 with :
123- no-sudo : ${{ inputs.build-environment == 'linux-s390x-binary-manywheel' }}
123+ no-sudo : ${{ inputs.build-environment == 'linux-s390x-binary-manywheel' || inputs.build-environment == 'linux-ppc64le-binary-manywheel' }}
124124
125125 - name : Setup Linux
126126 uses : ./.github/actions/setup-linux
127- if : inputs.build-environment != 'linux-s390x-binary-manywheel'
127+ if : inputs.build-environment != 'linux-s390x-binary-manywheel' && inputs.build-environment != 'linux-ppc64le-binary-manywheel'
128128
129129 - name : configure aws credentials
130130 uses : aws-actions/configure-aws-credentials@v3
131- if : ${{ inputs.aws-role-to-assume != '' && inputs.build-environment != 'linux-s390x-binary-manywheel' }}
131+ if : ${{ inputs.aws-role-to-assume != '' && inputs.build-environment != 'linux-s390x-binary-manywheel' && inputs.build-environment != 'linux-ppc64le-binary-manywheel' }}
132132 with :
133133 role-to-assume : ${{ inputs.aws-role-to-assume }}
134134 role-session-name : gha-linux-build
@@ -137,13 +137,13 @@ jobs:
137137 - name : Calculate docker image
138138 id : calculate-docker-image
139139 uses : pytorch/test-infra/.github/actions/calculate-docker-image@main
140- if : inputs.build-environment != 'linux-s390x-binary-manywheel'
140+ if : inputs.build-environment != 'linux-s390x-binary-manywheel' && inputs.build-environment != 'linux-ppc64le-binary-manywheel'
141141 with :
142142 docker-image-name : ${{ inputs.docker-image-name }}
143143
144144 - name : Use following to pull public copy of the image
145145 id : print-ghcr-mirror
146- if : inputs.build-environment != 'linux-s390x-binary-manywheel'
146+ if : inputs.build-environment != 'linux-s390x-binary-manywheel' && inputs.build-environment != 'linux-ppc64le-binary-manywheel'
147147 env :
148148 ECR_DOCKER_IMAGE : ${{ steps.calculate-docker-image.outputs.docker-image }}
149149 shell : bash
@@ -153,7 +153,7 @@ jobs:
153153
154154 - name : Pull docker image
155155 uses : pytorch/test-infra/.github/actions/pull-docker-image@main
156- if : inputs.build-environment != 'linux-s390x-binary-manywheel'
156+ if : inputs.build-environment != 'linux-s390x-binary-manywheel' && inputs.build-environment != 'linux-ppc64le-binary-manywheel'
157157 with :
158158 docker-image : ${{ steps.calculate-docker-image.outputs.docker-image }}
159159
@@ -212,7 +212,7 @@ jobs:
212212 SCRIBE_GRAPHQL_ACCESS_TOKEN : ${{ secrets.SCRIBE_GRAPHQL_ACCESS_TOKEN }}
213213 USE_SPLIT_BUILD : ${{ inputs.use_split_build }}
214214 run : |
215- if [[ ${BUILD_ENVIRONMENT} == *"s390x"* ]]; then
215+ if [[ ${BUILD_ENVIRONMENT} == *"s390x"* || ${BUILD_ENVIRONMENT} == *"ppc64le"* ]]; then
216216 JENKINS_USER=
217217 USED_IMAGE="${DOCKER_IMAGE_S390X}"
218218
@@ -278,7 +278,7 @@ jobs:
278278
279279 - name : Store PyTorch Build Artifacts on S3 for split build
280280 uses : seemethere/upload-artifact-s3@v5
281- if : inputs.build-generates-artifacts && steps.build.outcome != 'skipped' && inputs.use_split_build && inputs.build-environment != 'linux-s390x-binary-manywheel'
281+ if : inputs.build-generates-artifacts && steps.build.outcome != 'skipped' && inputs.use_split_build && inputs.build-environment != 'linux-s390x-binary-manywheel' && inputs.build-environment != 'linux-ppc64le-binary-manywheel'
282282 with :
283283 name : ${{ inputs.build-environment }}-experimental-split-build
284284 retention-days : 14
0 commit comments