• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# @generated DO NOT EDIT MANUALLY
2
3# Template is at:    .github/templates/linux_binary_build_workflow.yml.j2
4# Generation script: .github/scripts/generate_ci_workflows.py
5name: linux-binary-libtorch-pre-cxx11
6
7
8on:
9  push:
10    # NOTE: Meta Employees can trigger new nightlies using: https://fburl.com/trigger_pytorch_nightly_build
11    branches:
12      - nightly
13    tags:
14      # NOTE: Binary build pipelines should only get triggered on release candidate builds
15      # Release candidate tags look like: v1.11.0-rc1
16      - v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
17      - 'ciflow/binaries/*'
18      - 'ciflow/binaries_libtorch/*'
19  workflow_dispatch:
20
21env:
22  # Needed for conda builds
23  ALPINE_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/alpine"
24  ANACONDA_USER: pytorch
25  AWS_DEFAULT_REGION: us-east-1
26  BINARY_ENV_FILE: /tmp/env
27  BUILD_ENVIRONMENT: linux-binary-libtorch-pre-cxx11
28  BUILDER_ROOT: /builder
29  GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
30  PR_NUMBER: ${{ github.event.pull_request.number }}
31  PYTORCH_FINAL_PACKAGE_DIR: /artifacts
32  PYTORCH_ROOT: /pytorch
33  SHA1: ${{ github.event.pull_request.head.sha || github.sha }}
34  SKIP_ALL_TESTS: 0
35concurrency:
36  group: linux-binary-libtorch-pre-cxx11-${{ github.event.pull_request.number || github.ref_name }}-${{ github.ref_type == 'branch' && github.sha }}-${{ github.event_name == 'workflow_dispatch' }}
37  cancel-in-progress: true
38
39jobs:
40  libtorch-cpu-shared-with-deps-pre-cxx11-build:
41    if: ${{ github.repository_owner == 'pytorch' }}
42    uses: ./.github/workflows/_binary-build-linux.yml
43    with:
44      PYTORCH_ROOT: /pytorch
45      BUILDER_ROOT: /builder
46      PACKAGE_TYPE: libtorch
47      # TODO: This is a legacy variable that we eventually want to get rid of in
48      #       favor of GPU_ARCH_VERSION
49      DESIRED_CUDA: cpu
50      GPU_ARCH_TYPE: cpu
51      DOCKER_IMAGE: pytorch/manylinux-builder:cpu-2.4
52      LIBTORCH_VARIANT: shared-with-deps
53      DESIRED_DEVTOOLSET: pre-cxx11
54      build_name: libtorch-cpu-shared-with-deps-pre-cxx11
55      build_environment: linux-binary-libtorch-pre-cxx11
56    secrets:
57      github-token: ${{ secrets.GITHUB_TOKEN }}
58  libtorch-cpu-shared-with-deps-pre-cxx11-test:  # Testing
59    if: ${{ github.repository_owner == 'pytorch' }}
60    needs: libtorch-cpu-shared-with-deps-pre-cxx11-build
61    uses: ./.github/workflows/_binary-test-linux.yml
62    with:
63      PYTORCH_ROOT: /pytorch
64      BUILDER_ROOT: /builder
65      PACKAGE_TYPE: libtorch
66      # TODO: This is a legacy variable that we eventually want to get rid of in
67      #       favor of GPU_ARCH_VERSION
68      DESIRED_CUDA: cpu
69      GPU_ARCH_TYPE: cpu
70      DOCKER_IMAGE: pytorch/manylinux-builder:cpu-2.4
71      LIBTORCH_VARIANT: shared-with-deps
72      DESIRED_DEVTOOLSET: pre-cxx11
73      build_name: libtorch-cpu-shared-with-deps-pre-cxx11
74      build_environment: linux-binary-libtorch-pre-cxx11
75      runs_on: linux.4xlarge
76    secrets:
77      github-token: ${{ secrets.GITHUB_TOKEN }}
78  libtorch-cpu-shared-with-deps-pre-cxx11-upload:  # Uploading
79    if: ${{ github.repository_owner == 'pytorch' }}
80    permissions:
81      id-token: write
82      contents: read
83    needs: libtorch-cpu-shared-with-deps-pre-cxx11-test
84    with:
85      PYTORCH_ROOT: /pytorch
86      BUILDER_ROOT: /builder
87      PACKAGE_TYPE: libtorch
88      # TODO: This is a legacy variable that we eventually want to get rid of in
89      #       favor of GPU_ARCH_VERSION
90      DESIRED_CUDA: cpu
91      GPU_ARCH_TYPE: cpu
92      DOCKER_IMAGE: pytorch/manylinux-builder:cpu-2.4
93      LIBTORCH_VARIANT: shared-with-deps
94      DESIRED_DEVTOOLSET: pre-cxx11
95      build_name: libtorch-cpu-shared-with-deps-pre-cxx11
96    secrets:
97      github-token: ${{ secrets.GITHUB_TOKEN }}
98      conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
99      conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
100    uses: ./.github/workflows/_binary-upload.yml
101
102  libtorch-cuda11_8-shared-with-deps-pre-cxx11-build:
103    if: ${{ github.repository_owner == 'pytorch' }}
104    uses: ./.github/workflows/_binary-build-linux.yml
105    with:
106      PYTORCH_ROOT: /pytorch
107      BUILDER_ROOT: /builder
108      PACKAGE_TYPE: libtorch
109      # TODO: This is a legacy variable that we eventually want to get rid of in
110      #       favor of GPU_ARCH_VERSION
111      DESIRED_CUDA: cu118
112      GPU_ARCH_VERSION: 11.8
113      GPU_ARCH_TYPE: cuda
114      DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4
115      LIBTORCH_VARIANT: shared-with-deps
116      DESIRED_DEVTOOLSET: pre-cxx11
117      build_name: libtorch-cuda11_8-shared-with-deps-pre-cxx11
118      build_environment: linux-binary-libtorch-pre-cxx11
119    secrets:
120      github-token: ${{ secrets.GITHUB_TOKEN }}
121  libtorch-cuda11_8-shared-with-deps-pre-cxx11-test:  # Testing
122    if: ${{ github.repository_owner == 'pytorch' }}
123    needs: libtorch-cuda11_8-shared-with-deps-pre-cxx11-build
124    uses: ./.github/workflows/_binary-test-linux.yml
125    with:
126      PYTORCH_ROOT: /pytorch
127      BUILDER_ROOT: /builder
128      PACKAGE_TYPE: libtorch
129      # TODO: This is a legacy variable that we eventually want to get rid of in
130      #       favor of GPU_ARCH_VERSION
131      DESIRED_CUDA: cu118
132      GPU_ARCH_VERSION: 11.8
133      GPU_ARCH_TYPE: cuda
134      DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4
135      LIBTORCH_VARIANT: shared-with-deps
136      DESIRED_DEVTOOLSET: pre-cxx11
137      build_name: libtorch-cuda11_8-shared-with-deps-pre-cxx11
138      build_environment: linux-binary-libtorch-pre-cxx11
139      runs_on: linux.4xlarge.nvidia.gpu
140    secrets:
141      github-token: ${{ secrets.GITHUB_TOKEN }}
142  libtorch-cuda11_8-shared-with-deps-pre-cxx11-upload:  # Uploading
143    if: ${{ github.repository_owner == 'pytorch' }}
144    permissions:
145      id-token: write
146      contents: read
147    needs: libtorch-cuda11_8-shared-with-deps-pre-cxx11-test
148    with:
149      PYTORCH_ROOT: /pytorch
150      BUILDER_ROOT: /builder
151      PACKAGE_TYPE: libtorch
152      # TODO: This is a legacy variable that we eventually want to get rid of in
153      #       favor of GPU_ARCH_VERSION
154      DESIRED_CUDA: cu118
155      GPU_ARCH_VERSION: 11.8
156      GPU_ARCH_TYPE: cuda
157      DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4
158      LIBTORCH_VARIANT: shared-with-deps
159      DESIRED_DEVTOOLSET: pre-cxx11
160      build_name: libtorch-cuda11_8-shared-with-deps-pre-cxx11
161    secrets:
162      github-token: ${{ secrets.GITHUB_TOKEN }}
163      conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
164      conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
165    uses: ./.github/workflows/_binary-upload.yml
166
167  libtorch-cuda12_1-shared-with-deps-pre-cxx11-build:
168    if: ${{ github.repository_owner == 'pytorch' }}
169    uses: ./.github/workflows/_binary-build-linux.yml
170    with:
171      PYTORCH_ROOT: /pytorch
172      BUILDER_ROOT: /builder
173      PACKAGE_TYPE: libtorch
174      # TODO: This is a legacy variable that we eventually want to get rid of in
175      #       favor of GPU_ARCH_VERSION
176      DESIRED_CUDA: cu121
177      GPU_ARCH_VERSION: 12.1
178      GPU_ARCH_TYPE: cuda
179      DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
180      LIBTORCH_VARIANT: shared-with-deps
181      DESIRED_DEVTOOLSET: pre-cxx11
182      build_name: libtorch-cuda12_1-shared-with-deps-pre-cxx11
183      build_environment: linux-binary-libtorch-pre-cxx11
184    secrets:
185      github-token: ${{ secrets.GITHUB_TOKEN }}
186  libtorch-cuda12_1-shared-with-deps-pre-cxx11-test:  # Testing
187    if: ${{ github.repository_owner == 'pytorch' }}
188    needs: libtorch-cuda12_1-shared-with-deps-pre-cxx11-build
189    uses: ./.github/workflows/_binary-test-linux.yml
190    with:
191      PYTORCH_ROOT: /pytorch
192      BUILDER_ROOT: /builder
193      PACKAGE_TYPE: libtorch
194      # TODO: This is a legacy variable that we eventually want to get rid of in
195      #       favor of GPU_ARCH_VERSION
196      DESIRED_CUDA: cu121
197      GPU_ARCH_VERSION: 12.1
198      GPU_ARCH_TYPE: cuda
199      DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
200      LIBTORCH_VARIANT: shared-with-deps
201      DESIRED_DEVTOOLSET: pre-cxx11
202      build_name: libtorch-cuda12_1-shared-with-deps-pre-cxx11
203      build_environment: linux-binary-libtorch-pre-cxx11
204      runs_on: linux.4xlarge.nvidia.gpu
205    secrets:
206      github-token: ${{ secrets.GITHUB_TOKEN }}
207  libtorch-cuda12_1-shared-with-deps-pre-cxx11-upload:  # Uploading
208    if: ${{ github.repository_owner == 'pytorch' }}
209    permissions:
210      id-token: write
211      contents: read
212    needs: libtorch-cuda12_1-shared-with-deps-pre-cxx11-test
213    with:
214      PYTORCH_ROOT: /pytorch
215      BUILDER_ROOT: /builder
216      PACKAGE_TYPE: libtorch
217      # TODO: This is a legacy variable that we eventually want to get rid of in
218      #       favor of GPU_ARCH_VERSION
219      DESIRED_CUDA: cu121
220      GPU_ARCH_VERSION: 12.1
221      GPU_ARCH_TYPE: cuda
222      DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
223      LIBTORCH_VARIANT: shared-with-deps
224      DESIRED_DEVTOOLSET: pre-cxx11
225      build_name: libtorch-cuda12_1-shared-with-deps-pre-cxx11
226    secrets:
227      github-token: ${{ secrets.GITHUB_TOKEN }}
228      conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
229      conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
230    uses: ./.github/workflows/_binary-upload.yml
231
232  libtorch-cuda12_4-shared-with-deps-pre-cxx11-build:
233    if: ${{ github.repository_owner == 'pytorch' }}
234    uses: ./.github/workflows/_binary-build-linux.yml
235    with:
236      PYTORCH_ROOT: /pytorch
237      BUILDER_ROOT: /builder
238      PACKAGE_TYPE: libtorch
239      # TODO: This is a legacy variable that we eventually want to get rid of in
240      #       favor of GPU_ARCH_VERSION
241      DESIRED_CUDA: cu124
242      GPU_ARCH_VERSION: 12.4
243      GPU_ARCH_TYPE: cuda
244      DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4
245      LIBTORCH_VARIANT: shared-with-deps
246      DESIRED_DEVTOOLSET: pre-cxx11
247      build_name: libtorch-cuda12_4-shared-with-deps-pre-cxx11
248      build_environment: linux-binary-libtorch-pre-cxx11
249    secrets:
250      github-token: ${{ secrets.GITHUB_TOKEN }}
251  libtorch-cuda12_4-shared-with-deps-pre-cxx11-test:  # Testing
252    if: ${{ github.repository_owner == 'pytorch' }}
253    needs: libtorch-cuda12_4-shared-with-deps-pre-cxx11-build
254    uses: ./.github/workflows/_binary-test-linux.yml
255    with:
256      PYTORCH_ROOT: /pytorch
257      BUILDER_ROOT: /builder
258      PACKAGE_TYPE: libtorch
259      # TODO: This is a legacy variable that we eventually want to get rid of in
260      #       favor of GPU_ARCH_VERSION
261      DESIRED_CUDA: cu124
262      GPU_ARCH_VERSION: 12.4
263      GPU_ARCH_TYPE: cuda
264      DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4
265      LIBTORCH_VARIANT: shared-with-deps
266      DESIRED_DEVTOOLSET: pre-cxx11
267      build_name: libtorch-cuda12_4-shared-with-deps-pre-cxx11
268      build_environment: linux-binary-libtorch-pre-cxx11
269      runs_on: linux.4xlarge.nvidia.gpu
270    secrets:
271      github-token: ${{ secrets.GITHUB_TOKEN }}
272  libtorch-cuda12_4-shared-with-deps-pre-cxx11-upload:  # Uploading
273    if: ${{ github.repository_owner == 'pytorch' }}
274    permissions:
275      id-token: write
276      contents: read
277    needs: libtorch-cuda12_4-shared-with-deps-pre-cxx11-test
278    with:
279      PYTORCH_ROOT: /pytorch
280      BUILDER_ROOT: /builder
281      PACKAGE_TYPE: libtorch
282      # TODO: This is a legacy variable that we eventually want to get rid of in
283      #       favor of GPU_ARCH_VERSION
284      DESIRED_CUDA: cu124
285      GPU_ARCH_VERSION: 12.4
286      GPU_ARCH_TYPE: cuda
287      DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4
288      LIBTORCH_VARIANT: shared-with-deps
289      DESIRED_DEVTOOLSET: pre-cxx11
290      build_name: libtorch-cuda12_4-shared-with-deps-pre-cxx11
291    secrets:
292      github-token: ${{ secrets.GITHUB_TOKEN }}
293      conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
294      conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
295    uses: ./.github/workflows/_binary-upload.yml
296
297  libtorch-rocm6_0-shared-with-deps-pre-cxx11-build:
298    if: ${{ github.repository_owner == 'pytorch' }}
299    uses: ./.github/workflows/_binary-build-linux.yml
300    with:
301      PYTORCH_ROOT: /pytorch
302      BUILDER_ROOT: /builder
303      PACKAGE_TYPE: libtorch
304      # TODO: This is a legacy variable that we eventually want to get rid of in
305      #       favor of GPU_ARCH_VERSION
306      DESIRED_CUDA: rocm6.0
307      GPU_ARCH_VERSION: 6.0
308      GPU_ARCH_TYPE: rocm
309      DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-2.4
310      LIBTORCH_VARIANT: shared-with-deps
311      DESIRED_DEVTOOLSET: pre-cxx11
312      build_name: libtorch-rocm6_0-shared-with-deps-pre-cxx11
313      build_environment: linux-binary-libtorch-pre-cxx11
314    secrets:
315      github-token: ${{ secrets.GITHUB_TOKEN }}
316  libtorch-rocm6_0-shared-with-deps-pre-cxx11-test:  # Testing
317    if: ${{ github.repository_owner == 'pytorch' }}
318    needs: libtorch-rocm6_0-shared-with-deps-pre-cxx11-build
319    runs-on: linux.rocm.gpu
320    timeout-minutes: 240
321    env:
322      PYTORCH_ROOT: /pytorch
323      BUILDER_ROOT: /builder
324      PACKAGE_TYPE: libtorch
325      # TODO: This is a legacy variable that we eventually want to get rid of in
326      #       favor of GPU_ARCH_VERSION
327      DESIRED_CUDA: rocm6.0
328      GPU_ARCH_VERSION: 6.0
329      GPU_ARCH_TYPE: rocm
330      SKIP_ALL_TESTS: 1
331      DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-2.4
332      LIBTORCH_VARIANT: shared-with-deps
333      DESIRED_DEVTOOLSET: pre-cxx11
334    steps:
335      - name: Setup ROCm
336        uses: ./.github/actions/setup-rocm
337      - uses: actions/download-artifact@v3
338        name: Download Build Artifacts
339        with:
340          name: libtorch-rocm6_0-shared-with-deps-pre-cxx11
341          path: "${{ runner.temp }}/artifacts/"
342      - name: Checkout PyTorch
343        uses: malfet/checkout@silent-checkout
344        with:
345          submodules: recursive
346          path: pytorch
347          quiet-checkout: true
348      - name: Clean PyTorch checkout
349        run: |
350          # Remove any artifacts from the previous checkouts
351          git clean -fxd
352        working-directory: pytorch
353      - name: Checkout pytorch/builder
354        uses: malfet/checkout@silent-checkout
355        with:
356          ref: release/2.4
357          submodules: recursive
358          repository: pytorch/builder
359          path: builder
360          quiet-checkout: true
361      - name: Clean pytorch/builder checkout
362        run: |
363          # Remove any artifacts from the previous checkouts
364          git clean -fxd
365        working-directory: builder
366      - name: ROCm set GPU_FLAG
367        run: |
368          echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add daemon" >> "${GITHUB_ENV}"
369      - name: Pull Docker image
370        uses: pytorch/test-infra/.github/actions/pull-docker-image@main
371        with:
372          docker-image: pytorch/manylinux-builder:rocm6.0-2.4
373      - name: Test Pytorch binary
374        uses: ./pytorch/.github/actions/test-pytorch-binary
375      - name: Teardown ROCm
376        uses: ./.github/actions/teardown-rocm
377  libtorch-rocm6_0-shared-with-deps-pre-cxx11-upload:  # Uploading
378    if: ${{ github.repository_owner == 'pytorch' }}
379    permissions:
380      id-token: write
381      contents: read
382    needs: libtorch-rocm6_0-shared-with-deps-pre-cxx11-test
383    with:
384      PYTORCH_ROOT: /pytorch
385      BUILDER_ROOT: /builder
386      PACKAGE_TYPE: libtorch
387      # TODO: This is a legacy variable that we eventually want to get rid of in
388      #       favor of GPU_ARCH_VERSION
389      DESIRED_CUDA: rocm6.0
390      GPU_ARCH_VERSION: 6.0
391      GPU_ARCH_TYPE: rocm
392      DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-2.4
393      LIBTORCH_VARIANT: shared-with-deps
394      DESIRED_DEVTOOLSET: pre-cxx11
395      build_name: libtorch-rocm6_0-shared-with-deps-pre-cxx11
396    secrets:
397      github-token: ${{ secrets.GITHUB_TOKEN }}
398      conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
399      conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
400    uses: ./.github/workflows/_binary-upload.yml
401
402  libtorch-rocm6_1-shared-with-deps-pre-cxx11-build:
403    if: ${{ github.repository_owner == 'pytorch' }}
404    uses: ./.github/workflows/_binary-build-linux.yml
405    with:
406      PYTORCH_ROOT: /pytorch
407      BUILDER_ROOT: /builder
408      PACKAGE_TYPE: libtorch
409      # TODO: This is a legacy variable that we eventually want to get rid of in
410      #       favor of GPU_ARCH_VERSION
411      DESIRED_CUDA: rocm6.1
412      GPU_ARCH_VERSION: 6.1
413      GPU_ARCH_TYPE: rocm
414      DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-2.4
415      LIBTORCH_VARIANT: shared-with-deps
416      DESIRED_DEVTOOLSET: pre-cxx11
417      build_name: libtorch-rocm6_1-shared-with-deps-pre-cxx11
418      build_environment: linux-binary-libtorch-pre-cxx11
419    secrets:
420      github-token: ${{ secrets.GITHUB_TOKEN }}
421  libtorch-rocm6_1-shared-with-deps-pre-cxx11-test:  # Testing
422    if: ${{ github.repository_owner == 'pytorch' }}
423    needs: libtorch-rocm6_1-shared-with-deps-pre-cxx11-build
424    runs-on: linux.rocm.gpu
425    timeout-minutes: 240
426    env:
427      PYTORCH_ROOT: /pytorch
428      BUILDER_ROOT: /builder
429      PACKAGE_TYPE: libtorch
430      # TODO: This is a legacy variable that we eventually want to get rid of in
431      #       favor of GPU_ARCH_VERSION
432      DESIRED_CUDA: rocm6.1
433      GPU_ARCH_VERSION: 6.1
434      GPU_ARCH_TYPE: rocm
435      SKIP_ALL_TESTS: 1
436      DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-2.4
437      LIBTORCH_VARIANT: shared-with-deps
438      DESIRED_DEVTOOLSET: pre-cxx11
439    steps:
440      - name: Setup ROCm
441        uses: ./.github/actions/setup-rocm
442      - uses: actions/download-artifact@v3
443        name: Download Build Artifacts
444        with:
445          name: libtorch-rocm6_1-shared-with-deps-pre-cxx11
446          path: "${{ runner.temp }}/artifacts/"
447      - name: Checkout PyTorch
448        uses: malfet/checkout@silent-checkout
449        with:
450          submodules: recursive
451          path: pytorch
452          quiet-checkout: true
453      - name: Clean PyTorch checkout
454        run: |
455          # Remove any artifacts from the previous checkouts
456          git clean -fxd
457        working-directory: pytorch
458      - name: Checkout pytorch/builder
459        uses: malfet/checkout@silent-checkout
460        with:
461          ref: release/2.4
462          submodules: recursive
463          repository: pytorch/builder
464          path: builder
465          quiet-checkout: true
466      - name: Clean pytorch/builder checkout
467        run: |
468          # Remove any artifacts from the previous checkouts
469          git clean -fxd
470        working-directory: builder
471      - name: ROCm set GPU_FLAG
472        run: |
473          echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add daemon" >> "${GITHUB_ENV}"
474      - name: Pull Docker image
475        uses: pytorch/test-infra/.github/actions/pull-docker-image@main
476        with:
477          docker-image: pytorch/manylinux-builder:rocm6.1-2.4
478      - name: Test Pytorch binary
479        uses: ./pytorch/.github/actions/test-pytorch-binary
480      - name: Teardown ROCm
481        uses: ./.github/actions/teardown-rocm
482  libtorch-rocm6_1-shared-with-deps-pre-cxx11-upload:  # Uploading
483    if: ${{ github.repository_owner == 'pytorch' }}
484    permissions:
485      id-token: write
486      contents: read
487    needs: libtorch-rocm6_1-shared-with-deps-pre-cxx11-test
488    with:
489      PYTORCH_ROOT: /pytorch
490      BUILDER_ROOT: /builder
491      PACKAGE_TYPE: libtorch
492      # TODO: This is a legacy variable that we eventually want to get rid of in
493      #       favor of GPU_ARCH_VERSION
494      DESIRED_CUDA: rocm6.1
495      GPU_ARCH_VERSION: 6.1
496      GPU_ARCH_TYPE: rocm
497      DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-2.4
498      LIBTORCH_VARIANT: shared-with-deps
499      DESIRED_DEVTOOLSET: pre-cxx11
500      build_name: libtorch-rocm6_1-shared-with-deps-pre-cxx11
501    secrets:
502      github-token: ${{ secrets.GITHUB_TOKEN }}
503      conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
504      conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
505    uses: ./.github/workflows/_binary-upload.yml
506