Lines Matching +full:pytorchbot +full:- +full:token
5 name: linux-binary-manywheel
12 - nightly
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_wheel/*'
23 ALPINE_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/alpine"
25 AWS_DEFAULT_REGION: us-east-1
27 BUILD_ENVIRONMENT: linux-binary-manywheel
36 …group: linux-binary-manywheel-${{ github.event.pull_request.number || github.ref_name }}-${{ githu…
37 cancel-in-progress: true
40 manywheel-py3_8-cpu-build:
42 uses: ./.github/workflows/_binary-build-linux.yml
51 DOCKER_IMAGE: pytorch/manylinux-builder:cpu-2.4
53 build_name: manywheel-py3_8-cpu
54 build_environment: linux-binary-manywheel
56 github-token: ${{ secrets.GITHUB_TOKEN }}
57 manywheel-py3_8-cpu-test: # Testing
59 needs: manywheel-py3_8-cpu-build
60 uses: ./.github/workflows/_binary-test-linux.yml
69 DOCKER_IMAGE: pytorch/manylinux-builder:cpu-2.4
71 build_name: manywheel-py3_8-cpu
72 build_environment: linux-binary-manywheel
75 github-token: ${{ secrets.GITHUB_TOKEN }}
76 manywheel-py3_8-cpu-upload: # Uploading
79 id-token: write
81 needs: manywheel-py3_8-cpu-test
90 DOCKER_IMAGE: pytorch/manylinux-builder:cpu-2.4
92 build_name: manywheel-py3_8-cpu
94 github-token: ${{ secrets.GITHUB_TOKEN }}
95 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
96 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
97 uses: ./.github/workflows/_binary-upload.yml
99 manywheel-py3_8-cpu-cxx11-abi-build:
101 uses: ./.github/workflows/_binary-build-linux.yml
108 DESIRED_CUDA: cpu-cxx11-abi
109 GPU_ARCH_TYPE: cpu-cxx11-abi
110 DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-2.4
111 DESIRED_DEVTOOLSET: cxx11-abi
113 build_name: manywheel-py3_8-cpu-cxx11-abi
114 build_environment: linux-binary-manywheel
116 github-token: ${{ secrets.GITHUB_TOKEN }}
117 manywheel-py3_8-cpu-cxx11-abi-test: # Testing
119 needs: manywheel-py3_8-cpu-cxx11-abi-build
120 uses: ./.github/workflows/_binary-test-linux.yml
127 DESIRED_CUDA: cpu-cxx11-abi
128 GPU_ARCH_TYPE: cpu-cxx11-abi
129 DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-2.4
130 DESIRED_DEVTOOLSET: cxx11-abi
132 build_name: manywheel-py3_8-cpu-cxx11-abi
133 build_environment: linux-binary-manywheel
136 github-token: ${{ secrets.GITHUB_TOKEN }}
137 manywheel-py3_8-cpu-cxx11-abi-upload: # Uploading
140 id-token: write
142 needs: manywheel-py3_8-cpu-cxx11-abi-test
149 DESIRED_CUDA: cpu-cxx11-abi
150 GPU_ARCH_TYPE: cpu-cxx11-abi
151 DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-2.4
152 DESIRED_DEVTOOLSET: cxx11-abi
154 build_name: manywheel-py3_8-cpu-cxx11-abi
156 github-token: ${{ secrets.GITHUB_TOKEN }}
157 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
158 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
159 uses: ./.github/workflows/_binary-upload.yml
161 manywheel-py3_8-cuda11_8-build:
163 uses: ./.github/workflows/_binary-build-linux.yml
173 DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4
175 build_name: manywheel-py3_8-cuda11_8
176 build_environment: linux-binary-manywheel
177 …-cuda-nvrtc-cu11==11.8.89; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cu…
179 github-token: ${{ secrets.GITHUB_TOKEN }}
180 manywheel-py3_8-cuda11_8-test: # Testing
182 needs: manywheel-py3_8-cuda11_8-build
183 uses: ./.github/workflows/_binary-test-linux.yml
193 DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4
195 build_name: manywheel-py3_8-cuda11_8
196 build_environment: linux-binary-manywheel
199 github-token: ${{ secrets.GITHUB_TOKEN }}
200 manywheel-py3_8-cuda11_8-upload: # Uploading
203 id-token: write
205 needs: manywheel-py3_8-cuda11_8-test
215 DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4
217 build_name: manywheel-py3_8-cuda11_8
219 github-token: ${{ secrets.GITHUB_TOKEN }}
220 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
221 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
222 uses: ./.github/workflows/_binary-upload.yml
224 manywheel-py3_8-cuda12_1-build:
226 uses: ./.github/workflows/_binary-build-linux.yml
236 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
238 build_name: manywheel-py3_8-cuda12_1
239 build_environment: linux-binary-manywheel
240 …-cuda-nvrtc-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-c…
242 github-token: ${{ secrets.GITHUB_TOKEN }}
243 manywheel-py3_8-cuda12_1-test: # Testing
245 needs: manywheel-py3_8-cuda12_1-build
246 uses: ./.github/workflows/_binary-test-linux.yml
256 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
258 build_name: manywheel-py3_8-cuda12_1
259 build_environment: linux-binary-manywheel
262 github-token: ${{ secrets.GITHUB_TOKEN }}
263 manywheel-py3_8-cuda12_1-upload: # Uploading
266 id-token: write
268 needs: manywheel-py3_8-cuda12_1-test
278 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
280 build_name: manywheel-py3_8-cuda12_1
282 github-token: ${{ secrets.GITHUB_TOKEN }}
283 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
284 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
285 uses: ./.github/workflows/_binary-upload.yml
287 manywheel-py3_8-cuda12_4-build:
289 uses: ./.github/workflows/_binary-build-linux.yml
299 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4
301 build_name: manywheel-py3_8-cuda12_4
302 build_environment: linux-binary-manywheel
303 …-cuda-nvrtc-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cu…
305 github-token: ${{ secrets.GITHUB_TOKEN }}
306 manywheel-py3_8-cuda12_4-test: # Testing
308 needs: manywheel-py3_8-cuda12_4-build
309 uses: ./.github/workflows/_binary-test-linux.yml
319 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4
321 build_name: manywheel-py3_8-cuda12_4
322 build_environment: linux-binary-manywheel
325 github-token: ${{ secrets.GITHUB_TOKEN }}
326 manywheel-py3_8-cuda12_4-upload: # Uploading
329 id-token: write
331 needs: manywheel-py3_8-cuda12_4-test
341 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4
343 build_name: manywheel-py3_8-cuda12_4
345 github-token: ${{ secrets.GITHUB_TOKEN }}
346 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
347 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
348 uses: ./.github/workflows/_binary-upload.yml
350 manywheel-py3_8-rocm6_0-build:
352 uses: ./.github/workflows/_binary-build-linux.yml
362 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-2.4
364 build_name: manywheel-py3_8-rocm6_0
365 build_environment: linux-binary-manywheel
367 github-token: ${{ secrets.GITHUB_TOKEN }}
368 manywheel-py3_8-rocm6_0-test: # Testing
370 needs: manywheel-py3_8-rocm6_0-build
371 runs-on: linux.rocm.gpu
372 timeout-minutes: 240
383 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-2.4
386 - name: Setup ROCm
387 uses: ./.github/actions/setup-rocm
388 - uses: actions/download-artifact@v3
391 name: manywheel-py3_8-rocm6_0
393 - name: Checkout PyTorch
394 uses: malfet/checkout@silent-checkout
398 quiet-checkout: true
399 - name: Clean PyTorch checkout
402 git clean -fxd
403 working-directory: pytorch
404 - name: Checkout pytorch/builder
405 uses: malfet/checkout@silent-checkout
411 quiet-checkout: true
412 - name: Clean pytorch/builder checkout
415 git clean -fxd
416 working-directory: builder
417 - name: ROCm set GPU_FLAG
419 …echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add…
420 - name: Pull Docker image
421 uses: pytorch/test-infra/.github/actions/pull-docker-image@main
423 docker-image: pytorch/manylinux-builder:rocm6.0-2.4
424 - name: Test Pytorch binary
425 uses: ./pytorch/.github/actions/test-pytorch-binary
426 - name: Teardown ROCm
427 uses: ./.github/actions/teardown-rocm
428 manywheel-py3_8-rocm6_0-upload: # Uploading
431 id-token: write
433 needs: manywheel-py3_8-rocm6_0-test
443 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-2.4
445 build_name: manywheel-py3_8-rocm6_0
447 github-token: ${{ secrets.GITHUB_TOKEN }}
448 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
449 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
450 uses: ./.github/workflows/_binary-upload.yml
452 manywheel-py3_8-rocm6_1-build:
454 uses: ./.github/workflows/_binary-build-linux.yml
464 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-2.4
466 build_name: manywheel-py3_8-rocm6_1
467 build_environment: linux-binary-manywheel
469 github-token: ${{ secrets.GITHUB_TOKEN }}
470 manywheel-py3_8-rocm6_1-test: # Testing
472 needs: manywheel-py3_8-rocm6_1-build
473 runs-on: linux.rocm.gpu
474 timeout-minutes: 240
485 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-2.4
488 - name: Setup ROCm
489 uses: ./.github/actions/setup-rocm
490 - uses: actions/download-artifact@v3
493 name: manywheel-py3_8-rocm6_1
495 - name: Checkout PyTorch
496 uses: malfet/checkout@silent-checkout
500 quiet-checkout: true
501 - name: Clean PyTorch checkout
504 git clean -fxd
505 working-directory: pytorch
506 - name: Checkout pytorch/builder
507 uses: malfet/checkout@silent-checkout
513 quiet-checkout: true
514 - name: Clean pytorch/builder checkout
517 git clean -fxd
518 working-directory: builder
519 - name: ROCm set GPU_FLAG
521 …echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add…
522 - name: Pull Docker image
523 uses: pytorch/test-infra/.github/actions/pull-docker-image@main
525 docker-image: pytorch/manylinux-builder:rocm6.1-2.4
526 - name: Test Pytorch binary
527 uses: ./pytorch/.github/actions/test-pytorch-binary
528 - name: Teardown ROCm
529 uses: ./.github/actions/teardown-rocm
530 manywheel-py3_8-rocm6_1-upload: # Uploading
533 id-token: write
535 needs: manywheel-py3_8-rocm6_1-test
545 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-2.4
547 build_name: manywheel-py3_8-rocm6_1
549 github-token: ${{ secrets.GITHUB_TOKEN }}
550 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
551 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
552 uses: ./.github/workflows/_binary-upload.yml
554 manywheel-py3_9-cpu-build:
556 uses: ./.github/workflows/_binary-build-linux.yml
565 DOCKER_IMAGE: pytorch/manylinux-builder:cpu-2.4
567 build_name: manywheel-py3_9-cpu
568 build_environment: linux-binary-manywheel
570 github-token: ${{ secrets.GITHUB_TOKEN }}
571 manywheel-py3_9-cpu-test: # Testing
573 needs: manywheel-py3_9-cpu-build
574 uses: ./.github/workflows/_binary-test-linux.yml
583 DOCKER_IMAGE: pytorch/manylinux-builder:cpu-2.4
585 build_name: manywheel-py3_9-cpu
586 build_environment: linux-binary-manywheel
589 github-token: ${{ secrets.GITHUB_TOKEN }}
590 manywheel-py3_9-cpu-upload: # Uploading
593 id-token: write
595 needs: manywheel-py3_9-cpu-test
604 DOCKER_IMAGE: pytorch/manylinux-builder:cpu-2.4
606 build_name: manywheel-py3_9-cpu
608 github-token: ${{ secrets.GITHUB_TOKEN }}
609 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
610 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
611 uses: ./.github/workflows/_binary-upload.yml
613 manywheel-py3_9-cpu-cxx11-abi-build:
615 uses: ./.github/workflows/_binary-build-linux.yml
622 DESIRED_CUDA: cpu-cxx11-abi
623 GPU_ARCH_TYPE: cpu-cxx11-abi
624 DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-2.4
625 DESIRED_DEVTOOLSET: cxx11-abi
627 build_name: manywheel-py3_9-cpu-cxx11-abi
628 build_environment: linux-binary-manywheel
630 github-token: ${{ secrets.GITHUB_TOKEN }}
631 manywheel-py3_9-cpu-cxx11-abi-test: # Testing
633 needs: manywheel-py3_9-cpu-cxx11-abi-build
634 uses: ./.github/workflows/_binary-test-linux.yml
641 DESIRED_CUDA: cpu-cxx11-abi
642 GPU_ARCH_TYPE: cpu-cxx11-abi
643 DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-2.4
644 DESIRED_DEVTOOLSET: cxx11-abi
646 build_name: manywheel-py3_9-cpu-cxx11-abi
647 build_environment: linux-binary-manywheel
650 github-token: ${{ secrets.GITHUB_TOKEN }}
651 manywheel-py3_9-cpu-cxx11-abi-upload: # Uploading
654 id-token: write
656 needs: manywheel-py3_9-cpu-cxx11-abi-test
663 DESIRED_CUDA: cpu-cxx11-abi
664 GPU_ARCH_TYPE: cpu-cxx11-abi
665 DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-2.4
666 DESIRED_DEVTOOLSET: cxx11-abi
668 build_name: manywheel-py3_9-cpu-cxx11-abi
670 github-token: ${{ secrets.GITHUB_TOKEN }}
671 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
672 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
673 uses: ./.github/workflows/_binary-upload.yml
675 manywheel-py3_9-cuda11_8-build:
677 uses: ./.github/workflows/_binary-build-linux.yml
687 DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4
689 build_name: manywheel-py3_9-cuda11_8
690 build_environment: linux-binary-manywheel
691 …-cuda-nvrtc-cu11==11.8.89; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cu…
693 github-token: ${{ secrets.GITHUB_TOKEN }}
694 manywheel-py3_9-cuda11_8-test: # Testing
696 needs: manywheel-py3_9-cuda11_8-build
697 uses: ./.github/workflows/_binary-test-linux.yml
707 DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4
709 build_name: manywheel-py3_9-cuda11_8
710 build_environment: linux-binary-manywheel
713 github-token: ${{ secrets.GITHUB_TOKEN }}
714 manywheel-py3_9-cuda11_8-upload: # Uploading
717 id-token: write
719 needs: manywheel-py3_9-cuda11_8-test
729 DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4
731 build_name: manywheel-py3_9-cuda11_8
733 github-token: ${{ secrets.GITHUB_TOKEN }}
734 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
735 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
736 uses: ./.github/workflows/_binary-upload.yml
738 manywheel-py3_9-cuda12_1-build:
740 uses: ./.github/workflows/_binary-build-linux.yml
750 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
752 build_name: manywheel-py3_9-cuda12_1
753 build_environment: linux-binary-manywheel
754 …-cuda-nvrtc-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-c…
756 github-token: ${{ secrets.GITHUB_TOKEN }}
757 manywheel-py3_9-cuda12_1-test: # Testing
759 needs: manywheel-py3_9-cuda12_1-build
760 uses: ./.github/workflows/_binary-test-linux.yml
770 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
772 build_name: manywheel-py3_9-cuda12_1
773 build_environment: linux-binary-manywheel
776 github-token: ${{ secrets.GITHUB_TOKEN }}
777 manywheel-py3_9-cuda12_1-upload: # Uploading
780 id-token: write
782 needs: manywheel-py3_9-cuda12_1-test
792 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
794 build_name: manywheel-py3_9-cuda12_1
796 github-token: ${{ secrets.GITHUB_TOKEN }}
797 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
798 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
799 uses: ./.github/workflows/_binary-upload.yml
801 manywheel-py3_9-cuda12_4-build:
803 uses: ./.github/workflows/_binary-build-linux.yml
813 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4
815 build_name: manywheel-py3_9-cuda12_4
816 build_environment: linux-binary-manywheel
817 …-cuda-nvrtc-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cu…
819 github-token: ${{ secrets.GITHUB_TOKEN }}
820 manywheel-py3_9-cuda12_4-test: # Testing
822 needs: manywheel-py3_9-cuda12_4-build
823 uses: ./.github/workflows/_binary-test-linux.yml
833 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4
835 build_name: manywheel-py3_9-cuda12_4
836 build_environment: linux-binary-manywheel
839 github-token: ${{ secrets.GITHUB_TOKEN }}
840 manywheel-py3_9-cuda12_4-upload: # Uploading
843 id-token: write
845 needs: manywheel-py3_9-cuda12_4-test
855 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4
857 build_name: manywheel-py3_9-cuda12_4
859 github-token: ${{ secrets.GITHUB_TOKEN }}
860 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
861 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
862 uses: ./.github/workflows/_binary-upload.yml
864 manywheel-py3_9-rocm6_0-build:
866 uses: ./.github/workflows/_binary-build-linux.yml
876 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-2.4
878 build_name: manywheel-py3_9-rocm6_0
879 build_environment: linux-binary-manywheel
881 github-token: ${{ secrets.GITHUB_TOKEN }}
882 manywheel-py3_9-rocm6_0-test: # Testing
884 needs: manywheel-py3_9-rocm6_0-build
885 runs-on: linux.rocm.gpu
886 timeout-minutes: 240
897 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-2.4
900 - name: Setup ROCm
901 uses: ./.github/actions/setup-rocm
902 - uses: actions/download-artifact@v3
905 name: manywheel-py3_9-rocm6_0
907 - name: Checkout PyTorch
908 uses: malfet/checkout@silent-checkout
912 quiet-checkout: true
913 - name: Clean PyTorch checkout
916 git clean -fxd
917 working-directory: pytorch
918 - name: Checkout pytorch/builder
919 uses: malfet/checkout@silent-checkout
925 quiet-checkout: true
926 - name: Clean pytorch/builder checkout
929 git clean -fxd
930 working-directory: builder
931 - name: ROCm set GPU_FLAG
933 …echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add…
934 - name: Pull Docker image
935 uses: pytorch/test-infra/.github/actions/pull-docker-image@main
937 docker-image: pytorch/manylinux-builder:rocm6.0-2.4
938 - name: Test Pytorch binary
939 uses: ./pytorch/.github/actions/test-pytorch-binary
940 - name: Teardown ROCm
941 uses: ./.github/actions/teardown-rocm
942 manywheel-py3_9-rocm6_0-upload: # Uploading
945 id-token: write
947 needs: manywheel-py3_9-rocm6_0-test
957 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-2.4
959 build_name: manywheel-py3_9-rocm6_0
961 github-token: ${{ secrets.GITHUB_TOKEN }}
962 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
963 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
964 uses: ./.github/workflows/_binary-upload.yml
966 manywheel-py3_9-rocm6_1-build:
968 uses: ./.github/workflows/_binary-build-linux.yml
978 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-2.4
980 build_name: manywheel-py3_9-rocm6_1
981 build_environment: linux-binary-manywheel
983 github-token: ${{ secrets.GITHUB_TOKEN }}
984 manywheel-py3_9-rocm6_1-test: # Testing
986 needs: manywheel-py3_9-rocm6_1-build
987 runs-on: linux.rocm.gpu
988 timeout-minutes: 240
999 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-2.4
1002 - name: Setup ROCm
1003 uses: ./.github/actions/setup-rocm
1004 - uses: actions/download-artifact@v3
1007 name: manywheel-py3_9-rocm6_1
1009 - name: Checkout PyTorch
1010 uses: malfet/checkout@silent-checkout
1014 quiet-checkout: true
1015 - name: Clean PyTorch checkout
1018 git clean -fxd
1019 working-directory: pytorch
1020 - name: Checkout pytorch/builder
1021 uses: malfet/checkout@silent-checkout
1027 quiet-checkout: true
1028 - name: Clean pytorch/builder checkout
1031 git clean -fxd
1032 working-directory: builder
1033 - name: ROCm set GPU_FLAG
1035 …echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add…
1036 - name: Pull Docker image
1037 uses: pytorch/test-infra/.github/actions/pull-docker-image@main
1039 docker-image: pytorch/manylinux-builder:rocm6.1-2.4
1040 - name: Test Pytorch binary
1041 uses: ./pytorch/.github/actions/test-pytorch-binary
1042 - name: Teardown ROCm
1043 uses: ./.github/actions/teardown-rocm
1044 manywheel-py3_9-rocm6_1-upload: # Uploading
1047 id-token: write
1049 needs: manywheel-py3_9-rocm6_1-test
1059 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-2.4
1061 build_name: manywheel-py3_9-rocm6_1
1063 github-token: ${{ secrets.GITHUB_TOKEN }}
1064 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
1065 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
1066 uses: ./.github/workflows/_binary-upload.yml
1068 manywheel-py3_10-cpu-build:
1070 uses: ./.github/workflows/_binary-build-linux.yml
1079 DOCKER_IMAGE: pytorch/manylinux-builder:cpu-2.4
1081 build_name: manywheel-py3_10-cpu
1082 build_environment: linux-binary-manywheel
1084 github-token: ${{ secrets.GITHUB_TOKEN }}
1085 manywheel-py3_10-cpu-test: # Testing
1087 needs: manywheel-py3_10-cpu-build
1088 uses: ./.github/workflows/_binary-test-linux.yml
1097 DOCKER_IMAGE: pytorch/manylinux-builder:cpu-2.4
1099 build_name: manywheel-py3_10-cpu
1100 build_environment: linux-binary-manywheel
1103 github-token: ${{ secrets.GITHUB_TOKEN }}
1104 manywheel-py3_10-cpu-upload: # Uploading
1107 id-token: write
1109 needs: manywheel-py3_10-cpu-test
1118 DOCKER_IMAGE: pytorch/manylinux-builder:cpu-2.4
1120 build_name: manywheel-py3_10-cpu
1122 github-token: ${{ secrets.GITHUB_TOKEN }}
1123 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
1124 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
1125 uses: ./.github/workflows/_binary-upload.yml
1127 manywheel-py3_10-cpu-cxx11-abi-build:
1129 uses: ./.github/workflows/_binary-build-linux.yml
1136 DESIRED_CUDA: cpu-cxx11-abi
1137 GPU_ARCH_TYPE: cpu-cxx11-abi
1138 DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-2.4
1139 DESIRED_DEVTOOLSET: cxx11-abi
1141 build_name: manywheel-py3_10-cpu-cxx11-abi
1142 build_environment: linux-binary-manywheel
1144 github-token: ${{ secrets.GITHUB_TOKEN }}
1145 manywheel-py3_10-cpu-cxx11-abi-test: # Testing
1147 needs: manywheel-py3_10-cpu-cxx11-abi-build
1148 uses: ./.github/workflows/_binary-test-linux.yml
1155 DESIRED_CUDA: cpu-cxx11-abi
1156 GPU_ARCH_TYPE: cpu-cxx11-abi
1157 DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-2.4
1158 DESIRED_DEVTOOLSET: cxx11-abi
1160 build_name: manywheel-py3_10-cpu-cxx11-abi
1161 build_environment: linux-binary-manywheel
1164 github-token: ${{ secrets.GITHUB_TOKEN }}
1165 manywheel-py3_10-cpu-cxx11-abi-upload: # Uploading
1168 id-token: write
1170 needs: manywheel-py3_10-cpu-cxx11-abi-test
1177 DESIRED_CUDA: cpu-cxx11-abi
1178 GPU_ARCH_TYPE: cpu-cxx11-abi
1179 DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-2.4
1180 DESIRED_DEVTOOLSET: cxx11-abi
1182 build_name: manywheel-py3_10-cpu-cxx11-abi
1184 github-token: ${{ secrets.GITHUB_TOKEN }}
1185 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
1186 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
1187 uses: ./.github/workflows/_binary-upload.yml
1189 manywheel-py3_10-cuda11_8-build:
1191 uses: ./.github/workflows/_binary-build-linux.yml
1201 DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4
1203 build_name: manywheel-py3_10-cuda11_8
1204 build_environment: linux-binary-manywheel
1205 …-cuda-nvrtc-cu11==11.8.89; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cu…
1207 github-token: ${{ secrets.GITHUB_TOKEN }}
1208 manywheel-py3_10-cuda11_8-test: # Testing
1210 needs: manywheel-py3_10-cuda11_8-build
1211 uses: ./.github/workflows/_binary-test-linux.yml
1221 DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4
1223 build_name: manywheel-py3_10-cuda11_8
1224 build_environment: linux-binary-manywheel
1227 github-token: ${{ secrets.GITHUB_TOKEN }}
1228 manywheel-py3_10-cuda11_8-upload: # Uploading
1231 id-token: write
1233 needs: manywheel-py3_10-cuda11_8-test
1243 DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4
1245 build_name: manywheel-py3_10-cuda11_8
1247 github-token: ${{ secrets.GITHUB_TOKEN }}
1248 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
1249 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
1250 uses: ./.github/workflows/_binary-upload.yml
1252 manywheel-py3_10-cuda12_1-build:
1254 uses: ./.github/workflows/_binary-build-linux.yml
1264 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
1266 build_name: manywheel-py3_10-cuda12_1
1267 build_environment: linux-binary-manywheel
1268 …-cuda-nvrtc-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-c…
1270 github-token: ${{ secrets.GITHUB_TOKEN }}
1271 manywheel-py3_10-cuda12_1-test: # Testing
1273 needs: manywheel-py3_10-cuda12_1-build
1274 uses: ./.github/workflows/_binary-test-linux.yml
1284 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
1286 build_name: manywheel-py3_10-cuda12_1
1287 build_environment: linux-binary-manywheel
1290 github-token: ${{ secrets.GITHUB_TOKEN }}
1291 manywheel-py3_10-cuda12_1-upload: # Uploading
1294 id-token: write
1296 needs: manywheel-py3_10-cuda12_1-test
1306 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
1308 build_name: manywheel-py3_10-cuda12_1
1310 github-token: ${{ secrets.GITHUB_TOKEN }}
1311 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
1312 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
1313 uses: ./.github/workflows/_binary-upload.yml
1315 manywheel-py3_10-cuda12_1-full-build:
1317 uses: ./.github/workflows/_binary-build-linux.yml
1327 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
1329 build_name: manywheel-py3_10-cuda12_1-full
1330 build_environment: linux-binary-manywheel
1332 github-token: ${{ secrets.GITHUB_TOKEN }}
1333 manywheel-py3_10-cuda12_1-full-test: # Testing
1335 needs: manywheel-py3_10-cuda12_1-full-build
1336 uses: ./.github/workflows/_binary-test-linux.yml
1346 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
1348 build_name: manywheel-py3_10-cuda12_1-full
1349 build_environment: linux-binary-manywheel
1352 github-token: ${{ secrets.GITHUB_TOKEN }}
1353 manywheel-py3_10-cuda12_1-full-upload: # Uploading
1356 id-token: write
1358 needs: manywheel-py3_10-cuda12_1-full-test
1368 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
1370 build_name: manywheel-py3_10-cuda12_1-full
1372 github-token: ${{ secrets.GITHUB_TOKEN }}
1373 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
1374 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
1375 uses: ./.github/workflows/_binary-upload.yml
1377 manywheel-py3_10-cuda12_4-build:
1379 uses: ./.github/workflows/_binary-build-linux.yml
1389 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4
1391 build_name: manywheel-py3_10-cuda12_4
1392 build_environment: linux-binary-manywheel
1393 …-cuda-nvrtc-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cu…
1395 github-token: ${{ secrets.GITHUB_TOKEN }}
1396 manywheel-py3_10-cuda12_4-test: # Testing
1398 needs: manywheel-py3_10-cuda12_4-build
1399 uses: ./.github/workflows/_binary-test-linux.yml
1409 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4
1411 build_name: manywheel-py3_10-cuda12_4
1412 build_environment: linux-binary-manywheel
1415 github-token: ${{ secrets.GITHUB_TOKEN }}
1416 manywheel-py3_10-cuda12_4-upload: # Uploading
1419 id-token: write
1421 needs: manywheel-py3_10-cuda12_4-test
1431 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4
1433 build_name: manywheel-py3_10-cuda12_4
1435 github-token: ${{ secrets.GITHUB_TOKEN }}
1436 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
1437 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
1438 uses: ./.github/workflows/_binary-upload.yml
1440 manywheel-py3_10-rocm6_0-build:
1442 uses: ./.github/workflows/_binary-build-linux.yml
1452 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-2.4
1454 build_name: manywheel-py3_10-rocm6_0
1455 build_environment: linux-binary-manywheel
1457 github-token: ${{ secrets.GITHUB_TOKEN }}
1458 manywheel-py3_10-rocm6_0-test: # Testing
1460 needs: manywheel-py3_10-rocm6_0-build
1461 runs-on: linux.rocm.gpu
1462 timeout-minutes: 240
1473 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-2.4
1476 - name: Setup ROCm
1477 uses: ./.github/actions/setup-rocm
1478 - uses: actions/download-artifact@v3
1481 name: manywheel-py3_10-rocm6_0
1483 - name: Checkout PyTorch
1484 uses: malfet/checkout@silent-checkout
1488 quiet-checkout: true
1489 - name: Clean PyTorch checkout
1492 git clean -fxd
1493 working-directory: pytorch
1494 - name: Checkout pytorch/builder
1495 uses: malfet/checkout@silent-checkout
1501 quiet-checkout: true
1502 - name: Clean pytorch/builder checkout
1505 git clean -fxd
1506 working-directory: builder
1507 - name: ROCm set GPU_FLAG
1509 …echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add…
1510 - name: Pull Docker image
1511 uses: pytorch/test-infra/.github/actions/pull-docker-image@main
1513 docker-image: pytorch/manylinux-builder:rocm6.0-2.4
1514 - name: Test Pytorch binary
1515 uses: ./pytorch/.github/actions/test-pytorch-binary
1516 - name: Teardown ROCm
1517 uses: ./.github/actions/teardown-rocm
1518 manywheel-py3_10-rocm6_0-upload: # Uploading
1521 id-token: write
1523 needs: manywheel-py3_10-rocm6_0-test
1533 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-2.4
1535 build_name: manywheel-py3_10-rocm6_0
1537 github-token: ${{ secrets.GITHUB_TOKEN }}
1538 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
1539 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
1540 uses: ./.github/workflows/_binary-upload.yml
1542 manywheel-py3_10-rocm6_1-build:
1544 uses: ./.github/workflows/_binary-build-linux.yml
1554 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-2.4
1556 build_name: manywheel-py3_10-rocm6_1
1557 build_environment: linux-binary-manywheel
1559 github-token: ${{ secrets.GITHUB_TOKEN }}
1560 manywheel-py3_10-rocm6_1-test: # Testing
1562 needs: manywheel-py3_10-rocm6_1-build
1563 runs-on: linux.rocm.gpu
1564 timeout-minutes: 240
1575 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-2.4
1578 - name: Setup ROCm
1579 uses: ./.github/actions/setup-rocm
1580 - uses: actions/download-artifact@v3
1583 name: manywheel-py3_10-rocm6_1
1585 - name: Checkout PyTorch
1586 uses: malfet/checkout@silent-checkout
1590 quiet-checkout: true
1591 - name: Clean PyTorch checkout
1594 git clean -fxd
1595 working-directory: pytorch
1596 - name: Checkout pytorch/builder
1597 uses: malfet/checkout@silent-checkout
1603 quiet-checkout: true
1604 - name: Clean pytorch/builder checkout
1607 git clean -fxd
1608 working-directory: builder
1609 - name: ROCm set GPU_FLAG
1611 …echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add…
1612 - name: Pull Docker image
1613 uses: pytorch/test-infra/.github/actions/pull-docker-image@main
1615 docker-image: pytorch/manylinux-builder:rocm6.1-2.4
1616 - name: Test Pytorch binary
1617 uses: ./pytorch/.github/actions/test-pytorch-binary
1618 - name: Teardown ROCm
1619 uses: ./.github/actions/teardown-rocm
1620 manywheel-py3_10-rocm6_1-upload: # Uploading
1623 id-token: write
1625 needs: manywheel-py3_10-rocm6_1-test
1635 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-2.4
1637 build_name: manywheel-py3_10-rocm6_1
1639 github-token: ${{ secrets.GITHUB_TOKEN }}
1640 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
1641 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
1642 uses: ./.github/workflows/_binary-upload.yml
1644 manywheel-py3_11-cpu-build:
1646 uses: ./.github/workflows/_binary-build-linux.yml
1655 DOCKER_IMAGE: pytorch/manylinux-builder:cpu-2.4
1657 build_name: manywheel-py3_11-cpu
1658 build_environment: linux-binary-manywheel
1660 github-token: ${{ secrets.GITHUB_TOKEN }}
1661 manywheel-py3_11-cpu-test: # Testing
1663 needs: manywheel-py3_11-cpu-build
1664 uses: ./.github/workflows/_binary-test-linux.yml
1673 DOCKER_IMAGE: pytorch/manylinux-builder:cpu-2.4
1675 build_name: manywheel-py3_11-cpu
1676 build_environment: linux-binary-manywheel
1679 github-token: ${{ secrets.GITHUB_TOKEN }}
1680 manywheel-py3_11-cpu-upload: # Uploading
1683 id-token: write
1685 needs: manywheel-py3_11-cpu-test
1694 DOCKER_IMAGE: pytorch/manylinux-builder:cpu-2.4
1696 build_name: manywheel-py3_11-cpu
1698 github-token: ${{ secrets.GITHUB_TOKEN }}
1699 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
1700 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
1701 uses: ./.github/workflows/_binary-upload.yml
1703 manywheel-py3_11-cpu-cxx11-abi-build:
1705 uses: ./.github/workflows/_binary-build-linux.yml
1712 DESIRED_CUDA: cpu-cxx11-abi
1713 GPU_ARCH_TYPE: cpu-cxx11-abi
1714 DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-2.4
1715 DESIRED_DEVTOOLSET: cxx11-abi
1717 build_name: manywheel-py3_11-cpu-cxx11-abi
1718 build_environment: linux-binary-manywheel
1720 github-token: ${{ secrets.GITHUB_TOKEN }}
1721 manywheel-py3_11-cpu-cxx11-abi-test: # Testing
1723 needs: manywheel-py3_11-cpu-cxx11-abi-build
1724 uses: ./.github/workflows/_binary-test-linux.yml
1731 DESIRED_CUDA: cpu-cxx11-abi
1732 GPU_ARCH_TYPE: cpu-cxx11-abi
1733 DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-2.4
1734 DESIRED_DEVTOOLSET: cxx11-abi
1736 build_name: manywheel-py3_11-cpu-cxx11-abi
1737 build_environment: linux-binary-manywheel
1740 github-token: ${{ secrets.GITHUB_TOKEN }}
1741 manywheel-py3_11-cpu-cxx11-abi-upload: # Uploading
1744 id-token: write
1746 needs: manywheel-py3_11-cpu-cxx11-abi-test
1753 DESIRED_CUDA: cpu-cxx11-abi
1754 GPU_ARCH_TYPE: cpu-cxx11-abi
1755 DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-2.4
1756 DESIRED_DEVTOOLSET: cxx11-abi
1758 build_name: manywheel-py3_11-cpu-cxx11-abi
1760 github-token: ${{ secrets.GITHUB_TOKEN }}
1761 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
1762 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
1763 uses: ./.github/workflows/_binary-upload.yml
1765 manywheel-py3_11-cuda11_8-build:
1767 uses: ./.github/workflows/_binary-build-linux.yml
1777 DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4
1779 build_name: manywheel-py3_11-cuda11_8
1780 build_environment: linux-binary-manywheel
1781 …-cuda-nvrtc-cu11==11.8.89; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cu…
1783 github-token: ${{ secrets.GITHUB_TOKEN }}
1784 manywheel-py3_11-cuda11_8-test: # Testing
1786 needs: manywheel-py3_11-cuda11_8-build
1787 uses: ./.github/workflows/_binary-test-linux.yml
1797 DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4
1799 build_name: manywheel-py3_11-cuda11_8
1800 build_environment: linux-binary-manywheel
1803 github-token: ${{ secrets.GITHUB_TOKEN }}
1804 manywheel-py3_11-cuda11_8-upload: # Uploading
1807 id-token: write
1809 needs: manywheel-py3_11-cuda11_8-test
1819 DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4
1821 build_name: manywheel-py3_11-cuda11_8
1823 github-token: ${{ secrets.GITHUB_TOKEN }}
1824 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
1825 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
1826 uses: ./.github/workflows/_binary-upload.yml
1828 manywheel-py3_11-cuda12_1-build:
1830 uses: ./.github/workflows/_binary-build-linux.yml
1840 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
1842 build_name: manywheel-py3_11-cuda12_1
1843 build_environment: linux-binary-manywheel
1844 …-cuda-nvrtc-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-c…
1846 github-token: ${{ secrets.GITHUB_TOKEN }}
1847 manywheel-py3_11-cuda12_1-test: # Testing
1849 needs: manywheel-py3_11-cuda12_1-build
1850 uses: ./.github/workflows/_binary-test-linux.yml
1860 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
1862 build_name: manywheel-py3_11-cuda12_1
1863 build_environment: linux-binary-manywheel
1866 github-token: ${{ secrets.GITHUB_TOKEN }}
1867 manywheel-py3_11-cuda12_1-upload: # Uploading
1870 id-token: write
1872 needs: manywheel-py3_11-cuda12_1-test
1882 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
1884 build_name: manywheel-py3_11-cuda12_1
1886 github-token: ${{ secrets.GITHUB_TOKEN }}
1887 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
1888 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
1889 uses: ./.github/workflows/_binary-upload.yml
1891 manywheel-py3_11-cuda12_4-build:
1893 uses: ./.github/workflows/_binary-build-linux.yml
1903 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4
1905 build_name: manywheel-py3_11-cuda12_4
1906 build_environment: linux-binary-manywheel
1907 …-cuda-nvrtc-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cu…
1909 github-token: ${{ secrets.GITHUB_TOKEN }}
1910 manywheel-py3_11-cuda12_4-test: # Testing
1912 needs: manywheel-py3_11-cuda12_4-build
1913 uses: ./.github/workflows/_binary-test-linux.yml
1923 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4
1925 build_name: manywheel-py3_11-cuda12_4
1926 build_environment: linux-binary-manywheel
1929 github-token: ${{ secrets.GITHUB_TOKEN }}
1930 manywheel-py3_11-cuda12_4-upload: # Uploading
1933 id-token: write
1935 needs: manywheel-py3_11-cuda12_4-test
1945 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4
1947 build_name: manywheel-py3_11-cuda12_4
1949 github-token: ${{ secrets.GITHUB_TOKEN }}
1950 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
1951 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
1952 uses: ./.github/workflows/_binary-upload.yml
1954 manywheel-py3_11-rocm6_0-build:
1956 uses: ./.github/workflows/_binary-build-linux.yml
1966 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-2.4
1968 build_name: manywheel-py3_11-rocm6_0
1969 build_environment: linux-binary-manywheel
1971 github-token: ${{ secrets.GITHUB_TOKEN }}
1972 manywheel-py3_11-rocm6_0-test: # Testing
1974 needs: manywheel-py3_11-rocm6_0-build
1975 runs-on: linux.rocm.gpu
1976 timeout-minutes: 240
1987 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-2.4
1990 - name: Setup ROCm
1991 uses: ./.github/actions/setup-rocm
1992 - uses: actions/download-artifact@v3
1995 name: manywheel-py3_11-rocm6_0
1997 - name: Checkout PyTorch
1998 uses: malfet/checkout@silent-checkout
2002 quiet-checkout: true
2003 - name: Clean PyTorch checkout
2006 git clean -fxd
2007 working-directory: pytorch
2008 - name: Checkout pytorch/builder
2009 uses: malfet/checkout@silent-checkout
2015 quiet-checkout: true
2016 - name: Clean pytorch/builder checkout
2019 git clean -fxd
2020 working-directory: builder
2021 - name: ROCm set GPU_FLAG
2023 …echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add…
2024 - name: Pull Docker image
2025 uses: pytorch/test-infra/.github/actions/pull-docker-image@main
2027 docker-image: pytorch/manylinux-builder:rocm6.0-2.4
2028 - name: Test Pytorch binary
2029 uses: ./pytorch/.github/actions/test-pytorch-binary
2030 - name: Teardown ROCm
2031 uses: ./.github/actions/teardown-rocm
2032 manywheel-py3_11-rocm6_0-upload: # Uploading
2035 id-token: write
2037 needs: manywheel-py3_11-rocm6_0-test
2047 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-2.4
2049 build_name: manywheel-py3_11-rocm6_0
2051 github-token: ${{ secrets.GITHUB_TOKEN }}
2052 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
2053 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
2054 uses: ./.github/workflows/_binary-upload.yml
2056 manywheel-py3_11-rocm6_1-build:
2058 uses: ./.github/workflows/_binary-build-linux.yml
2068 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-2.4
2070 build_name: manywheel-py3_11-rocm6_1
2071 build_environment: linux-binary-manywheel
2073 github-token: ${{ secrets.GITHUB_TOKEN }}
2074 manywheel-py3_11-rocm6_1-test: # Testing
2076 needs: manywheel-py3_11-rocm6_1-build
2077 runs-on: linux.rocm.gpu
2078 timeout-minutes: 240
2089 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-2.4
2092 - name: Setup ROCm
2093 uses: ./.github/actions/setup-rocm
2094 - uses: actions/download-artifact@v3
2097 name: manywheel-py3_11-rocm6_1
2099 - name: Checkout PyTorch
2100 uses: malfet/checkout@silent-checkout
2104 quiet-checkout: true
2105 - name: Clean PyTorch checkout
2108 git clean -fxd
2109 working-directory: pytorch
2110 - name: Checkout pytorch/builder
2111 uses: malfet/checkout@silent-checkout
2117 quiet-checkout: true
2118 - name: Clean pytorch/builder checkout
2121 git clean -fxd
2122 working-directory: builder
2123 - name: ROCm set GPU_FLAG
2125 …echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add…
2126 - name: Pull Docker image
2127 uses: pytorch/test-infra/.github/actions/pull-docker-image@main
2129 docker-image: pytorch/manylinux-builder:rocm6.1-2.4
2130 - name: Test Pytorch binary
2131 uses: ./pytorch/.github/actions/test-pytorch-binary
2132 - name: Teardown ROCm
2133 uses: ./.github/actions/teardown-rocm
2134 manywheel-py3_11-rocm6_1-upload: # Uploading
2137 id-token: write
2139 needs: manywheel-py3_11-rocm6_1-test
2149 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-2.4
2151 build_name: manywheel-py3_11-rocm6_1
2153 github-token: ${{ secrets.GITHUB_TOKEN }}
2154 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
2155 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
2156 uses: ./.github/workflows/_binary-upload.yml
2158 manywheel-py3_12-cpu-build:
2160 uses: ./.github/workflows/_binary-build-linux.yml
2169 DOCKER_IMAGE: pytorch/manylinux-builder:cpu-2.4
2171 build_name: manywheel-py3_12-cpu
2172 build_environment: linux-binary-manywheel
2174 github-token: ${{ secrets.GITHUB_TOKEN }}
2175 manywheel-py3_12-cpu-test: # Testing
2177 needs: manywheel-py3_12-cpu-build
2178 uses: ./.github/workflows/_binary-test-linux.yml
2187 DOCKER_IMAGE: pytorch/manylinux-builder:cpu-2.4
2189 build_name: manywheel-py3_12-cpu
2190 build_environment: linux-binary-manywheel
2193 github-token: ${{ secrets.GITHUB_TOKEN }}
2194 manywheel-py3_12-cpu-upload: # Uploading
2197 id-token: write
2199 needs: manywheel-py3_12-cpu-test
2208 DOCKER_IMAGE: pytorch/manylinux-builder:cpu-2.4
2210 build_name: manywheel-py3_12-cpu
2212 github-token: ${{ secrets.GITHUB_TOKEN }}
2213 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
2214 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
2215 uses: ./.github/workflows/_binary-upload.yml
2217 manywheel-py3_12-cpu-cxx11-abi-build:
2219 uses: ./.github/workflows/_binary-build-linux.yml
2226 DESIRED_CUDA: cpu-cxx11-abi
2227 GPU_ARCH_TYPE: cpu-cxx11-abi
2228 DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-2.4
2229 DESIRED_DEVTOOLSET: cxx11-abi
2231 build_name: manywheel-py3_12-cpu-cxx11-abi
2232 build_environment: linux-binary-manywheel
2234 github-token: ${{ secrets.GITHUB_TOKEN }}
2235 manywheel-py3_12-cpu-cxx11-abi-test: # Testing
2237 needs: manywheel-py3_12-cpu-cxx11-abi-build
2238 uses: ./.github/workflows/_binary-test-linux.yml
2245 DESIRED_CUDA: cpu-cxx11-abi
2246 GPU_ARCH_TYPE: cpu-cxx11-abi
2247 DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-2.4
2248 DESIRED_DEVTOOLSET: cxx11-abi
2250 build_name: manywheel-py3_12-cpu-cxx11-abi
2251 build_environment: linux-binary-manywheel
2254 github-token: ${{ secrets.GITHUB_TOKEN }}
2255 manywheel-py3_12-cpu-cxx11-abi-upload: # Uploading
2258 id-token: write
2260 needs: manywheel-py3_12-cpu-cxx11-abi-test
2267 DESIRED_CUDA: cpu-cxx11-abi
2268 GPU_ARCH_TYPE: cpu-cxx11-abi
2269 DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-2.4
2270 DESIRED_DEVTOOLSET: cxx11-abi
2272 build_name: manywheel-py3_12-cpu-cxx11-abi
2274 github-token: ${{ secrets.GITHUB_TOKEN }}
2275 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
2276 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
2277 uses: ./.github/workflows/_binary-upload.yml
2279 manywheel-py3_12-cuda11_8-build:
2281 uses: ./.github/workflows/_binary-build-linux.yml
2291 DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4
2293 build_name: manywheel-py3_12-cuda11_8
2294 build_environment: linux-binary-manywheel
2295 …-cuda-nvrtc-cu11==11.8.89; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cu…
2297 github-token: ${{ secrets.GITHUB_TOKEN }}
2298 manywheel-py3_12-cuda11_8-test: # Testing
2300 needs: manywheel-py3_12-cuda11_8-build
2301 uses: ./.github/workflows/_binary-test-linux.yml
2311 DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4
2313 build_name: manywheel-py3_12-cuda11_8
2314 build_environment: linux-binary-manywheel
2317 github-token: ${{ secrets.GITHUB_TOKEN }}
2318 manywheel-py3_12-cuda11_8-upload: # Uploading
2321 id-token: write
2323 needs: manywheel-py3_12-cuda11_8-test
2333 DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-2.4
2335 build_name: manywheel-py3_12-cuda11_8
2337 github-token: ${{ secrets.GITHUB_TOKEN }}
2338 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
2339 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
2340 uses: ./.github/workflows/_binary-upload.yml
2342 manywheel-py3_12-cuda12_1-build:
2344 uses: ./.github/workflows/_binary-build-linux.yml
2354 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
2356 build_name: manywheel-py3_12-cuda12_1
2357 build_environment: linux-binary-manywheel
2358 …-cuda-nvrtc-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-c…
2360 github-token: ${{ secrets.GITHUB_TOKEN }}
2361 manywheel-py3_12-cuda12_1-test: # Testing
2363 needs: manywheel-py3_12-cuda12_1-build
2364 uses: ./.github/workflows/_binary-test-linux.yml
2374 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
2376 build_name: manywheel-py3_12-cuda12_1
2377 build_environment: linux-binary-manywheel
2380 github-token: ${{ secrets.GITHUB_TOKEN }}
2381 manywheel-py3_12-cuda12_1-upload: # Uploading
2384 id-token: write
2386 needs: manywheel-py3_12-cuda12_1-test
2396 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-2.4
2398 build_name: manywheel-py3_12-cuda12_1
2400 github-token: ${{ secrets.GITHUB_TOKEN }}
2401 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
2402 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
2403 uses: ./.github/workflows/_binary-upload.yml
2405 manywheel-py3_12-cuda12_4-build:
2407 uses: ./.github/workflows/_binary-build-linux.yml
2417 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4
2419 build_name: manywheel-py3_12-cuda12_4
2420 build_environment: linux-binary-manywheel
2421 …-cuda-nvrtc-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cu…
2423 github-token: ${{ secrets.GITHUB_TOKEN }}
2424 manywheel-py3_12-cuda12_4-test: # Testing
2426 needs: manywheel-py3_12-cuda12_4-build
2427 uses: ./.github/workflows/_binary-test-linux.yml
2437 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4
2439 build_name: manywheel-py3_12-cuda12_4
2440 build_environment: linux-binary-manywheel
2443 github-token: ${{ secrets.GITHUB_TOKEN }}
2444 manywheel-py3_12-cuda12_4-upload: # Uploading
2447 id-token: write
2449 needs: manywheel-py3_12-cuda12_4-test
2459 DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-2.4
2461 build_name: manywheel-py3_12-cuda12_4
2463 github-token: ${{ secrets.GITHUB_TOKEN }}
2464 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
2465 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
2466 uses: ./.github/workflows/_binary-upload.yml
2468 manywheel-py3_12-rocm6_0-build:
2470 uses: ./.github/workflows/_binary-build-linux.yml
2480 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-2.4
2482 build_name: manywheel-py3_12-rocm6_0
2483 build_environment: linux-binary-manywheel
2485 github-token: ${{ secrets.GITHUB_TOKEN }}
2486 manywheel-py3_12-rocm6_0-test: # Testing
2488 needs: manywheel-py3_12-rocm6_0-build
2489 runs-on: linux.rocm.gpu
2490 timeout-minutes: 240
2501 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-2.4
2504 - name: Setup ROCm
2505 uses: ./.github/actions/setup-rocm
2506 - uses: actions/download-artifact@v3
2509 name: manywheel-py3_12-rocm6_0
2511 - name: Checkout PyTorch
2512 uses: malfet/checkout@silent-checkout
2516 quiet-checkout: true
2517 - name: Clean PyTorch checkout
2520 git clean -fxd
2521 working-directory: pytorch
2522 - name: Checkout pytorch/builder
2523 uses: malfet/checkout@silent-checkout
2529 quiet-checkout: true
2530 - name: Clean pytorch/builder checkout
2533 git clean -fxd
2534 working-directory: builder
2535 - name: ROCm set GPU_FLAG
2537 …echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add…
2538 - name: Pull Docker image
2539 uses: pytorch/test-infra/.github/actions/pull-docker-image@main
2541 docker-image: pytorch/manylinux-builder:rocm6.0-2.4
2542 - name: Test Pytorch binary
2543 uses: ./pytorch/.github/actions/test-pytorch-binary
2544 - name: Teardown ROCm
2545 uses: ./.github/actions/teardown-rocm
2546 manywheel-py3_12-rocm6_0-upload: # Uploading
2549 id-token: write
2551 needs: manywheel-py3_12-rocm6_0-test
2561 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-2.4
2563 build_name: manywheel-py3_12-rocm6_0
2565 github-token: ${{ secrets.GITHUB_TOKEN }}
2566 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
2567 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
2568 uses: ./.github/workflows/_binary-upload.yml
2570 manywheel-py3_12-rocm6_1-build:
2572 uses: ./.github/workflows/_binary-build-linux.yml
2582 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-2.4
2584 build_name: manywheel-py3_12-rocm6_1
2585 build_environment: linux-binary-manywheel
2587 github-token: ${{ secrets.GITHUB_TOKEN }}
2588 manywheel-py3_12-rocm6_1-test: # Testing
2590 needs: manywheel-py3_12-rocm6_1-build
2591 runs-on: linux.rocm.gpu
2592 timeout-minutes: 240
2603 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-2.4
2606 - name: Setup ROCm
2607 uses: ./.github/actions/setup-rocm
2608 - uses: actions/download-artifact@v3
2611 name: manywheel-py3_12-rocm6_1
2613 - name: Checkout PyTorch
2614 uses: malfet/checkout@silent-checkout
2618 quiet-checkout: true
2619 - name: Clean PyTorch checkout
2622 git clean -fxd
2623 working-directory: pytorch
2624 - name: Checkout pytorch/builder
2625 uses: malfet/checkout@silent-checkout
2631 quiet-checkout: true
2632 - name: Clean pytorch/builder checkout
2635 git clean -fxd
2636 working-directory: builder
2637 - name: ROCm set GPU_FLAG
2639 …echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add…
2640 - name: Pull Docker image
2641 uses: pytorch/test-infra/.github/actions/pull-docker-image@main
2643 docker-image: pytorch/manylinux-builder:rocm6.1-2.4
2644 - name: Test Pytorch binary
2645 uses: ./pytorch/.github/actions/test-pytorch-binary
2646 - name: Teardown ROCm
2647 uses: ./.github/actions/teardown-rocm
2648 manywheel-py3_12-rocm6_1-upload: # Uploading
2651 id-token: write
2653 needs: manywheel-py3_12-rocm6_1-test
2663 DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-2.4
2665 build_name: manywheel-py3_12-rocm6_1
2667 github-token: ${{ secrets.GITHUB_TOKEN }}
2668 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
2669 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
2670 uses: ./.github/workflows/_binary-upload.yml