|
43 | 43 | curr_branch: ${{ github.head_ref || github.ref_name }}
|
44 | 44 | curr_ref_type: ${{ github.ref_type }}
|
45 | 45 |
|
46 |
| - build-docker-cuda: |
47 |
| - environment: ${{ (github.ref == 'refs/heads/main' || startsWith(github.event.ref, 'refs/tags/v')) && 'docker-build' || '' }} |
48 |
| - needs: get-label-type |
49 |
| - runs-on: "${{ needs.get-label-type.outputs.label-type }}linux.9xlarge.ephemeral" |
50 |
| - strategy: |
51 |
| - matrix: |
52 |
| - cuda_version: ["12.8", "12.6", "12.4", "11.8"] |
53 |
| - env: |
54 |
| - GPU_ARCH_TYPE: cuda |
55 |
| - GPU_ARCH_VERSION: ${{ matrix.cuda_version }} |
56 |
| - steps: |
57 |
| - - name: Purge tools folder (free space for build) |
58 |
| - run: rm -rf /opt/hostedtoolcache |
59 |
| - - name: Checkout PyTorch |
60 |
| - uses: pytorch/pytorch/.github/actions/checkout-pytorch@main |
61 |
| - with: |
62 |
| - submodules: false |
63 |
| - - name: Calculate docker image |
64 |
| - if: env.WITH_PUSH == 'false' |
65 |
| - uses: pytorch/test-infra/.github/actions/calculate-docker-image@main |
66 |
| - with: |
67 |
| - docker-image-name: manylinux-builder-cuda${{matrix.cuda_version}} |
68 |
| - docker-build-dir: .ci/docker/manywheel |
69 |
| - always-rebuild: true |
70 |
| - push: true |
71 |
| - - name: Authenticate if WITH_PUSH |
72 |
| - if: env.WITH_PUSH == 'true' |
73 |
| - env: |
74 |
| - DOCKER_TOKEN: ${{ secrets.DOCKER_TOKEN }} |
75 |
| - DOCKER_ID: ${{ secrets.DOCKER_ID }} |
76 |
| - run: | |
77 |
| - if [[ "${WITH_PUSH}" == true ]]; then |
78 |
| - echo "${DOCKER_TOKEN}" | docker login -u "${DOCKER_ID}" --password-stdin |
79 |
| - fi |
80 |
| - - name: Build Docker Image |
81 |
| - if: env.WITH_PUSH == 'true' |
82 |
| - uses: nick-fields/retry@v3.0.0 |
83 |
| - with: |
84 |
| - shell: bash |
85 |
| - timeout_minutes: 90 |
86 |
| - max_attempts: 3 |
87 |
| - retry_wait_seconds: 90 |
88 |
| - command: | |
89 |
| - .ci/docker/manywheel/build.sh manylinux-builder:cuda${{matrix.cuda_version}} |
90 |
| - # NOTE: manylinux_2_28 are still experimental, see https://github.com/pytorch/pytorch/issues/123649 |
91 | 46 | build-docker-cuda-manylinux_2_28:
|
92 | 47 | environment: ${{ (github.ref == 'refs/heads/main' || startsWith(github.event.ref, 'refs/tags/v')) && 'docker-build' || '' }}
|
93 | 48 | needs: get-label-type
|
@@ -214,42 +169,6 @@ jobs:
|
214 | 169 | retry_wait_seconds: 90
|
215 | 170 | command: |
|
216 | 171 | .ci/docker/manywheel/build.sh manylinux2_28-builder:rocm${{matrix.rocm_version}}
|
217 |
| - build-docker-cpu: |
218 |
| - environment: ${{ (github.ref == 'refs/heads/main' || startsWith(github.event.ref, 'refs/tags/v')) && 'docker-build' || '' }} |
219 |
| - needs: get-label-type |
220 |
| - runs-on: "${{ needs.get-label-type.outputs.label-type }}linux.9xlarge.ephemeral" |
221 |
| - steps: |
222 |
| - - name: Checkout PyTorch |
223 |
| - uses: pytorch/pytorch/.github/actions/checkout-pytorch@main |
224 |
| - with: |
225 |
| - submodules: false |
226 |
| - - name: Calculate docker image |
227 |
| - if: env.WITH_PUSH == 'false' |
228 |
| - uses: pytorch/test-infra/.github/actions/calculate-docker-image@main |
229 |
| - with: |
230 |
| - docker-image-name: manylinux-builder-cpu |
231 |
| - docker-build-dir: .ci/docker/manywheel |
232 |
| - always-rebuild: true |
233 |
| - push: true |
234 |
| - - name: Authenticate if WITH_PUSH |
235 |
| - if: env.WITH_PUSH == 'true' |
236 |
| - env: |
237 |
| - DOCKER_TOKEN: ${{ secrets.DOCKER_TOKEN }} |
238 |
| - DOCKER_ID: ${{ secrets.DOCKER_ID }} |
239 |
| - run: | |
240 |
| - if [[ "${WITH_PUSH}" == true ]]; then |
241 |
| - echo "${DOCKER_TOKEN}" | docker login -u "${DOCKER_ID}" --password-stdin |
242 |
| - fi |
243 |
| - - name: Build Docker Image |
244 |
| - if: env.WITH_PUSH == 'true' |
245 |
| - uses: nick-fields/retry@v3.0.0 |
246 |
| - with: |
247 |
| - shell: bash |
248 |
| - timeout_minutes: 90 |
249 |
| - max_attempts: 3 |
250 |
| - retry_wait_seconds: 90 |
251 |
| - command: | |
252 |
| - .ci/docker/manywheel/build.sh manylinux-builder:cpu |
253 | 172 | build-docker-cpu-manylinux_2_28:
|
254 | 173 | environment: ${{ (github.ref == 'refs/heads/main' || startsWith(github.event.ref, 'refs/tags/v')) && 'docker-build' || '' }}
|
255 | 174 | needs: get-label-type
|
|
0 commit comments