linux-binary-manywheel #3676

Workflow file for this run

.github/workflows/generated-linux-binary-manywheel-nightly.yml at 35aa14e

	# @generated DO NOT EDIT MANUALLY

	# Template is at: .github/templates/linux_binary_build_workflow.yml.j2
	# Generation script: .github/scripts/generate_ci_workflows.py
	name: linux-binary-manywheel


	on:
	push:
	# NOTE: Meta Employees can trigger new nightlies using: https://fburl.com/trigger_pytorch_nightly_build
	branches:
	- nightly
	tags:
	# NOTE: Binary build pipelines should only get triggered on release candidate builds
	# Release candidate tags look like: v1.11.0-rc1
	- v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
	- 'ciflow/binaries/*'
	- 'ciflow/binaries_wheel/*'
	workflow_dispatch:

	env:
	# Needed for conda builds
	ALPINE_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/alpine"
	ANACONDA_USER: pytorch
	AWS_DEFAULT_REGION: us-east-1
	BINARY_ENV_FILE: /tmp/env
	BUILD_ENVIRONMENT: linux-binary-manywheel
	BUILDER_ROOT: /builder
	GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
	PR_NUMBER: ${{ github.event.pull_request.number }}
	PYTORCH_FINAL_PACKAGE_DIR: /artifacts
	PYTORCH_ROOT: /pytorch
	SHA1: ${{ github.event.pull_request.head.sha \|\| github.sha }}
	SKIP_ALL_TESTS: 0
	concurrency:
	group: linux-binary-manywheel-${{ github.event.pull_request.number \|\| github.ref_name }}-${{ github.ref_type == 'branch' && github.sha }}-${{ github.event_name == 'workflow_dispatch' }}
	cancel-in-progress: true

	jobs:
	manywheel-py3_8-cpu-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu
	GPU_ARCH_TYPE: cpu
	DOCKER_IMAGE: pytorch/manylinux-builder:cpu-main
	DESIRED_PYTHON: "3.8"
	build_name: manywheel-py3_8-cpu
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_8-cpu-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_8-cpu-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu
	GPU_ARCH_TYPE: cpu
	DOCKER_IMAGE: pytorch/manylinux-builder:cpu-main
	DESIRED_PYTHON: "3.8"
	build_name: manywheel-py3_8-cpu
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_8-cpu-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_8-cpu-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu
	GPU_ARCH_TYPE: cpu
	DOCKER_IMAGE: pytorch/manylinux-builder:cpu-main
	DESIRED_PYTHON: "3.8"
	build_name: manywheel-py3_8-cpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_8-cpu-cxx11-abi-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu-cxx11-abi
	GPU_ARCH_TYPE: cpu-cxx11-abi
	DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-main
	DESIRED_DEVTOOLSET: cxx11-abi
	DESIRED_PYTHON: "3.8"
	build_name: manywheel-py3_8-cpu-cxx11-abi
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_8-cpu-cxx11-abi-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_8-cpu-cxx11-abi-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu-cxx11-abi
	GPU_ARCH_TYPE: cpu-cxx11-abi
	DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-main
	DESIRED_DEVTOOLSET: cxx11-abi
	DESIRED_PYTHON: "3.8"
	build_name: manywheel-py3_8-cpu-cxx11-abi
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_8-cpu-cxx11-abi-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_8-cpu-cxx11-abi-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu-cxx11-abi
	GPU_ARCH_TYPE: cpu-cxx11-abi
	DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-main
	DESIRED_DEVTOOLSET: cxx11-abi
	DESIRED_PYTHON: "3.8"
	build_name: manywheel-py3_8-cpu-cxx11-abi
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_8-cuda11_8-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu118
	GPU_ARCH_VERSION: 11.8
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-main
	DESIRED_PYTHON: "3.8"
	build_name: manywheel-py3_8-cuda11_8
	build_environment: linux-binary-manywheel
	PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu11==11.8.89; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-runtime-cu11==11.8.89; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-cupti-cu11==11.8.87; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cudnn-cu11==8.7.0.84; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cublas-cu11==11.11.3.6; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cufft-cu11==10.9.0.58; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-curand-cu11==10.3.0.86; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusolver-cu11==11.4.1.48; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusparse-cu11==11.7.5.86; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nccl-cu11==2.20.5; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvtx-cu11==11.8.86; platform_system == 'Linux' and platform_machine == 'x86_64'
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_8-cuda11_8-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_8-cuda11_8-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu118
	GPU_ARCH_VERSION: 11.8
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-main
	DESIRED_PYTHON: "3.8"
	build_name: manywheel-py3_8-cuda11_8
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge.nvidia.gpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_8-cuda11_8-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_8-cuda11_8-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu118
	GPU_ARCH_VERSION: 11.8
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-main
	DESIRED_PYTHON: "3.8"
	build_name: manywheel-py3_8-cuda11_8
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_8-cuda12_1-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu121
	GPU_ARCH_VERSION: 12.1
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-main
	DESIRED_PYTHON: "3.8"
	build_name: manywheel-py3_8-cuda12_1
	build_environment: linux-binary-manywheel
	PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-runtime-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-cupti-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cudnn-cu12==8.9.2.26; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cublas-cu12==12.1.3.1; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cufft-cu12==11.0.2.54; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-curand-cu12==10.3.2.106; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusolver-cu12==11.4.5.107; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusparse-cu12==12.1.0.106; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nccl-cu12==2.20.5; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvtx-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64'
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_8-cuda12_1-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_8-cuda12_1-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu121
	GPU_ARCH_VERSION: 12.1
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-main
	DESIRED_PYTHON: "3.8"
	build_name: manywheel-py3_8-cuda12_1
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge.nvidia.gpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_8-cuda12_1-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_8-cuda12_1-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu121
	GPU_ARCH_VERSION: 12.1
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-main
	DESIRED_PYTHON: "3.8"
	build_name: manywheel-py3_8-cuda12_1
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_8-cuda12_4-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu124
	GPU_ARCH_VERSION: 12.4
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-main
	DESIRED_PYTHON: "3.8"
	build_name: manywheel-py3_8-cuda12_4
	build_environment: linux-binary-manywheel
	PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-runtime-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-cupti-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cudnn-cu12==8.9.7.29; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cublas-cu12==12.4.2.65; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cufft-cu12==11.2.0.44; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-curand-cu12==10.3.5.119; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusolver-cu12==11.6.0.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusparse-cu12==12.3.0.142; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nccl-cu12==2.20.5; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvtx-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvjitlink-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64'
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_8-cuda12_4-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_8-cuda12_4-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu124
	GPU_ARCH_VERSION: 12.4
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-main
	DESIRED_PYTHON: "3.8"
	build_name: manywheel-py3_8-cuda12_4
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge.nvidia.gpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_8-cuda12_4-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_8-cuda12_4-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu124
	GPU_ARCH_VERSION: 12.4
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-main
	DESIRED_PYTHON: "3.8"
	build_name: manywheel-py3_8-cuda12_4
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_8-rocm6_0-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.0
	GPU_ARCH_VERSION: 6.0
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-main
	DESIRED_PYTHON: "3.8"
	build_name: manywheel-py3_8-rocm6_0
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_8-rocm6_0-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_8-rocm6_0-build
	runs-on: linux.rocm.gpu
	timeout-minutes: 240
	env:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.0
	GPU_ARCH_VERSION: 6.0
	GPU_ARCH_TYPE: rocm
	SKIP_ALL_TESTS: 1
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-main
	DESIRED_PYTHON: "3.8"
	steps:
	- name: Setup ROCm
	uses: ./.github/actions/setup-rocm
	- uses: actions/download-artifact@v3
	name: Download Build Artifacts
	with:
	name: manywheel-py3_8-rocm6_0
	path: "${{ runner.temp }}/artifacts/"
	- name: Checkout PyTorch
	uses: malfet/checkout@silent-checkout
	with:
	ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha \|\| github.sha }}
	submodules: recursive
	path: pytorch
	quiet-checkout: true
	- name: Clean PyTorch checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: pytorch
	- name: Checkout nWEIdia/builder
	uses: malfet/checkout@silent-checkout
	with:
	ref: cudnn_v9
	submodules: recursive
	repository: nWEIdia/builder
	path: builder
	quiet-checkout: true
	- name: Clean nWEIdia/builder checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: builder
	- name: ROCm set GPU_FLAG
	run: \|
	echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add daemon" >> "${GITHUB_ENV}"
	- name: Pull Docker image
	uses: pytorch/test-infra/.github/actions/pull-docker-image@main
	with:
	docker-image: pytorch/manylinux-builder:rocm6.0-main
	- name: Test Pytorch binary
	uses: ./pytorch/.github/actions/test-pytorch-binary
	- name: Teardown ROCm
	uses: ./.github/actions/teardown-rocm
	manywheel-py3_8-rocm6_0-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_8-rocm6_0-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.0
	GPU_ARCH_VERSION: 6.0
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-main
	DESIRED_PYTHON: "3.8"
	build_name: manywheel-py3_8-rocm6_0
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_8-rocm6_1-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.1
	GPU_ARCH_VERSION: 6.1
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-main
	DESIRED_PYTHON: "3.8"
	build_name: manywheel-py3_8-rocm6_1
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_8-rocm6_1-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_8-rocm6_1-build
	runs-on: linux.rocm.gpu
	timeout-minutes: 240
	env:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.1
	GPU_ARCH_VERSION: 6.1
	GPU_ARCH_TYPE: rocm
	SKIP_ALL_TESTS: 1
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-main
	DESIRED_PYTHON: "3.8"
	steps:
	- name: Setup ROCm
	uses: ./.github/actions/setup-rocm
	- uses: actions/download-artifact@v3
	name: Download Build Artifacts
	with:
	name: manywheel-py3_8-rocm6_1
	path: "${{ runner.temp }}/artifacts/"
	- name: Checkout PyTorch
	uses: malfet/checkout@silent-checkout
	with:
	ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha \|\| github.sha }}
	submodules: recursive
	path: pytorch
	quiet-checkout: true
	- name: Clean PyTorch checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: pytorch
	- name: Checkout nWEIdia/builder
	uses: malfet/checkout@silent-checkout
	with:
	ref: cudnn_v9
	submodules: recursive
	repository: nWEIdia/builder
	path: builder
	quiet-checkout: true
	- name: Clean nWEIdia/builder checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: builder
	- name: ROCm set GPU_FLAG
	run: \|
	echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add daemon" >> "${GITHUB_ENV}"
	- name: Pull Docker image
	uses: pytorch/test-infra/.github/actions/pull-docker-image@main
	with:
	docker-image: pytorch/manylinux-builder:rocm6.1-main
	- name: Test Pytorch binary
	uses: ./pytorch/.github/actions/test-pytorch-binary
	- name: Teardown ROCm
	uses: ./.github/actions/teardown-rocm
	manywheel-py3_8-rocm6_1-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_8-rocm6_1-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.1
	GPU_ARCH_VERSION: 6.1
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-main
	DESIRED_PYTHON: "3.8"
	build_name: manywheel-py3_8-rocm6_1
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_9-cpu-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu
	GPU_ARCH_TYPE: cpu
	DOCKER_IMAGE: pytorch/manylinux-builder:cpu-main
	DESIRED_PYTHON: "3.9"
	build_name: manywheel-py3_9-cpu
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_9-cpu-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_9-cpu-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu
	GPU_ARCH_TYPE: cpu
	DOCKER_IMAGE: pytorch/manylinux-builder:cpu-main
	DESIRED_PYTHON: "3.9"
	build_name: manywheel-py3_9-cpu
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_9-cpu-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_9-cpu-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu
	GPU_ARCH_TYPE: cpu
	DOCKER_IMAGE: pytorch/manylinux-builder:cpu-main
	DESIRED_PYTHON: "3.9"
	build_name: manywheel-py3_9-cpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_9-cpu-cxx11-abi-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu-cxx11-abi
	GPU_ARCH_TYPE: cpu-cxx11-abi
	DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-main
	DESIRED_DEVTOOLSET: cxx11-abi
	DESIRED_PYTHON: "3.9"
	build_name: manywheel-py3_9-cpu-cxx11-abi
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_9-cpu-cxx11-abi-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_9-cpu-cxx11-abi-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu-cxx11-abi
	GPU_ARCH_TYPE: cpu-cxx11-abi
	DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-main
	DESIRED_DEVTOOLSET: cxx11-abi
	DESIRED_PYTHON: "3.9"
	build_name: manywheel-py3_9-cpu-cxx11-abi
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_9-cpu-cxx11-abi-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_9-cpu-cxx11-abi-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu-cxx11-abi
	GPU_ARCH_TYPE: cpu-cxx11-abi
	DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-main
	DESIRED_DEVTOOLSET: cxx11-abi
	DESIRED_PYTHON: "3.9"
	build_name: manywheel-py3_9-cpu-cxx11-abi
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_9-cuda11_8-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu118
	GPU_ARCH_VERSION: 11.8
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-main
	DESIRED_PYTHON: "3.9"
	build_name: manywheel-py3_9-cuda11_8
	build_environment: linux-binary-manywheel
	PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu11==11.8.89; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-runtime-cu11==11.8.89; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-cupti-cu11==11.8.87; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cudnn-cu11==8.7.0.84; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cublas-cu11==11.11.3.6; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cufft-cu11==10.9.0.58; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-curand-cu11==10.3.0.86; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusolver-cu11==11.4.1.48; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusparse-cu11==11.7.5.86; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nccl-cu11==2.20.5; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvtx-cu11==11.8.86; platform_system == 'Linux' and platform_machine == 'x86_64'
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_9-cuda11_8-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_9-cuda11_8-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu118
	GPU_ARCH_VERSION: 11.8
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-main
	DESIRED_PYTHON: "3.9"
	build_name: manywheel-py3_9-cuda11_8
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge.nvidia.gpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_9-cuda11_8-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_9-cuda11_8-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu118
	GPU_ARCH_VERSION: 11.8
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-main
	DESIRED_PYTHON: "3.9"
	build_name: manywheel-py3_9-cuda11_8
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_9-cuda12_1-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu121
	GPU_ARCH_VERSION: 12.1
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-main
	DESIRED_PYTHON: "3.9"
	build_name: manywheel-py3_9-cuda12_1
	build_environment: linux-binary-manywheel
	PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-runtime-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-cupti-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cudnn-cu12==8.9.2.26; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cublas-cu12==12.1.3.1; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cufft-cu12==11.0.2.54; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-curand-cu12==10.3.2.106; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusolver-cu12==11.4.5.107; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusparse-cu12==12.1.0.106; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nccl-cu12==2.20.5; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvtx-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64'
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_9-cuda12_1-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_9-cuda12_1-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu121
	GPU_ARCH_VERSION: 12.1
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-main
	DESIRED_PYTHON: "3.9"
	build_name: manywheel-py3_9-cuda12_1
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge.nvidia.gpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_9-cuda12_1-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_9-cuda12_1-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu121
	GPU_ARCH_VERSION: 12.1
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-main
	DESIRED_PYTHON: "3.9"
	build_name: manywheel-py3_9-cuda12_1
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_9-cuda12_4-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu124
	GPU_ARCH_VERSION: 12.4
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-main
	DESIRED_PYTHON: "3.9"
	build_name: manywheel-py3_9-cuda12_4
	build_environment: linux-binary-manywheel
	PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-runtime-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-cupti-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cudnn-cu12==8.9.7.29; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cublas-cu12==12.4.2.65; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cufft-cu12==11.2.0.44; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-curand-cu12==10.3.5.119; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusolver-cu12==11.6.0.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusparse-cu12==12.3.0.142; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nccl-cu12==2.20.5; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvtx-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvjitlink-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64'
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_9-cuda12_4-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_9-cuda12_4-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu124
	GPU_ARCH_VERSION: 12.4
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-main
	DESIRED_PYTHON: "3.9"
	build_name: manywheel-py3_9-cuda12_4
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge.nvidia.gpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_9-cuda12_4-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_9-cuda12_4-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu124
	GPU_ARCH_VERSION: 12.4
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-main
	DESIRED_PYTHON: "3.9"
	build_name: manywheel-py3_9-cuda12_4
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_9-rocm6_0-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.0
	GPU_ARCH_VERSION: 6.0
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-main
	DESIRED_PYTHON: "3.9"
	build_name: manywheel-py3_9-rocm6_0
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_9-rocm6_0-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_9-rocm6_0-build
	runs-on: linux.rocm.gpu
	timeout-minutes: 240
	env:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.0
	GPU_ARCH_VERSION: 6.0
	GPU_ARCH_TYPE: rocm
	SKIP_ALL_TESTS: 1
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-main
	DESIRED_PYTHON: "3.9"
	steps:
	- name: Setup ROCm
	uses: ./.github/actions/setup-rocm
	- uses: actions/download-artifact@v3
	name: Download Build Artifacts
	with:
	name: manywheel-py3_9-rocm6_0
	path: "${{ runner.temp }}/artifacts/"
	- name: Checkout PyTorch
	uses: malfet/checkout@silent-checkout
	with:
	ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha \|\| github.sha }}
	submodules: recursive
	path: pytorch
	quiet-checkout: true
	- name: Clean PyTorch checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: pytorch
	- name: Checkout nWEIdia/builder
	uses: malfet/checkout@silent-checkout
	with:
	ref: cudnn_v9
	submodules: recursive
	repository: nWEIdia/builder
	path: builder
	quiet-checkout: true
	- name: Clean nWEIdia/builder checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: builder
	- name: ROCm set GPU_FLAG
	run: \|
	echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add daemon" >> "${GITHUB_ENV}"
	- name: Pull Docker image
	uses: pytorch/test-infra/.github/actions/pull-docker-image@main
	with:
	docker-image: pytorch/manylinux-builder:rocm6.0-main
	- name: Test Pytorch binary
	uses: ./pytorch/.github/actions/test-pytorch-binary
	- name: Teardown ROCm
	uses: ./.github/actions/teardown-rocm
	manywheel-py3_9-rocm6_0-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_9-rocm6_0-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.0
	GPU_ARCH_VERSION: 6.0
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-main
	DESIRED_PYTHON: "3.9"
	build_name: manywheel-py3_9-rocm6_0
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_9-rocm6_1-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.1
	GPU_ARCH_VERSION: 6.1
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-main
	DESIRED_PYTHON: "3.9"
	build_name: manywheel-py3_9-rocm6_1
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_9-rocm6_1-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_9-rocm6_1-build
	runs-on: linux.rocm.gpu
	timeout-minutes: 240
	env:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.1
	GPU_ARCH_VERSION: 6.1
	GPU_ARCH_TYPE: rocm
	SKIP_ALL_TESTS: 1
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-main
	DESIRED_PYTHON: "3.9"
	steps:
	- name: Setup ROCm
	uses: ./.github/actions/setup-rocm
	- uses: actions/download-artifact@v3
	name: Download Build Artifacts
	with:
	name: manywheel-py3_9-rocm6_1
	path: "${{ runner.temp }}/artifacts/"
	- name: Checkout PyTorch
	uses: malfet/checkout@silent-checkout
	with:
	ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha \|\| github.sha }}
	submodules: recursive
	path: pytorch
	quiet-checkout: true
	- name: Clean PyTorch checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: pytorch
	- name: Checkout nWEIdia/builder
	uses: malfet/checkout@silent-checkout
	with:
	ref: cudnn_v9
	submodules: recursive
	repository: nWEIdia/builder
	path: builder
	quiet-checkout: true
	- name: Clean nWEIdia/builder checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: builder
	- name: ROCm set GPU_FLAG
	run: \|
	echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add daemon" >> "${GITHUB_ENV}"
	- name: Pull Docker image
	uses: pytorch/test-infra/.github/actions/pull-docker-image@main
	with:
	docker-image: pytorch/manylinux-builder:rocm6.1-main
	- name: Test Pytorch binary
	uses: ./pytorch/.github/actions/test-pytorch-binary
	- name: Teardown ROCm
	uses: ./.github/actions/teardown-rocm
	manywheel-py3_9-rocm6_1-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_9-rocm6_1-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.1
	GPU_ARCH_VERSION: 6.1
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-main
	DESIRED_PYTHON: "3.9"
	build_name: manywheel-py3_9-rocm6_1
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_10-cpu-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu
	GPU_ARCH_TYPE: cpu
	DOCKER_IMAGE: pytorch/manylinux-builder:cpu-main
	DESIRED_PYTHON: "3.10"
	build_name: manywheel-py3_10-cpu
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_10-cpu-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_10-cpu-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu
	GPU_ARCH_TYPE: cpu
	DOCKER_IMAGE: pytorch/manylinux-builder:cpu-main
	DESIRED_PYTHON: "3.10"
	build_name: manywheel-py3_10-cpu
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_10-cpu-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_10-cpu-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu
	GPU_ARCH_TYPE: cpu
	DOCKER_IMAGE: pytorch/manylinux-builder:cpu-main
	DESIRED_PYTHON: "3.10"
	build_name: manywheel-py3_10-cpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_10-cpu-cxx11-abi-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu-cxx11-abi
	GPU_ARCH_TYPE: cpu-cxx11-abi
	DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-main
	DESIRED_DEVTOOLSET: cxx11-abi
	DESIRED_PYTHON: "3.10"
	build_name: manywheel-py3_10-cpu-cxx11-abi
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_10-cpu-cxx11-abi-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_10-cpu-cxx11-abi-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu-cxx11-abi
	GPU_ARCH_TYPE: cpu-cxx11-abi
	DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-main
	DESIRED_DEVTOOLSET: cxx11-abi
	DESIRED_PYTHON: "3.10"
	build_name: manywheel-py3_10-cpu-cxx11-abi
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_10-cpu-cxx11-abi-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_10-cpu-cxx11-abi-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu-cxx11-abi
	GPU_ARCH_TYPE: cpu-cxx11-abi
	DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-main
	DESIRED_DEVTOOLSET: cxx11-abi
	DESIRED_PYTHON: "3.10"
	build_name: manywheel-py3_10-cpu-cxx11-abi
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_10-cuda11_8-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu118
	GPU_ARCH_VERSION: 11.8
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-main
	DESIRED_PYTHON: "3.10"
	build_name: manywheel-py3_10-cuda11_8
	build_environment: linux-binary-manywheel
	PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu11==11.8.89; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-runtime-cu11==11.8.89; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-cupti-cu11==11.8.87; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cudnn-cu11==8.7.0.84; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cublas-cu11==11.11.3.6; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cufft-cu11==10.9.0.58; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-curand-cu11==10.3.0.86; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusolver-cu11==11.4.1.48; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusparse-cu11==11.7.5.86; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nccl-cu11==2.20.5; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvtx-cu11==11.8.86; platform_system == 'Linux' and platform_machine == 'x86_64'
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_10-cuda11_8-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_10-cuda11_8-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu118
	GPU_ARCH_VERSION: 11.8
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-main
	DESIRED_PYTHON: "3.10"
	build_name: manywheel-py3_10-cuda11_8
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge.nvidia.gpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_10-cuda11_8-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_10-cuda11_8-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu118
	GPU_ARCH_VERSION: 11.8
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-main
	DESIRED_PYTHON: "3.10"
	build_name: manywheel-py3_10-cuda11_8
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_10-cuda12_1-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu121
	GPU_ARCH_VERSION: 12.1
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-main
	DESIRED_PYTHON: "3.10"
	build_name: manywheel-py3_10-cuda12_1
	build_environment: linux-binary-manywheel
	PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-runtime-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-cupti-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cudnn-cu12==8.9.2.26; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cublas-cu12==12.1.3.1; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cufft-cu12==11.0.2.54; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-curand-cu12==10.3.2.106; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusolver-cu12==11.4.5.107; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusparse-cu12==12.1.0.106; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nccl-cu12==2.20.5; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvtx-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64'
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_10-cuda12_1-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_10-cuda12_1-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu121
	GPU_ARCH_VERSION: 12.1
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-main
	DESIRED_PYTHON: "3.10"
	build_name: manywheel-py3_10-cuda12_1
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge.nvidia.gpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_10-cuda12_1-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_10-cuda12_1-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu121
	GPU_ARCH_VERSION: 12.1
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-main
	DESIRED_PYTHON: "3.10"
	build_name: manywheel-py3_10-cuda12_1
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_10-cuda12_4-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu124
	GPU_ARCH_VERSION: 12.4
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-main
	DESIRED_PYTHON: "3.10"
	build_name: manywheel-py3_10-cuda12_4
	build_environment: linux-binary-manywheel
	PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-runtime-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-cupti-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cudnn-cu12==8.9.7.29; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cublas-cu12==12.4.2.65; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cufft-cu12==11.2.0.44; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-curand-cu12==10.3.5.119; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusolver-cu12==11.6.0.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusparse-cu12==12.3.0.142; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nccl-cu12==2.20.5; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvtx-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvjitlink-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64'
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_10-cuda12_4-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_10-cuda12_4-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu124
	GPU_ARCH_VERSION: 12.4
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-main
	DESIRED_PYTHON: "3.10"
	build_name: manywheel-py3_10-cuda12_4
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge.nvidia.gpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_10-cuda12_4-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_10-cuda12_4-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu124
	GPU_ARCH_VERSION: 12.4
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-main
	DESIRED_PYTHON: "3.10"
	build_name: manywheel-py3_10-cuda12_4
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_10-rocm6_0-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.0
	GPU_ARCH_VERSION: 6.0
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-main
	DESIRED_PYTHON: "3.10"
	build_name: manywheel-py3_10-rocm6_0
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_10-rocm6_0-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_10-rocm6_0-build
	runs-on: linux.rocm.gpu
	timeout-minutes: 240
	env:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.0
	GPU_ARCH_VERSION: 6.0
	GPU_ARCH_TYPE: rocm
	SKIP_ALL_TESTS: 1
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-main
	DESIRED_PYTHON: "3.10"
	steps:
	- name: Setup ROCm
	uses: ./.github/actions/setup-rocm
	- uses: actions/download-artifact@v3
	name: Download Build Artifacts
	with:
	name: manywheel-py3_10-rocm6_0
	path: "${{ runner.temp }}/artifacts/"
	- name: Checkout PyTorch
	uses: malfet/checkout@silent-checkout
	with:
	ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha \|\| github.sha }}
	submodules: recursive
	path: pytorch
	quiet-checkout: true
	- name: Clean PyTorch checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: pytorch
	- name: Checkout nWEIdia/builder
	uses: malfet/checkout@silent-checkout
	with:
	ref: cudnn_v9
	submodules: recursive
	repository: nWEIdia/builder
	path: builder
	quiet-checkout: true
	- name: Clean nWEIdia/builder checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: builder
	- name: ROCm set GPU_FLAG
	run: \|
	echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add daemon" >> "${GITHUB_ENV}"
	- name: Pull Docker image
	uses: pytorch/test-infra/.github/actions/pull-docker-image@main
	with:
	docker-image: pytorch/manylinux-builder:rocm6.0-main
	- name: Test Pytorch binary
	uses: ./pytorch/.github/actions/test-pytorch-binary
	- name: Teardown ROCm
	uses: ./.github/actions/teardown-rocm
	manywheel-py3_10-rocm6_0-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_10-rocm6_0-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.0
	GPU_ARCH_VERSION: 6.0
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-main
	DESIRED_PYTHON: "3.10"
	build_name: manywheel-py3_10-rocm6_0
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_10-rocm6_1-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.1
	GPU_ARCH_VERSION: 6.1
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-main
	DESIRED_PYTHON: "3.10"
	build_name: manywheel-py3_10-rocm6_1
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_10-rocm6_1-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_10-rocm6_1-build
	runs-on: linux.rocm.gpu
	timeout-minutes: 240
	env:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.1
	GPU_ARCH_VERSION: 6.1
	GPU_ARCH_TYPE: rocm
	SKIP_ALL_TESTS: 1
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-main
	DESIRED_PYTHON: "3.10"
	steps:
	- name: Setup ROCm
	uses: ./.github/actions/setup-rocm
	- uses: actions/download-artifact@v3
	name: Download Build Artifacts
	with:
	name: manywheel-py3_10-rocm6_1
	path: "${{ runner.temp }}/artifacts/"
	- name: Checkout PyTorch
	uses: malfet/checkout@silent-checkout
	with:
	ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha \|\| github.sha }}
	submodules: recursive
	path: pytorch
	quiet-checkout: true
	- name: Clean PyTorch checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: pytorch
	- name: Checkout nWEIdia/builder
	uses: malfet/checkout@silent-checkout
	with:
	ref: cudnn_v9
	submodules: recursive
	repository: nWEIdia/builder
	path: builder
	quiet-checkout: true
	- name: Clean nWEIdia/builder checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: builder
	- name: ROCm set GPU_FLAG
	run: \|
	echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add daemon" >> "${GITHUB_ENV}"
	- name: Pull Docker image
	uses: pytorch/test-infra/.github/actions/pull-docker-image@main
	with:
	docker-image: pytorch/manylinux-builder:rocm6.1-main
	- name: Test Pytorch binary
	uses: ./pytorch/.github/actions/test-pytorch-binary
	- name: Teardown ROCm
	uses: ./.github/actions/teardown-rocm
	manywheel-py3_10-rocm6_1-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_10-rocm6_1-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.1
	GPU_ARCH_VERSION: 6.1
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-main
	DESIRED_PYTHON: "3.10"
	build_name: manywheel-py3_10-rocm6_1
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_11-cpu-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu
	GPU_ARCH_TYPE: cpu
	DOCKER_IMAGE: pytorch/manylinux-builder:cpu-main
	DESIRED_PYTHON: "3.11"
	build_name: manywheel-py3_11-cpu
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_11-cpu-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_11-cpu-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu
	GPU_ARCH_TYPE: cpu
	DOCKER_IMAGE: pytorch/manylinux-builder:cpu-main
	DESIRED_PYTHON: "3.11"
	build_name: manywheel-py3_11-cpu
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_11-cpu-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_11-cpu-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu
	GPU_ARCH_TYPE: cpu
	DOCKER_IMAGE: pytorch/manylinux-builder:cpu-main
	DESIRED_PYTHON: "3.11"
	build_name: manywheel-py3_11-cpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_11-cpu-cxx11-abi-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu-cxx11-abi
	GPU_ARCH_TYPE: cpu-cxx11-abi
	DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-main
	DESIRED_DEVTOOLSET: cxx11-abi
	DESIRED_PYTHON: "3.11"
	build_name: manywheel-py3_11-cpu-cxx11-abi
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_11-cpu-cxx11-abi-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_11-cpu-cxx11-abi-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu-cxx11-abi
	GPU_ARCH_TYPE: cpu-cxx11-abi
	DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-main
	DESIRED_DEVTOOLSET: cxx11-abi
	DESIRED_PYTHON: "3.11"
	build_name: manywheel-py3_11-cpu-cxx11-abi
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_11-cpu-cxx11-abi-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_11-cpu-cxx11-abi-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu-cxx11-abi
	GPU_ARCH_TYPE: cpu-cxx11-abi
	DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-main
	DESIRED_DEVTOOLSET: cxx11-abi
	DESIRED_PYTHON: "3.11"
	build_name: manywheel-py3_11-cpu-cxx11-abi
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_11-cuda11_8-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu118
	GPU_ARCH_VERSION: 11.8
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-main
	DESIRED_PYTHON: "3.11"
	build_name: manywheel-py3_11-cuda11_8
	build_environment: linux-binary-manywheel
	PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu11==11.8.89; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-runtime-cu11==11.8.89; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-cupti-cu11==11.8.87; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cudnn-cu11==8.7.0.84; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cublas-cu11==11.11.3.6; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cufft-cu11==10.9.0.58; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-curand-cu11==10.3.0.86; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusolver-cu11==11.4.1.48; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusparse-cu11==11.7.5.86; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nccl-cu11==2.20.5; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvtx-cu11==11.8.86; platform_system == 'Linux' and platform_machine == 'x86_64'
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_11-cuda11_8-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_11-cuda11_8-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu118
	GPU_ARCH_VERSION: 11.8
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-main
	DESIRED_PYTHON: "3.11"
	build_name: manywheel-py3_11-cuda11_8
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge.nvidia.gpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_11-cuda11_8-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_11-cuda11_8-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu118
	GPU_ARCH_VERSION: 11.8
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-main
	DESIRED_PYTHON: "3.11"
	build_name: manywheel-py3_11-cuda11_8
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_11-cuda12_1-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu121
	GPU_ARCH_VERSION: 12.1
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-main
	DESIRED_PYTHON: "3.11"
	build_name: manywheel-py3_11-cuda12_1
	build_environment: linux-binary-manywheel
	PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-runtime-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-cupti-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cudnn-cu12==8.9.2.26; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cublas-cu12==12.1.3.1; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cufft-cu12==11.0.2.54; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-curand-cu12==10.3.2.106; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusolver-cu12==11.4.5.107; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusparse-cu12==12.1.0.106; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nccl-cu12==2.20.5; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvtx-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64'
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_11-cuda12_1-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_11-cuda12_1-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu121
	GPU_ARCH_VERSION: 12.1
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-main
	DESIRED_PYTHON: "3.11"
	build_name: manywheel-py3_11-cuda12_1
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge.nvidia.gpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_11-cuda12_1-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_11-cuda12_1-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu121
	GPU_ARCH_VERSION: 12.1
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-main
	DESIRED_PYTHON: "3.11"
	build_name: manywheel-py3_11-cuda12_1
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_11-cuda12_4-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu124
	GPU_ARCH_VERSION: 12.4
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-main
	DESIRED_PYTHON: "3.11"
	build_name: manywheel-py3_11-cuda12_4
	build_environment: linux-binary-manywheel
	PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-runtime-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-cupti-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cudnn-cu12==8.9.7.29; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cublas-cu12==12.4.2.65; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cufft-cu12==11.2.0.44; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-curand-cu12==10.3.5.119; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusolver-cu12==11.6.0.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusparse-cu12==12.3.0.142; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nccl-cu12==2.20.5; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvtx-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvjitlink-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64'
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_11-cuda12_4-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_11-cuda12_4-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu124
	GPU_ARCH_VERSION: 12.4
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-main
	DESIRED_PYTHON: "3.11"
	build_name: manywheel-py3_11-cuda12_4
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge.nvidia.gpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_11-cuda12_4-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_11-cuda12_4-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu124
	GPU_ARCH_VERSION: 12.4
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-main
	DESIRED_PYTHON: "3.11"
	build_name: manywheel-py3_11-cuda12_4
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_11-rocm6_0-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.0
	GPU_ARCH_VERSION: 6.0
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-main
	DESIRED_PYTHON: "3.11"
	build_name: manywheel-py3_11-rocm6_0
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_11-rocm6_0-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_11-rocm6_0-build
	runs-on: linux.rocm.gpu
	timeout-minutes: 240
	env:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.0
	GPU_ARCH_VERSION: 6.0
	GPU_ARCH_TYPE: rocm
	SKIP_ALL_TESTS: 1
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-main
	DESIRED_PYTHON: "3.11"
	steps:
	- name: Setup ROCm
	uses: ./.github/actions/setup-rocm
	- uses: actions/download-artifact@v3
	name: Download Build Artifacts
	with:
	name: manywheel-py3_11-rocm6_0
	path: "${{ runner.temp }}/artifacts/"
	- name: Checkout PyTorch
	uses: malfet/checkout@silent-checkout
	with:
	ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha \|\| github.sha }}
	submodules: recursive
	path: pytorch
	quiet-checkout: true
	- name: Clean PyTorch checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: pytorch
	- name: Checkout nWEIdia/builder
	uses: malfet/checkout@silent-checkout
	with:
	ref: cudnn_v9
	submodules: recursive
	repository: nWEIdia/builder
	path: builder
	quiet-checkout: true
	- name: Clean nWEIdia/builder checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: builder
	- name: ROCm set GPU_FLAG
	run: \|
	echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add daemon" >> "${GITHUB_ENV}"
	- name: Pull Docker image
	uses: pytorch/test-infra/.github/actions/pull-docker-image@main
	with:
	docker-image: pytorch/manylinux-builder:rocm6.0-main
	- name: Test Pytorch binary
	uses: ./pytorch/.github/actions/test-pytorch-binary
	- name: Teardown ROCm
	uses: ./.github/actions/teardown-rocm
	manywheel-py3_11-rocm6_0-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_11-rocm6_0-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.0
	GPU_ARCH_VERSION: 6.0
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-main
	DESIRED_PYTHON: "3.11"
	build_name: manywheel-py3_11-rocm6_0
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_11-rocm6_1-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.1
	GPU_ARCH_VERSION: 6.1
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-main
	DESIRED_PYTHON: "3.11"
	build_name: manywheel-py3_11-rocm6_1
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_11-rocm6_1-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_11-rocm6_1-build
	runs-on: linux.rocm.gpu
	timeout-minutes: 240
	env:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.1
	GPU_ARCH_VERSION: 6.1
	GPU_ARCH_TYPE: rocm
	SKIP_ALL_TESTS: 1
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-main
	DESIRED_PYTHON: "3.11"
	steps:
	- name: Setup ROCm
	uses: ./.github/actions/setup-rocm
	- uses: actions/download-artifact@v3
	name: Download Build Artifacts
	with:
	name: manywheel-py3_11-rocm6_1
	path: "${{ runner.temp }}/artifacts/"
	- name: Checkout PyTorch
	uses: malfet/checkout@silent-checkout
	with:
	ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha \|\| github.sha }}
	submodules: recursive
	path: pytorch
	quiet-checkout: true
	- name: Clean PyTorch checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: pytorch
	- name: Checkout nWEIdia/builder
	uses: malfet/checkout@silent-checkout
	with:
	ref: cudnn_v9
	submodules: recursive
	repository: nWEIdia/builder
	path: builder
	quiet-checkout: true
	- name: Clean nWEIdia/builder checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: builder
	- name: ROCm set GPU_FLAG
	run: \|
	echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add daemon" >> "${GITHUB_ENV}"
	- name: Pull Docker image
	uses: pytorch/test-infra/.github/actions/pull-docker-image@main
	with:
	docker-image: pytorch/manylinux-builder:rocm6.1-main
	- name: Test Pytorch binary
	uses: ./pytorch/.github/actions/test-pytorch-binary
	- name: Teardown ROCm
	uses: ./.github/actions/teardown-rocm
	manywheel-py3_11-rocm6_1-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_11-rocm6_1-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.1
	GPU_ARCH_VERSION: 6.1
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-main
	DESIRED_PYTHON: "3.11"
	build_name: manywheel-py3_11-rocm6_1
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_12-cpu-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu
	GPU_ARCH_TYPE: cpu
	DOCKER_IMAGE: pytorch/manylinux-builder:cpu-main
	DESIRED_PYTHON: "3.12"
	build_name: manywheel-py3_12-cpu
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_12-cpu-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_12-cpu-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu
	GPU_ARCH_TYPE: cpu
	DOCKER_IMAGE: pytorch/manylinux-builder:cpu-main
	DESIRED_PYTHON: "3.12"
	build_name: manywheel-py3_12-cpu
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_12-cpu-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_12-cpu-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu
	GPU_ARCH_TYPE: cpu
	DOCKER_IMAGE: pytorch/manylinux-builder:cpu-main
	DESIRED_PYTHON: "3.12"
	build_name: manywheel-py3_12-cpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_12-cpu-cxx11-abi-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu-cxx11-abi
	GPU_ARCH_TYPE: cpu-cxx11-abi
	DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-main
	DESIRED_DEVTOOLSET: cxx11-abi
	DESIRED_PYTHON: "3.12"
	build_name: manywheel-py3_12-cpu-cxx11-abi
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_12-cpu-cxx11-abi-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_12-cpu-cxx11-abi-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu-cxx11-abi
	GPU_ARCH_TYPE: cpu-cxx11-abi
	DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-main
	DESIRED_DEVTOOLSET: cxx11-abi
	DESIRED_PYTHON: "3.12"
	build_name: manywheel-py3_12-cpu-cxx11-abi
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_12-cpu-cxx11-abi-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_12-cpu-cxx11-abi-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cpu-cxx11-abi
	GPU_ARCH_TYPE: cpu-cxx11-abi
	DOCKER_IMAGE: pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-main
	DESIRED_DEVTOOLSET: cxx11-abi
	DESIRED_PYTHON: "3.12"
	build_name: manywheel-py3_12-cpu-cxx11-abi
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_12-cuda11_8-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu118
	GPU_ARCH_VERSION: 11.8
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-main
	DESIRED_PYTHON: "3.12"
	build_name: manywheel-py3_12-cuda11_8
	build_environment: linux-binary-manywheel
	PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu11==11.8.89; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-runtime-cu11==11.8.89; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-cupti-cu11==11.8.87; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cudnn-cu11==8.7.0.84; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cublas-cu11==11.11.3.6; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cufft-cu11==10.9.0.58; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-curand-cu11==10.3.0.86; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusolver-cu11==11.4.1.48; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusparse-cu11==11.7.5.86; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nccl-cu11==2.20.5; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvtx-cu11==11.8.86; platform_system == 'Linux' and platform_machine == 'x86_64'
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_12-cuda11_8-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_12-cuda11_8-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu118
	GPU_ARCH_VERSION: 11.8
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-main
	DESIRED_PYTHON: "3.12"
	build_name: manywheel-py3_12-cuda11_8
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge.nvidia.gpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_12-cuda11_8-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_12-cuda11_8-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu118
	GPU_ARCH_VERSION: 11.8
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda11.8-main
	DESIRED_PYTHON: "3.12"
	build_name: manywheel-py3_12-cuda11_8
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_12-cuda12_1-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu121
	GPU_ARCH_VERSION: 12.1
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-main
	DESIRED_PYTHON: "3.12"
	build_name: manywheel-py3_12-cuda12_1
	build_environment: linux-binary-manywheel
	PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-runtime-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-cupti-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cudnn-cu12==8.9.2.26; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cublas-cu12==12.1.3.1; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cufft-cu12==11.0.2.54; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-curand-cu12==10.3.2.106; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusolver-cu12==11.4.5.107; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusparse-cu12==12.1.0.106; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nccl-cu12==2.20.5; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvtx-cu12==12.1.105; platform_system == 'Linux' and platform_machine == 'x86_64'
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_12-cuda12_1-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_12-cuda12_1-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu121
	GPU_ARCH_VERSION: 12.1
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-main
	DESIRED_PYTHON: "3.12"
	build_name: manywheel-py3_12-cuda12_1
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge.nvidia.gpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_12-cuda12_1-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_12-cuda12_1-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu121
	GPU_ARCH_VERSION: 12.1
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.1-main
	DESIRED_PYTHON: "3.12"
	build_name: manywheel-py3_12-cuda12_1
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_12-cuda12_4-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu124
	GPU_ARCH_VERSION: 12.4
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-main
	DESIRED_PYTHON: "3.12"
	build_name: manywheel-py3_12-cuda12_4
	build_environment: linux-binary-manywheel
	PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-runtime-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cuda-cupti-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cudnn-cu12==8.9.7.29; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cublas-cu12==12.4.2.65; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cufft-cu12==11.2.0.44; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-curand-cu12==10.3.5.119; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusolver-cu12==11.6.0.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-cusparse-cu12==12.3.0.142; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nccl-cu12==2.20.5; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvtx-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64' \| nvidia-nvjitlink-cu12==12.4.99; platform_system == 'Linux' and platform_machine == 'x86_64'
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_12-cuda12_4-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_12-cuda12_4-build
	uses: ./.github/workflows/_binary-test-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu124
	GPU_ARCH_VERSION: 12.4
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-main
	DESIRED_PYTHON: "3.12"
	build_name: manywheel-py3_12-cuda12_4
	build_environment: linux-binary-manywheel
	runs_on: linux.4xlarge.nvidia.gpu
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_12-cuda12_4-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_12-cuda12_4-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: cu124
	GPU_ARCH_VERSION: 12.4
	GPU_ARCH_TYPE: cuda
	DOCKER_IMAGE: pytorch/manylinux-builder:cuda12.4-main
	DESIRED_PYTHON: "3.12"
	build_name: manywheel-py3_12-cuda12_4
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_12-rocm6_0-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.0
	GPU_ARCH_VERSION: 6.0
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-main
	DESIRED_PYTHON: "3.12"
	build_name: manywheel-py3_12-rocm6_0
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_12-rocm6_0-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_12-rocm6_0-build
	runs-on: linux.rocm.gpu
	timeout-minutes: 240
	env:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.0
	GPU_ARCH_VERSION: 6.0
	GPU_ARCH_TYPE: rocm
	SKIP_ALL_TESTS: 1
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-main
	DESIRED_PYTHON: "3.12"
	steps:
	- name: Setup ROCm
	uses: ./.github/actions/setup-rocm
	- uses: actions/download-artifact@v3
	name: Download Build Artifacts
	with:
	name: manywheel-py3_12-rocm6_0
	path: "${{ runner.temp }}/artifacts/"
	- name: Checkout PyTorch
	uses: malfet/checkout@silent-checkout
	with:
	ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha \|\| github.sha }}
	submodules: recursive
	path: pytorch
	quiet-checkout: true
	- name: Clean PyTorch checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: pytorch
	- name: Checkout nWEIdia/builder
	uses: malfet/checkout@silent-checkout
	with:
	ref: cudnn_v9
	submodules: recursive
	repository: nWEIdia/builder
	path: builder
	quiet-checkout: true
	- name: Clean nWEIdia/builder checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: builder
	- name: ROCm set GPU_FLAG
	run: \|
	echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add daemon" >> "${GITHUB_ENV}"
	- name: Pull Docker image
	uses: pytorch/test-infra/.github/actions/pull-docker-image@main
	with:
	docker-image: pytorch/manylinux-builder:rocm6.0-main
	- name: Test Pytorch binary
	uses: ./pytorch/.github/actions/test-pytorch-binary
	- name: Teardown ROCm
	uses: ./.github/actions/teardown-rocm
	manywheel-py3_12-rocm6_0-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_12-rocm6_0-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.0
	GPU_ARCH_VERSION: 6.0
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.0-main
	DESIRED_PYTHON: "3.12"
	build_name: manywheel-py3_12-rocm6_0
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

	manywheel-py3_12-rocm6_1-build:
	if: ${{ github.repository_owner == 'pytorch' }}
	uses: ./.github/workflows/_binary-build-linux.yml
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.1
	GPU_ARCH_VERSION: 6.1
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-main
	DESIRED_PYTHON: "3.12"
	build_name: manywheel-py3_12-rocm6_1
	build_environment: linux-binary-manywheel
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	manywheel-py3_12-rocm6_1-test: # Testing
	if: ${{ github.repository_owner == 'pytorch' }}
	needs: manywheel-py3_12-rocm6_1-build
	runs-on: linux.rocm.gpu
	timeout-minutes: 240
	env:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.1
	GPU_ARCH_VERSION: 6.1
	GPU_ARCH_TYPE: rocm
	SKIP_ALL_TESTS: 1
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-main
	DESIRED_PYTHON: "3.12"
	steps:
	- name: Setup ROCm
	uses: ./.github/actions/setup-rocm
	- uses: actions/download-artifact@v3
	name: Download Build Artifacts
	with:
	name: manywheel-py3_12-rocm6_1
	path: "${{ runner.temp }}/artifacts/"
	- name: Checkout PyTorch
	uses: malfet/checkout@silent-checkout
	with:
	ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha \|\| github.sha }}
	submodules: recursive
	path: pytorch
	quiet-checkout: true
	- name: Clean PyTorch checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: pytorch
	- name: Checkout nWEIdia/builder
	uses: malfet/checkout@silent-checkout
	with:
	ref: cudnn_v9
	submodules: recursive
	repository: nWEIdia/builder
	path: builder
	quiet-checkout: true
	- name: Clean nWEIdia/builder checkout
	run: \|
	# Remove any artifacts from the previous checkouts
	git clean -fxd
	working-directory: builder
	- name: ROCm set GPU_FLAG
	run: \|
	echo "GPU_FLAG=--device=/dev/mem --device=/dev/kfd --device=/dev/dri --group-add video --group-add daemon" >> "${GITHUB_ENV}"
	- name: Pull Docker image
	uses: pytorch/test-infra/.github/actions/pull-docker-image@main
	with:
	docker-image: pytorch/manylinux-builder:rocm6.1-main
	- name: Test Pytorch binary
	uses: ./pytorch/.github/actions/test-pytorch-binary
	- name: Teardown ROCm
	uses: ./.github/actions/teardown-rocm
	manywheel-py3_12-rocm6_1-upload: # Uploading
	if: ${{ github.repository_owner == 'pytorch' }}
	permissions:
	id-token: write
	contents: read
	needs: manywheel-py3_12-rocm6_1-test
	with:
	PYTORCH_ROOT: /pytorch
	BUILDER_ROOT: /builder
	PACKAGE_TYPE: manywheel
	# TODO: This is a legacy variable that we eventually want to get rid of in
	# favor of GPU_ARCH_VERSION
	DESIRED_CUDA: rocm6.1
	GPU_ARCH_VERSION: 6.1
	GPU_ARCH_TYPE: rocm
	DOCKER_IMAGE: pytorch/manylinux-builder:rocm6.1-main
	DESIRED_PYTHON: "3.12"
	build_name: manywheel-py3_12-rocm6_1
	secrets:
	github-token: ${{ secrets.GITHUB_TOKEN }}
	conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }}
	conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }}
	uses: ./.github/workflows/_binary-upload.yml

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

linux-binary-manywheel #3676

Workflow file

linux-binary-manywheel #3676

Jobs

Run details

Workflow file for this run