mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
Continue to build nightly CUDA 12.9 for internal (#163029)
Revert part of https://github.com/pytorch/pytorch/pull/161916 to continue building CUDA 12.9 nightly Pull Request resolved: https://github.com/pytorch/pytorch/pull/163029 Approved by: https://github.com/malfet
This commit is contained in:
Huy Do
PyTorch MergeBot
@ -8,6 +8,8 @@ if [[ "$GPU_ARCH_VERSION" == *"12.6"* ]]; then
|
|||||||
export TORCH_CUDA_ARCH_LIST="8.0;9.0"
|
export TORCH_CUDA_ARCH_LIST="8.0;9.0"
|
||||||
elif [[ "$GPU_ARCH_VERSION" == *"12.8"* ]]; then
|
elif [[ "$GPU_ARCH_VERSION" == *"12.8"* ]]; then
|
||||||
export TORCH_CUDA_ARCH_LIST="8.0;9.0;10.0;12.0"
|
export TORCH_CUDA_ARCH_LIST="8.0;9.0;10.0;12.0"
|
||||||
|
elif [[ "$GPU_ARCH_VERSION" == *"12.9"* ]]; then
|
||||||
|
export TORCH_CUDA_ARCH_LIST="8.0;9.0;10.0;12.0"
|
||||||
elif [[ "$GPU_ARCH_VERSION" == *"13.0"* ]]; then
|
elif [[ "$GPU_ARCH_VERSION" == *"13.0"* ]]; then
|
||||||
export TORCH_CUDA_ARCH_LIST="8.0;9.0;10.0;11.0;12.0+PTX"
|
export TORCH_CUDA_ARCH_LIST="8.0;9.0;10.0;11.0;12.0+PTX"
|
||||||
fi
|
fi
|
||||||
|
|||||||
26
.github/scripts/generate_binary_build_matrix.py
vendored
26
.github/scripts/generate_binary_build_matrix.py
vendored
@ -16,16 +16,18 @@ from typing import Optional
|
|||||||
|
|
||||||
|
|
||||||
# NOTE: Please also update the CUDA sources in `PIP_SOURCES` in tools/nightly.py when changing this
|
# NOTE: Please also update the CUDA sources in `PIP_SOURCES` in tools/nightly.py when changing this
|
||||||
CUDA_ARCHES = ["12.6", "12.8", "13.0"]
|
CUDA_ARCHES = ["12.6", "12.8", "12.9", "13.0"]
|
||||||
CUDA_STABLE = "12.8"
|
CUDA_STABLE = "12.8"
|
||||||
CUDA_ARCHES_FULL_VERSION = {
|
CUDA_ARCHES_FULL_VERSION = {
|
||||||
"12.6": "12.6.3",
|
"12.6": "12.6.3",
|
||||||
"12.8": "12.8.1",
|
"12.8": "12.8.1",
|
||||||
|
"12.9": "12.9.1",
|
||||||
"13.0": "13.0.0",
|
"13.0": "13.0.0",
|
||||||
}
|
}
|
||||||
CUDA_ARCHES_CUDNN_VERSION = {
|
CUDA_ARCHES_CUDNN_VERSION = {
|
||||||
"12.6": "9",
|
"12.6": "9",
|
||||||
"12.8": "9",
|
"12.8": "9",
|
||||||
|
"12.9": "9",
|
||||||
"13.0": "9",
|
"13.0": "9",
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -38,7 +40,7 @@ CPU_AARCH64_ARCH = ["cpu-aarch64"]
|
|||||||
|
|
||||||
CPU_S390X_ARCH = ["cpu-s390x"]
|
CPU_S390X_ARCH = ["cpu-s390x"]
|
||||||
|
|
||||||
CUDA_AARCH64_ARCHES = ["12.6-aarch64", "12.8-aarch64", "13.0-aarch64"]
|
CUDA_AARCH64_ARCHES = ["12.6-aarch64", "12.8-aarch64", "12.9-aarch64", "13.0-aarch64"]
|
||||||
|
|
||||||
|
|
||||||
PYTORCH_EXTRA_INSTALL_REQUIREMENTS = {
|
PYTORCH_EXTRA_INSTALL_REQUIREMENTS = {
|
||||||
@ -76,6 +78,23 @@ PYTORCH_EXTRA_INSTALL_REQUIREMENTS = {
|
|||||||
"nvidia-nvjitlink-cu12==12.8.93; platform_system == 'Linux' | "
|
"nvidia-nvjitlink-cu12==12.8.93; platform_system == 'Linux' | "
|
||||||
"nvidia-cufile-cu12==1.13.1.3; platform_system == 'Linux'"
|
"nvidia-cufile-cu12==1.13.1.3; platform_system == 'Linux'"
|
||||||
),
|
),
|
||||||
|
"12.9": (
|
||||||
|
"nvidia-cuda-nvrtc-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | "
|
||||||
|
"nvidia-cuda-runtime-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | "
|
||||||
|
"nvidia-cuda-cupti-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | "
|
||||||
|
"nvidia-cudnn-cu12==9.10.2.21; platform_system == 'Linux' and platform_machine == 'x86_64' | "
|
||||||
|
"nvidia-cublas-cu12==12.9.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | "
|
||||||
|
"nvidia-cufft-cu12==11.4.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | "
|
||||||
|
"nvidia-curand-cu12==10.3.10.19; platform_system == 'Linux' and platform_machine == 'x86_64' | "
|
||||||
|
"nvidia-cusolver-cu12==11.7.5.82; platform_system == 'Linux' and platform_machine == 'x86_64' | "
|
||||||
|
"nvidia-cusparse-cu12==12.5.10.65; platform_system == 'Linux' and platform_machine == 'x86_64' | "
|
||||||
|
"nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' and platform_machine == 'x86_64' | "
|
||||||
|
"nvidia-nccl-cu12==2.27.5; platform_system == 'Linux' and platform_machine == 'x86_64' | "
|
||||||
|
"nvidia-nvshmem-cu12==3.3.20; platform_system == 'Linux' and platform_machine == 'x86_64' | "
|
||||||
|
"nvidia-nvtx-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | "
|
||||||
|
"nvidia-nvjitlink-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | "
|
||||||
|
"nvidia-cufile-cu12==1.14.1.1; platform_system == 'Linux' and platform_machine == 'x86_64'"
|
||||||
|
),
|
||||||
"13.0": (
|
"13.0": (
|
||||||
"nvidia-cuda-nvrtc==13.0.48; platform_system == 'Linux' | "
|
"nvidia-cuda-nvrtc==13.0.48; platform_system == 'Linux' | "
|
||||||
"nvidia-cuda-runtime==13.0.48; platform_system == 'Linux' | "
|
"nvidia-cuda-runtime==13.0.48; platform_system == 'Linux' | "
|
||||||
@ -322,7 +341,7 @@ def generate_wheels_matrix(
|
|||||||
# cuda linux wheels require PYTORCH_EXTRA_INSTALL_REQUIREMENTS to install
|
# cuda linux wheels require PYTORCH_EXTRA_INSTALL_REQUIREMENTS to install
|
||||||
|
|
||||||
if (
|
if (
|
||||||
arch_version in ["13.0", "12.8", "12.6"]
|
arch_version in ["13.0", "12.9", "12.8", "12.6"]
|
||||||
and os == "linux"
|
and os == "linux"
|
||||||
or arch_version in CUDA_AARCH64_ARCHES
|
or arch_version in CUDA_AARCH64_ARCHES
|
||||||
):
|
):
|
||||||
@ -386,5 +405,6 @@ def generate_wheels_matrix(
|
|||||||
|
|
||||||
|
|
||||||
validate_nccl_dep_consistency("13.0")
|
validate_nccl_dep_consistency("13.0")
|
||||||
|
validate_nccl_dep_consistency("12.9")
|
||||||
validate_nccl_dep_consistency("12.8")
|
validate_nccl_dep_consistency("12.8")
|
||||||
validate_nccl_dep_consistency("12.6")
|
validate_nccl_dep_consistency("12.6")
|
||||||
|
|||||||
4
.github/workflows/build-manywheel-images.yml
vendored
4
.github/workflows/build-manywheel-images.yml
vendored
@ -46,10 +46,12 @@ jobs:
|
|||||||
fail-fast: false
|
fail-fast: false
|
||||||
matrix:
|
matrix:
|
||||||
include: [
|
include: [
|
||||||
{ name: "manylinux2_28-builder", tag: "cuda13.0", runner: "linux.9xlarge.ephemeral" },
|
{ name: "manylinux2_28-builder", tag: "cuda13.0", runner: "linux.9xlarge.ephemeral" },
|
||||||
{ name: "manylinux2_28-builder", tag: "cuda12.8", runner: "linux.9xlarge.ephemeral" },
|
{ name: "manylinux2_28-builder", tag: "cuda12.8", runner: "linux.9xlarge.ephemeral" },
|
||||||
|
{ name: "manylinux2_28-builder", tag: "cuda12.9", runner: "linux.9xlarge.ephemeral" },
|
||||||
{ name: "manylinux2_28-builder", tag: "cuda12.6", runner: "linux.9xlarge.ephemeral" },
|
{ name: "manylinux2_28-builder", tag: "cuda12.6", runner: "linux.9xlarge.ephemeral" },
|
||||||
{ name: "manylinuxaarch64-builder", tag: "cuda13.0", runner: "linux.arm64.2xlarge.ephemeral" },
|
{ name: "manylinuxaarch64-builder", tag: "cuda13.0", runner: "linux.arm64.2xlarge.ephemeral" },
|
||||||
|
{ name: "manylinuxaarch64-builder", tag: "cuda12.9", runner: "linux.arm64.2xlarge.ephemeral" },
|
||||||
{ name: "manylinuxaarch64-builder", tag: "cuda12.8", runner: "linux.arm64.2xlarge.ephemeral" },
|
{ name: "manylinuxaarch64-builder", tag: "cuda12.8", runner: "linux.arm64.2xlarge.ephemeral" },
|
||||||
{ name: "manylinuxaarch64-builder", tag: "cuda12.6", runner: "linux.arm64.2xlarge.ephemeral" },
|
{ name: "manylinuxaarch64-builder", tag: "cuda12.6", runner: "linux.arm64.2xlarge.ephemeral" },
|
||||||
{ name: "manylinux2_28-builder", tag: "rocm6.4", runner: "linux.9xlarge.ephemeral" },
|
{ name: "manylinux2_28-builder", tag: "rocm6.4", runner: "linux.9xlarge.ephemeral" },
|
||||||
|
|||||||
322
.github/workflows/generated-linux-aarch64-binary-manywheel-nightly.yml
generated
vendored
322
.github/workflows/generated-linux-aarch64-binary-manywheel-nightly.yml
generated
vendored
@ -204,6 +204,52 @@ jobs:
|
|||||||
github-token: ${{ secrets.GITHUB_TOKEN }}
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
uses: ./.github/workflows/_binary-upload.yml
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
|
manywheel-py3_10-cuda-aarch64-12_9-build:
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
|
needs: get-label-type
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9-aarch64"
|
||||||
|
GPU_ARCH_TYPE: cuda-aarch64
|
||||||
|
DOCKER_IMAGE: manylinuxaarch64-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.10"
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
runs_on: linux.arm64.r7g.12xlarge.memory
|
||||||
|
ALPINE_IMAGE: "arm64v8/alpine"
|
||||||
|
build_name: manywheel-py3_10-cuda-aarch64-12_9
|
||||||
|
build_environment: linux-aarch64-binary-manywheel
|
||||||
|
PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-runtime-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-cupti-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cudnn-cu12==9.10.2.21; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cublas-cu12==12.9.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufft-cu12==11.4.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-curand-cu12==10.3.10.19; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusolver-cu12==11.7.5.82; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparse-cu12==12.5.10.65; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nccl-cu12==2.27.5; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvshmem-cu12==3.3.20; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvtx-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvjitlink-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufile-cu12==1.14.1.1; platform_system == 'Linux' and platform_machine == 'x86_64'
|
||||||
|
timeout-minutes: 420
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_10-cuda-aarch64-12_9-upload: # Uploading
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
permissions:
|
||||||
|
id-token: write
|
||||||
|
contents: read
|
||||||
|
needs: manywheel-py3_10-cuda-aarch64-12_9-build
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9-aarch64"
|
||||||
|
GPU_ARCH_TYPE: cuda-aarch64
|
||||||
|
DOCKER_IMAGE: manylinuxaarch64-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.10"
|
||||||
|
build_name: manywheel-py3_10-cuda-aarch64-12_9
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
manywheel-py3_10-cuda-aarch64-13_0-build:
|
manywheel-py3_10-cuda-aarch64-13_0-build:
|
||||||
if: ${{ github.repository_owner == 'pytorch' }}
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
uses: ./.github/workflows/_binary-build-linux.yml
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
@ -407,6 +453,52 @@ jobs:
|
|||||||
github-token: ${{ secrets.GITHUB_TOKEN }}
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
uses: ./.github/workflows/_binary-upload.yml
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
|
manywheel-py3_11-cuda-aarch64-12_9-build:
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
|
needs: get-label-type
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9-aarch64"
|
||||||
|
GPU_ARCH_TYPE: cuda-aarch64
|
||||||
|
DOCKER_IMAGE: manylinuxaarch64-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.11"
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
runs_on: linux.arm64.r7g.12xlarge.memory
|
||||||
|
ALPINE_IMAGE: "arm64v8/alpine"
|
||||||
|
build_name: manywheel-py3_11-cuda-aarch64-12_9
|
||||||
|
build_environment: linux-aarch64-binary-manywheel
|
||||||
|
PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-runtime-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-cupti-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cudnn-cu12==9.10.2.21; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cublas-cu12==12.9.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufft-cu12==11.4.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-curand-cu12==10.3.10.19; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusolver-cu12==11.7.5.82; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparse-cu12==12.5.10.65; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nccl-cu12==2.27.5; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvshmem-cu12==3.3.20; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvtx-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvjitlink-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufile-cu12==1.14.1.1; platform_system == 'Linux' and platform_machine == 'x86_64'
|
||||||
|
timeout-minutes: 420
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_11-cuda-aarch64-12_9-upload: # Uploading
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
permissions:
|
||||||
|
id-token: write
|
||||||
|
contents: read
|
||||||
|
needs: manywheel-py3_11-cuda-aarch64-12_9-build
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9-aarch64"
|
||||||
|
GPU_ARCH_TYPE: cuda-aarch64
|
||||||
|
DOCKER_IMAGE: manylinuxaarch64-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.11"
|
||||||
|
build_name: manywheel-py3_11-cuda-aarch64-12_9
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
manywheel-py3_11-cuda-aarch64-13_0-build:
|
manywheel-py3_11-cuda-aarch64-13_0-build:
|
||||||
if: ${{ github.repository_owner == 'pytorch' }}
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
uses: ./.github/workflows/_binary-build-linux.yml
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
@ -610,6 +702,52 @@ jobs:
|
|||||||
github-token: ${{ secrets.GITHUB_TOKEN }}
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
uses: ./.github/workflows/_binary-upload.yml
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
|
manywheel-py3_12-cuda-aarch64-12_9-build:
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
|
needs: get-label-type
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9-aarch64"
|
||||||
|
GPU_ARCH_TYPE: cuda-aarch64
|
||||||
|
DOCKER_IMAGE: manylinuxaarch64-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.12"
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
runs_on: linux.arm64.r7g.12xlarge.memory
|
||||||
|
ALPINE_IMAGE: "arm64v8/alpine"
|
||||||
|
build_name: manywheel-py3_12-cuda-aarch64-12_9
|
||||||
|
build_environment: linux-aarch64-binary-manywheel
|
||||||
|
PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-runtime-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-cupti-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cudnn-cu12==9.10.2.21; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cublas-cu12==12.9.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufft-cu12==11.4.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-curand-cu12==10.3.10.19; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusolver-cu12==11.7.5.82; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparse-cu12==12.5.10.65; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nccl-cu12==2.27.5; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvshmem-cu12==3.3.20; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvtx-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvjitlink-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufile-cu12==1.14.1.1; platform_system == 'Linux' and platform_machine == 'x86_64'
|
||||||
|
timeout-minutes: 420
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_12-cuda-aarch64-12_9-upload: # Uploading
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
permissions:
|
||||||
|
id-token: write
|
||||||
|
contents: read
|
||||||
|
needs: manywheel-py3_12-cuda-aarch64-12_9-build
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9-aarch64"
|
||||||
|
GPU_ARCH_TYPE: cuda-aarch64
|
||||||
|
DOCKER_IMAGE: manylinuxaarch64-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.12"
|
||||||
|
build_name: manywheel-py3_12-cuda-aarch64-12_9
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
manywheel-py3_12-cuda-aarch64-13_0-build:
|
manywheel-py3_12-cuda-aarch64-13_0-build:
|
||||||
if: ${{ github.repository_owner == 'pytorch' }}
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
uses: ./.github/workflows/_binary-build-linux.yml
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
@ -813,6 +951,52 @@ jobs:
|
|||||||
github-token: ${{ secrets.GITHUB_TOKEN }}
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
uses: ./.github/workflows/_binary-upload.yml
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
|
manywheel-py3_13-cuda-aarch64-12_9-build:
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
|
needs: get-label-type
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9-aarch64"
|
||||||
|
GPU_ARCH_TYPE: cuda-aarch64
|
||||||
|
DOCKER_IMAGE: manylinuxaarch64-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.13"
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
runs_on: linux.arm64.r7g.12xlarge.memory
|
||||||
|
ALPINE_IMAGE: "arm64v8/alpine"
|
||||||
|
build_name: manywheel-py3_13-cuda-aarch64-12_9
|
||||||
|
build_environment: linux-aarch64-binary-manywheel
|
||||||
|
PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-runtime-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-cupti-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cudnn-cu12==9.10.2.21; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cublas-cu12==12.9.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufft-cu12==11.4.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-curand-cu12==10.3.10.19; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusolver-cu12==11.7.5.82; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparse-cu12==12.5.10.65; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nccl-cu12==2.27.5; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvshmem-cu12==3.3.20; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvtx-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvjitlink-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufile-cu12==1.14.1.1; platform_system == 'Linux' and platform_machine == 'x86_64'
|
||||||
|
timeout-minutes: 420
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_13-cuda-aarch64-12_9-upload: # Uploading
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
permissions:
|
||||||
|
id-token: write
|
||||||
|
contents: read
|
||||||
|
needs: manywheel-py3_13-cuda-aarch64-12_9-build
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9-aarch64"
|
||||||
|
GPU_ARCH_TYPE: cuda-aarch64
|
||||||
|
DOCKER_IMAGE: manylinuxaarch64-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.13"
|
||||||
|
build_name: manywheel-py3_13-cuda-aarch64-12_9
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
manywheel-py3_13-cuda-aarch64-13_0-build:
|
manywheel-py3_13-cuda-aarch64-13_0-build:
|
||||||
if: ${{ github.repository_owner == 'pytorch' }}
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
uses: ./.github/workflows/_binary-build-linux.yml
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
@ -1016,6 +1200,52 @@ jobs:
|
|||||||
github-token: ${{ secrets.GITHUB_TOKEN }}
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
uses: ./.github/workflows/_binary-upload.yml
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
|
manywheel-py3_13t-cuda-aarch64-12_9-build:
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
|
needs: get-label-type
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9-aarch64"
|
||||||
|
GPU_ARCH_TYPE: cuda-aarch64
|
||||||
|
DOCKER_IMAGE: manylinuxaarch64-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.13t"
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
runs_on: linux.arm64.r7g.12xlarge.memory
|
||||||
|
ALPINE_IMAGE: "arm64v8/alpine"
|
||||||
|
build_name: manywheel-py3_13t-cuda-aarch64-12_9
|
||||||
|
build_environment: linux-aarch64-binary-manywheel
|
||||||
|
PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-runtime-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-cupti-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cudnn-cu12==9.10.2.21; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cublas-cu12==12.9.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufft-cu12==11.4.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-curand-cu12==10.3.10.19; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusolver-cu12==11.7.5.82; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparse-cu12==12.5.10.65; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nccl-cu12==2.27.5; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvshmem-cu12==3.3.20; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvtx-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvjitlink-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufile-cu12==1.14.1.1; platform_system == 'Linux' and platform_machine == 'x86_64'
|
||||||
|
timeout-minutes: 420
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_13t-cuda-aarch64-12_9-upload: # Uploading
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
permissions:
|
||||||
|
id-token: write
|
||||||
|
contents: read
|
||||||
|
needs: manywheel-py3_13t-cuda-aarch64-12_9-build
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9-aarch64"
|
||||||
|
GPU_ARCH_TYPE: cuda-aarch64
|
||||||
|
DOCKER_IMAGE: manylinuxaarch64-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.13t"
|
||||||
|
build_name: manywheel-py3_13t-cuda-aarch64-12_9
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
manywheel-py3_13t-cuda-aarch64-13_0-build:
|
manywheel-py3_13t-cuda-aarch64-13_0-build:
|
||||||
if: ${{ github.repository_owner == 'pytorch' }}
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
uses: ./.github/workflows/_binary-build-linux.yml
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
@ -1219,6 +1449,52 @@ jobs:
|
|||||||
github-token: ${{ secrets.GITHUB_TOKEN }}
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
uses: ./.github/workflows/_binary-upload.yml
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
|
manywheel-py3_14-cuda-aarch64-12_9-build:
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
|
needs: get-label-type
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9-aarch64"
|
||||||
|
GPU_ARCH_TYPE: cuda-aarch64
|
||||||
|
DOCKER_IMAGE: manylinuxaarch64-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.14"
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
runs_on: linux.arm64.r7g.12xlarge.memory
|
||||||
|
ALPINE_IMAGE: "arm64v8/alpine"
|
||||||
|
build_name: manywheel-py3_14-cuda-aarch64-12_9
|
||||||
|
build_environment: linux-aarch64-binary-manywheel
|
||||||
|
PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-runtime-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-cupti-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cudnn-cu12==9.10.2.21; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cublas-cu12==12.9.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufft-cu12==11.4.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-curand-cu12==10.3.10.19; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusolver-cu12==11.7.5.82; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparse-cu12==12.5.10.65; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nccl-cu12==2.27.5; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvshmem-cu12==3.3.20; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvtx-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvjitlink-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufile-cu12==1.14.1.1; platform_system == 'Linux' and platform_machine == 'x86_64'
|
||||||
|
timeout-minutes: 420
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_14-cuda-aarch64-12_9-upload: # Uploading
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
permissions:
|
||||||
|
id-token: write
|
||||||
|
contents: read
|
||||||
|
needs: manywheel-py3_14-cuda-aarch64-12_9-build
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9-aarch64"
|
||||||
|
GPU_ARCH_TYPE: cuda-aarch64
|
||||||
|
DOCKER_IMAGE: manylinuxaarch64-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.14"
|
||||||
|
build_name: manywheel-py3_14-cuda-aarch64-12_9
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
manywheel-py3_14-cuda-aarch64-13_0-build:
|
manywheel-py3_14-cuda-aarch64-13_0-build:
|
||||||
if: ${{ github.repository_owner == 'pytorch' }}
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
uses: ./.github/workflows/_binary-build-linux.yml
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
@ -1422,6 +1698,52 @@ jobs:
|
|||||||
github-token: ${{ secrets.GITHUB_TOKEN }}
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
uses: ./.github/workflows/_binary-upload.yml
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
|
manywheel-py3_14t-cuda-aarch64-12_9-build:
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
|
needs: get-label-type
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9-aarch64"
|
||||||
|
GPU_ARCH_TYPE: cuda-aarch64
|
||||||
|
DOCKER_IMAGE: manylinuxaarch64-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.14t"
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
runs_on: linux.arm64.r7g.12xlarge.memory
|
||||||
|
ALPINE_IMAGE: "arm64v8/alpine"
|
||||||
|
build_name: manywheel-py3_14t-cuda-aarch64-12_9
|
||||||
|
build_environment: linux-aarch64-binary-manywheel
|
||||||
|
PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-runtime-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-cupti-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cudnn-cu12==9.10.2.21; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cublas-cu12==12.9.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufft-cu12==11.4.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-curand-cu12==10.3.10.19; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusolver-cu12==11.7.5.82; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparse-cu12==12.5.10.65; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nccl-cu12==2.27.5; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvshmem-cu12==3.3.20; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvtx-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvjitlink-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufile-cu12==1.14.1.1; platform_system == 'Linux' and platform_machine == 'x86_64'
|
||||||
|
timeout-minutes: 420
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_14t-cuda-aarch64-12_9-upload: # Uploading
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
permissions:
|
||||||
|
id-token: write
|
||||||
|
contents: read
|
||||||
|
needs: manywheel-py3_14t-cuda-aarch64-12_9-build
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9-aarch64"
|
||||||
|
GPU_ARCH_TYPE: cuda-aarch64
|
||||||
|
DOCKER_IMAGE: manylinuxaarch64-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.14t"
|
||||||
|
build_name: manywheel-py3_14t-cuda-aarch64-12_9
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
manywheel-py3_14t-cuda-aarch64-13_0-build:
|
manywheel-py3_14t-cuda-aarch64-13_0-build:
|
||||||
if: ${{ github.repository_owner == 'pytorch' }}
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
uses: ./.github/workflows/_binary-build-linux.yml
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
|
|||||||
68
.github/workflows/generated-linux-binary-libtorch-nightly.yml
generated
vendored
68
.github/workflows/generated-linux-binary-libtorch-nightly.yml
generated
vendored
@ -248,6 +248,74 @@ jobs:
|
|||||||
github-token: ${{ secrets.GITHUB_TOKEN }}
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
uses: ./.github/workflows/_binary-upload.yml
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
|
libtorch-cuda12_9-shared-with-deps-release-build:
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
|
needs: get-label-type
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
PACKAGE_TYPE: libtorch
|
||||||
|
# TODO: This is a legacy variable that we eventually want to get rid of in
|
||||||
|
# favor of GPU_ARCH_VERSION
|
||||||
|
DESIRED_CUDA: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: libtorch-cxx11-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
LIBTORCH_CONFIG: release
|
||||||
|
LIBTORCH_VARIANT: shared-with-deps
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
build_name: libtorch-cuda12_9-shared-with-deps-release
|
||||||
|
build_environment: linux-binary-libtorch
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
libtorch-cuda12_9-shared-with-deps-release-test: # Testing
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
needs:
|
||||||
|
- libtorch-cuda12_9-shared-with-deps-release-build
|
||||||
|
- get-label-type
|
||||||
|
uses: ./.github/workflows/_binary-test-linux.yml
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
PACKAGE_TYPE: libtorch
|
||||||
|
# TODO: This is a legacy variable that we eventually want to get rid of in
|
||||||
|
# favor of GPU_ARCH_VERSION
|
||||||
|
DESIRED_CUDA: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: libtorch-cxx11-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
LIBTORCH_CONFIG: release
|
||||||
|
LIBTORCH_VARIANT: shared-with-deps
|
||||||
|
build_name: libtorch-cuda12_9-shared-with-deps-release
|
||||||
|
build_environment: linux-binary-libtorch
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
runs_on: linux.g4dn.4xlarge.nvidia.gpu # 12.8+ builds need sm_70+ runner
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
libtorch-cuda12_9-shared-with-deps-release-upload: # Uploading
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
permissions:
|
||||||
|
id-token: write
|
||||||
|
contents: read
|
||||||
|
needs: libtorch-cuda12_9-shared-with-deps-release-test
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
PACKAGE_TYPE: libtorch
|
||||||
|
# TODO: This is a legacy variable that we eventually want to get rid of in
|
||||||
|
# favor of GPU_ARCH_VERSION
|
||||||
|
DESIRED_CUDA: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: libtorch-cxx11-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
LIBTORCH_CONFIG: release
|
||||||
|
LIBTORCH_VARIANT: shared-with-deps
|
||||||
|
build_name: libtorch-cuda12_9-shared-with-deps-release
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
libtorch-cuda13_0-shared-with-deps-release-build:
|
libtorch-cuda13_0-shared-with-deps-release-build:
|
||||||
if: ${{ github.repository_owner == 'pytorch' }}
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
uses: ./.github/workflows/_binary-build-linux.yml
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
|
|||||||
462
.github/workflows/generated-linux-binary-manywheel-nightly.yml
generated
vendored
462
.github/workflows/generated-linux-binary-manywheel-nightly.yml
generated
vendored
@ -241,6 +241,72 @@ jobs:
|
|||||||
github-token: ${{ secrets.GITHUB_TOKEN }}
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
uses: ./.github/workflows/_binary-upload.yml
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
|
manywheel-py3_10-cuda12_9-build:
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
|
needs: get-label-type
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.10"
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
build_name: manywheel-py3_10-cuda12_9
|
||||||
|
build_environment: linux-binary-manywheel
|
||||||
|
PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-runtime-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-cupti-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cudnn-cu12==9.10.2.21; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cublas-cu12==12.9.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufft-cu12==11.4.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-curand-cu12==10.3.10.19; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusolver-cu12==11.7.5.82; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparse-cu12==12.5.10.65; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nccl-cu12==2.27.5; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvshmem-cu12==3.3.20; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvtx-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvjitlink-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufile-cu12==1.14.1.1; platform_system == 'Linux' and platform_machine == 'x86_64'
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_10-cuda12_9-test: # Testing
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
needs:
|
||||||
|
- manywheel-py3_10-cuda12_9-build
|
||||||
|
- get-label-type
|
||||||
|
uses: ./.github/workflows/_binary-test-linux.yml
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.10"
|
||||||
|
build_name: manywheel-py3_10-cuda12_9
|
||||||
|
build_environment: linux-binary-manywheel
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
runs_on: linux.g4dn.4xlarge.nvidia.gpu # 12.8+ builds need sm_70+ runner
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_10-cuda12_9-upload: # Uploading
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
permissions:
|
||||||
|
id-token: write
|
||||||
|
contents: read
|
||||||
|
needs: manywheel-py3_10-cuda12_9-test
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.10"
|
||||||
|
build_name: manywheel-py3_10-cuda12_9
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
manywheel-py3_10-cuda13_0-build:
|
manywheel-py3_10-cuda13_0-build:
|
||||||
if: ${{ github.repository_owner == 'pytorch' }}
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
uses: ./.github/workflows/_binary-build-linux.yml
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
@ -841,6 +907,72 @@ jobs:
|
|||||||
github-token: ${{ secrets.GITHUB_TOKEN }}
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
uses: ./.github/workflows/_binary-upload.yml
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
|
manywheel-py3_11-cuda12_9-build:
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
|
needs: get-label-type
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.11"
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
build_name: manywheel-py3_11-cuda12_9
|
||||||
|
build_environment: linux-binary-manywheel
|
||||||
|
PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-runtime-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-cupti-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cudnn-cu12==9.10.2.21; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cublas-cu12==12.9.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufft-cu12==11.4.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-curand-cu12==10.3.10.19; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusolver-cu12==11.7.5.82; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparse-cu12==12.5.10.65; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nccl-cu12==2.27.5; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvshmem-cu12==3.3.20; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvtx-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvjitlink-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufile-cu12==1.14.1.1; platform_system == 'Linux' and platform_machine == 'x86_64'
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_11-cuda12_9-test: # Testing
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
needs:
|
||||||
|
- manywheel-py3_11-cuda12_9-build
|
||||||
|
- get-label-type
|
||||||
|
uses: ./.github/workflows/_binary-test-linux.yml
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.11"
|
||||||
|
build_name: manywheel-py3_11-cuda12_9
|
||||||
|
build_environment: linux-binary-manywheel
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
runs_on: linux.g4dn.4xlarge.nvidia.gpu # 12.8+ builds need sm_70+ runner
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_11-cuda12_9-upload: # Uploading
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
permissions:
|
||||||
|
id-token: write
|
||||||
|
contents: read
|
||||||
|
needs: manywheel-py3_11-cuda12_9-test
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.11"
|
||||||
|
build_name: manywheel-py3_11-cuda12_9
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
manywheel-py3_11-cuda13_0-build:
|
manywheel-py3_11-cuda13_0-build:
|
||||||
if: ${{ github.repository_owner == 'pytorch' }}
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
uses: ./.github/workflows/_binary-build-linux.yml
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
@ -1441,6 +1573,72 @@ jobs:
|
|||||||
github-token: ${{ secrets.GITHUB_TOKEN }}
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
uses: ./.github/workflows/_binary-upload.yml
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
|
manywheel-py3_12-cuda12_9-build:
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
|
needs: get-label-type
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.12"
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
build_name: manywheel-py3_12-cuda12_9
|
||||||
|
build_environment: linux-binary-manywheel
|
||||||
|
PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-runtime-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-cupti-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cudnn-cu12==9.10.2.21; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cublas-cu12==12.9.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufft-cu12==11.4.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-curand-cu12==10.3.10.19; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusolver-cu12==11.7.5.82; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparse-cu12==12.5.10.65; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nccl-cu12==2.27.5; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvshmem-cu12==3.3.20; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvtx-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvjitlink-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufile-cu12==1.14.1.1; platform_system == 'Linux' and platform_machine == 'x86_64'
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_12-cuda12_9-test: # Testing
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
needs:
|
||||||
|
- manywheel-py3_12-cuda12_9-build
|
||||||
|
- get-label-type
|
||||||
|
uses: ./.github/workflows/_binary-test-linux.yml
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.12"
|
||||||
|
build_name: manywheel-py3_12-cuda12_9
|
||||||
|
build_environment: linux-binary-manywheel
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
runs_on: linux.g4dn.4xlarge.nvidia.gpu # 12.8+ builds need sm_70+ runner
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_12-cuda12_9-upload: # Uploading
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
permissions:
|
||||||
|
id-token: write
|
||||||
|
contents: read
|
||||||
|
needs: manywheel-py3_12-cuda12_9-test
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.12"
|
||||||
|
build_name: manywheel-py3_12-cuda12_9
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
manywheel-py3_12-cuda13_0-build:
|
manywheel-py3_12-cuda13_0-build:
|
||||||
if: ${{ github.repository_owner == 'pytorch' }}
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
uses: ./.github/workflows/_binary-build-linux.yml
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
@ -2041,6 +2239,72 @@ jobs:
|
|||||||
github-token: ${{ secrets.GITHUB_TOKEN }}
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
uses: ./.github/workflows/_binary-upload.yml
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
|
manywheel-py3_13-cuda12_9-build:
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
|
needs: get-label-type
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.13"
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
build_name: manywheel-py3_13-cuda12_9
|
||||||
|
build_environment: linux-binary-manywheel
|
||||||
|
PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-runtime-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-cupti-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cudnn-cu12==9.10.2.21; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cublas-cu12==12.9.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufft-cu12==11.4.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-curand-cu12==10.3.10.19; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusolver-cu12==11.7.5.82; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparse-cu12==12.5.10.65; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nccl-cu12==2.27.5; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvshmem-cu12==3.3.20; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvtx-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvjitlink-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufile-cu12==1.14.1.1; platform_system == 'Linux' and platform_machine == 'x86_64'
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_13-cuda12_9-test: # Testing
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
needs:
|
||||||
|
- manywheel-py3_13-cuda12_9-build
|
||||||
|
- get-label-type
|
||||||
|
uses: ./.github/workflows/_binary-test-linux.yml
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.13"
|
||||||
|
build_name: manywheel-py3_13-cuda12_9
|
||||||
|
build_environment: linux-binary-manywheel
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
runs_on: linux.g4dn.4xlarge.nvidia.gpu # 12.8+ builds need sm_70+ runner
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_13-cuda12_9-upload: # Uploading
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
permissions:
|
||||||
|
id-token: write
|
||||||
|
contents: read
|
||||||
|
needs: manywheel-py3_13-cuda12_9-test
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.13"
|
||||||
|
build_name: manywheel-py3_13-cuda12_9
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
manywheel-py3_13-cuda13_0-build:
|
manywheel-py3_13-cuda13_0-build:
|
||||||
if: ${{ github.repository_owner == 'pytorch' }}
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
uses: ./.github/workflows/_binary-build-linux.yml
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
@ -2641,6 +2905,72 @@ jobs:
|
|||||||
github-token: ${{ secrets.GITHUB_TOKEN }}
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
uses: ./.github/workflows/_binary-upload.yml
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
|
manywheel-py3_13t-cuda12_9-build:
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
|
needs: get-label-type
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.13t"
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
build_name: manywheel-py3_13t-cuda12_9
|
||||||
|
build_environment: linux-binary-manywheel
|
||||||
|
PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-runtime-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-cupti-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cudnn-cu12==9.10.2.21; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cublas-cu12==12.9.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufft-cu12==11.4.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-curand-cu12==10.3.10.19; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusolver-cu12==11.7.5.82; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparse-cu12==12.5.10.65; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nccl-cu12==2.27.5; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvshmem-cu12==3.3.20; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvtx-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvjitlink-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufile-cu12==1.14.1.1; platform_system == 'Linux' and platform_machine == 'x86_64'
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_13t-cuda12_9-test: # Testing
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
needs:
|
||||||
|
- manywheel-py3_13t-cuda12_9-build
|
||||||
|
- get-label-type
|
||||||
|
uses: ./.github/workflows/_binary-test-linux.yml
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.13t"
|
||||||
|
build_name: manywheel-py3_13t-cuda12_9
|
||||||
|
build_environment: linux-binary-manywheel
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
runs_on: linux.g4dn.4xlarge.nvidia.gpu # 12.8+ builds need sm_70+ runner
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_13t-cuda12_9-upload: # Uploading
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
permissions:
|
||||||
|
id-token: write
|
||||||
|
contents: read
|
||||||
|
needs: manywheel-py3_13t-cuda12_9-test
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.13t"
|
||||||
|
build_name: manywheel-py3_13t-cuda12_9
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
manywheel-py3_13t-cuda13_0-build:
|
manywheel-py3_13t-cuda13_0-build:
|
||||||
if: ${{ github.repository_owner == 'pytorch' }}
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
uses: ./.github/workflows/_binary-build-linux.yml
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
@ -3241,6 +3571,72 @@ jobs:
|
|||||||
github-token: ${{ secrets.GITHUB_TOKEN }}
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
uses: ./.github/workflows/_binary-upload.yml
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
|
manywheel-py3_14-cuda12_9-build:
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
|
needs: get-label-type
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.14"
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
build_name: manywheel-py3_14-cuda12_9
|
||||||
|
build_environment: linux-binary-manywheel
|
||||||
|
PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-runtime-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-cupti-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cudnn-cu12==9.10.2.21; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cublas-cu12==12.9.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufft-cu12==11.4.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-curand-cu12==10.3.10.19; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusolver-cu12==11.7.5.82; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparse-cu12==12.5.10.65; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nccl-cu12==2.27.5; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvshmem-cu12==3.3.20; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvtx-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvjitlink-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufile-cu12==1.14.1.1; platform_system == 'Linux' and platform_machine == 'x86_64'
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_14-cuda12_9-test: # Testing
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
needs:
|
||||||
|
- manywheel-py3_14-cuda12_9-build
|
||||||
|
- get-label-type
|
||||||
|
uses: ./.github/workflows/_binary-test-linux.yml
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.14"
|
||||||
|
build_name: manywheel-py3_14-cuda12_9
|
||||||
|
build_environment: linux-binary-manywheel
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
runs_on: linux.g4dn.4xlarge.nvidia.gpu # 12.8+ builds need sm_70+ runner
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_14-cuda12_9-upload: # Uploading
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
permissions:
|
||||||
|
id-token: write
|
||||||
|
contents: read
|
||||||
|
needs: manywheel-py3_14-cuda12_9-test
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.14"
|
||||||
|
build_name: manywheel-py3_14-cuda12_9
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
manywheel-py3_14-cuda13_0-build:
|
manywheel-py3_14-cuda13_0-build:
|
||||||
if: ${{ github.repository_owner == 'pytorch' }}
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
uses: ./.github/workflows/_binary-build-linux.yml
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
@ -3841,6 +4237,72 @@ jobs:
|
|||||||
github-token: ${{ secrets.GITHUB_TOKEN }}
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
uses: ./.github/workflows/_binary-upload.yml
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
|
manywheel-py3_14t-cuda12_9-build:
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
|
needs: get-label-type
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.14t"
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
build_name: manywheel-py3_14t-cuda12_9
|
||||||
|
build_environment: linux-binary-manywheel
|
||||||
|
PYTORCH_EXTRA_INSTALL_REQUIREMENTS: nvidia-cuda-nvrtc-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-runtime-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cuda-cupti-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cudnn-cu12==9.10.2.21; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cublas-cu12==12.9.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufft-cu12==11.4.1.4; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-curand-cu12==10.3.10.19; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusolver-cu12==11.7.5.82; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparse-cu12==12.5.10.65; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nccl-cu12==2.27.5; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvshmem-cu12==3.3.20; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvtx-cu12==12.9.79; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-nvjitlink-cu12==12.9.86; platform_system == 'Linux' and platform_machine == 'x86_64' | nvidia-cufile-cu12==1.14.1.1; platform_system == 'Linux' and platform_machine == 'x86_64'
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_14t-cuda12_9-test: # Testing
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
needs:
|
||||||
|
- manywheel-py3_14t-cuda12_9-build
|
||||||
|
- get-label-type
|
||||||
|
uses: ./.github/workflows/_binary-test-linux.yml
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.14t"
|
||||||
|
build_name: manywheel-py3_14t-cuda12_9
|
||||||
|
build_environment: linux-binary-manywheel
|
||||||
|
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
|
||||||
|
runs_on: linux.g4dn.4xlarge.nvidia.gpu # 12.8+ builds need sm_70+ runner
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
manywheel-py3_14t-cuda12_9-upload: # Uploading
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
permissions:
|
||||||
|
id-token: write
|
||||||
|
contents: read
|
||||||
|
needs: manywheel-py3_14t-cuda12_9-test
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: /pytorch
|
||||||
|
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: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
DOCKER_IMAGE: manylinux2_28-builder
|
||||||
|
DOCKER_IMAGE_TAG_PREFIX: cuda12.9
|
||||||
|
DESIRED_PYTHON: "3.14t"
|
||||||
|
build_name: manywheel-py3_14t-cuda12_9
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
|
||||||
manywheel-py3_14t-cuda13_0-build:
|
manywheel-py3_14t-cuda13_0-build:
|
||||||
if: ${{ github.repository_owner == 'pytorch' }}
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
uses: ./.github/workflows/_binary-build-linux.yml
|
uses: ./.github/workflows/_binary-build-linux.yml
|
||||||
|
|||||||
250
.github/workflows/generated-windows-binary-libtorch-debug-nightly.yml
generated
vendored
250
.github/workflows/generated-windows-binary-libtorch-debug-nightly.yml
generated
vendored
@ -788,6 +788,256 @@ jobs:
|
|||||||
secrets:
|
secrets:
|
||||||
github-token: ${{ secrets.GITHUB_TOKEN }}
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
uses: ./.github/workflows/_binary-upload.yml
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
libtorch-cuda12_9-shared-with-deps-debug-build:
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
needs: get-label-type
|
||||||
|
runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge"
|
||||||
|
timeout-minutes: 360
|
||||||
|
env:
|
||||||
|
PYTORCH_ROOT: ${{ github.workspace }}/pytorch
|
||||||
|
PACKAGE_TYPE: libtorch
|
||||||
|
# TODO: This is a legacy variable that we eventually want to get rid of in
|
||||||
|
# favor of GPU_ARCH_VERSION
|
||||||
|
DESIRED_CUDA: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
SKIP_ALL_TESTS: 1
|
||||||
|
LIBTORCH_CONFIG: debug
|
||||||
|
LIBTORCH_VARIANT: shared-with-deps
|
||||||
|
# This is a dummy value for libtorch to work correctly with our batch scripts
|
||||||
|
# without this value pip does not get installed for some reason
|
||||||
|
DESIRED_PYTHON: "3.10"
|
||||||
|
steps:
|
||||||
|
# NOTE: These environment variables are put here so that they can be applied on every job equally
|
||||||
|
# They are also here because setting them at a workflow level doesn't give us access to the
|
||||||
|
# runner.temp variable, which we need.
|
||||||
|
- name: Populate binary env
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}"
|
||||||
|
echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}"
|
||||||
|
echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}"
|
||||||
|
- name: Display EC2 information
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
set -euo pipefail
|
||||||
|
function get_ec2_metadata() {
|
||||||
|
# Pulled from instance metadata endpoint for EC2
|
||||||
|
# see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html
|
||||||
|
category=$1
|
||||||
|
curl -H "X-aws-ec2-metadata-token: $(curl -s -X PUT "http://169.254.169.254/latest/api/token" -H "X-aws-ec2-metadata-token-ttl-seconds: 30")" -fsSL "http://169.254.169.254/latest/meta-data/${category}"
|
||||||
|
}
|
||||||
|
echo "ami-id: $(get_ec2_metadata ami-id)"
|
||||||
|
echo "instance-id: $(get_ec2_metadata instance-id)"
|
||||||
|
echo "instance-type: $(get_ec2_metadata instance-type)"
|
||||||
|
echo "system info $(uname -a)"
|
||||||
|
- name: "[FB EMPLOYEES] Enable SSH (Click me for login details)"
|
||||||
|
uses: pytorch/test-infra/.github/actions/setup-ssh@main
|
||||||
|
continue-on-error: true
|
||||||
|
with:
|
||||||
|
github-secret: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
- name: Enable git long paths and symlinks on Windows and disable fsmonitor daemon
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
git config --global core.longpaths true
|
||||||
|
git config --global core.symlinks true
|
||||||
|
|
||||||
|
# https://git-scm.com/docs/git-fsmonitor--daemon. The daemon could lock
|
||||||
|
# the directory on Windows and prevent GHA from checking out as reported
|
||||||
|
# in https://github.com/actions/checkout/issues/1018
|
||||||
|
git config --global core.fsmonitor false
|
||||||
|
# Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560
|
||||||
|
- name: Enable long paths on Windows
|
||||||
|
shell: powershell
|
||||||
|
run: |
|
||||||
|
Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1
|
||||||
|
# Since it's just a defensive command, the workflow should continue even the command fails. This step can be
|
||||||
|
# removed once Windows Defender is removed from the AMI
|
||||||
|
- name: Disables Windows Defender scheduled and real-time scanning for files in directories used by PyTorch
|
||||||
|
continue-on-error: true
|
||||||
|
shell: powershell
|
||||||
|
run: |
|
||||||
|
Add-MpPreference -ExclusionPath $(Get-Location).tostring(),$Env:TEMP -ErrorAction Ignore
|
||||||
|
# Let's both exclude the path and disable Windows Defender completely just to be sure
|
||||||
|
# that it doesn't interfere
|
||||||
|
Set-MpPreference -DisableRealtimeMonitoring $True -ErrorAction Ignore
|
||||||
|
- name: Checkout PyTorch
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
|
||||||
|
submodules: recursive
|
||||||
|
path: pytorch
|
||||||
|
show-progress: false
|
||||||
|
- name: Clean PyTorch checkout
|
||||||
|
run: |
|
||||||
|
# Remove any artifacts from the previous checkouts
|
||||||
|
git clean -fxd
|
||||||
|
working-directory: pytorch
|
||||||
|
- name: Populate binary env
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
"${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh"
|
||||||
|
- name: Build PyTorch binary
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
"${PYTORCH_ROOT}/.circleci/scripts/binary_windows_build.sh"
|
||||||
|
- uses: actions/upload-artifact@v4.4.0
|
||||||
|
if: always()
|
||||||
|
with:
|
||||||
|
name: libtorch-cuda12_9-shared-with-deps-debug
|
||||||
|
retention-days: 14
|
||||||
|
if-no-files-found: error
|
||||||
|
path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}"
|
||||||
|
- name: Wait until all sessions have drained
|
||||||
|
shell: powershell
|
||||||
|
working-directory: pytorch
|
||||||
|
if: always()
|
||||||
|
timeout-minutes: 120
|
||||||
|
run: |
|
||||||
|
.github\scripts\wait_for_ssh_to_drain.ps1
|
||||||
|
- name: Kill active ssh sessions if still around (Useful if workflow was cancelled)
|
||||||
|
shell: powershell
|
||||||
|
working-directory: pytorch
|
||||||
|
if: always()
|
||||||
|
run: |
|
||||||
|
.github\scripts\kill_active_ssh_sessions.ps1
|
||||||
|
|
||||||
|
libtorch-cuda12_9-shared-with-deps-debug-test: # Testing
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
needs:
|
||||||
|
- libtorch-cuda12_9-shared-with-deps-debug-build
|
||||||
|
- get-label-type
|
||||||
|
runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.g4dn.xlarge"
|
||||||
|
timeout-minutes: 360
|
||||||
|
env:
|
||||||
|
PYTORCH_ROOT: ${{ github.workspace }}/pytorch
|
||||||
|
PACKAGE_TYPE: libtorch
|
||||||
|
# TODO: This is a legacy variable that we eventually want to get rid of in
|
||||||
|
# favor of GPU_ARCH_VERSION
|
||||||
|
DESIRED_CUDA: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
SKIP_ALL_TESTS: 1
|
||||||
|
LIBTORCH_CONFIG: debug
|
||||||
|
LIBTORCH_VARIANT: shared-with-deps
|
||||||
|
# This is a dummy value for libtorch to work correctly with our batch scripts
|
||||||
|
# without this value pip does not get installed for some reason
|
||||||
|
DESIRED_PYTHON: "3.10"
|
||||||
|
steps:
|
||||||
|
- name: Display EC2 information
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
set -euo pipefail
|
||||||
|
function get_ec2_metadata() {
|
||||||
|
# Pulled from instance metadata endpoint for EC2
|
||||||
|
# see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html
|
||||||
|
category=$1
|
||||||
|
curl -H "X-aws-ec2-metadata-token: $(curl -s -X PUT "http://169.254.169.254/latest/api/token" -H "X-aws-ec2-metadata-token-ttl-seconds: 30")" -fsSL "http://169.254.169.254/latest/meta-data/${category}"
|
||||||
|
}
|
||||||
|
echo "ami-id: $(get_ec2_metadata ami-id)"
|
||||||
|
echo "instance-id: $(get_ec2_metadata instance-id)"
|
||||||
|
echo "instance-type: $(get_ec2_metadata instance-type)"
|
||||||
|
echo "system info $(uname -a)"
|
||||||
|
- name: "[FB EMPLOYEES] Enable SSH (Click me for login details)"
|
||||||
|
uses: pytorch/test-infra/.github/actions/setup-ssh@main
|
||||||
|
continue-on-error: true
|
||||||
|
with:
|
||||||
|
github-secret: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
- name: Enable git long paths and symlinks on Windows and disable fsmonitor daemon
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
git config --global core.longpaths true
|
||||||
|
git config --global core.symlinks true
|
||||||
|
|
||||||
|
# https://git-scm.com/docs/git-fsmonitor--daemon. The daemon could lock
|
||||||
|
# the directory on Windows and prevent GHA from checking out as reported
|
||||||
|
# in https://github.com/actions/checkout/issues/1018
|
||||||
|
git config --global core.fsmonitor false
|
||||||
|
# Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560
|
||||||
|
- name: Enable long paths on Windows
|
||||||
|
shell: powershell
|
||||||
|
run: |
|
||||||
|
Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1
|
||||||
|
# Since it's just a defensive command, the workflow should continue even the command fails. This step can be
|
||||||
|
# removed once Windows Defender is removed from the AMI
|
||||||
|
- name: Disables Windows Defender scheduled and real-time scanning for files in directories used by PyTorch
|
||||||
|
continue-on-error: true
|
||||||
|
shell: powershell
|
||||||
|
run: |
|
||||||
|
Add-MpPreference -ExclusionPath $(Get-Location).tostring(),$Env:TEMP -ErrorAction Ignore
|
||||||
|
# Let's both exclude the path and disable Windows Defender completely just to be sure
|
||||||
|
# that it doesn't interfere
|
||||||
|
Set-MpPreference -DisableRealtimeMonitoring $True -ErrorAction Ignore
|
||||||
|
- name: Checkout PyTorch
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
|
||||||
|
submodules: recursive
|
||||||
|
path: pytorch
|
||||||
|
show-progress: false
|
||||||
|
- name: Clean PyTorch checkout
|
||||||
|
run: |
|
||||||
|
# Remove any artifacts from the previous checkouts
|
||||||
|
git clean -fxd
|
||||||
|
working-directory: pytorch
|
||||||
|
# NOTE: These environment variables are put here so that they can be applied on every job equally
|
||||||
|
# They are also here because setting them at a workflow level doesn't give us access to the
|
||||||
|
# runner.temp variable, which we need.
|
||||||
|
- name: Populate binary env
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}"
|
||||||
|
echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}"
|
||||||
|
echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}"
|
||||||
|
- uses: actions/download-artifact@v4.1.7
|
||||||
|
name: Download Build Artifacts
|
||||||
|
with:
|
||||||
|
name: libtorch-cuda12_9-shared-with-deps-debug
|
||||||
|
path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}"
|
||||||
|
- name: Populate binary env
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
"${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh"
|
||||||
|
- name: Test PyTorch binary
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
"${PYTORCH_ROOT}/.circleci/scripts/binary_windows_test.sh"
|
||||||
|
- name: Wait until all sessions have drained
|
||||||
|
shell: powershell
|
||||||
|
working-directory: pytorch
|
||||||
|
if: always()
|
||||||
|
timeout-minutes: 120
|
||||||
|
run: |
|
||||||
|
.github\scripts\wait_for_ssh_to_drain.ps1
|
||||||
|
- name: Kill active ssh sessions if still around (Useful if workflow was cancelled)
|
||||||
|
shell: powershell
|
||||||
|
working-directory: pytorch
|
||||||
|
if: always()
|
||||||
|
run: |
|
||||||
|
.github\scripts\kill_active_ssh_sessions.ps1
|
||||||
|
libtorch-cuda12_9-shared-with-deps-debug-upload: # Uploading
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
permissions:
|
||||||
|
id-token: write
|
||||||
|
contents: read
|
||||||
|
needs: libtorch-cuda12_9-shared-with-deps-debug-test
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: ${{ github.workspace }}/pytorch
|
||||||
|
PACKAGE_TYPE: libtorch
|
||||||
|
# TODO: This is a legacy variable that we eventually want to get rid of in
|
||||||
|
# favor of GPU_ARCH_VERSION
|
||||||
|
DESIRED_CUDA: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
LIBTORCH_CONFIG: debug
|
||||||
|
LIBTORCH_VARIANT: shared-with-deps
|
||||||
|
# This is a dummy value for libtorch to work correctly with our batch scripts
|
||||||
|
# without this value pip does not get installed for some reason
|
||||||
|
DESIRED_PYTHON: "3.10"
|
||||||
|
build_name: libtorch-cuda12_9-shared-with-deps-debug
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
libtorch-cuda13_0-shared-with-deps-debug-build:
|
libtorch-cuda13_0-shared-with-deps-debug-build:
|
||||||
if: ${{ github.repository_owner == 'pytorch' }}
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
needs: get-label-type
|
needs: get-label-type
|
||||||
|
|||||||
250
.github/workflows/generated-windows-binary-libtorch-release-nightly.yml
generated
vendored
250
.github/workflows/generated-windows-binary-libtorch-release-nightly.yml
generated
vendored
@ -788,6 +788,256 @@ jobs:
|
|||||||
secrets:
|
secrets:
|
||||||
github-token: ${{ secrets.GITHUB_TOKEN }}
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
uses: ./.github/workflows/_binary-upload.yml
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
|
libtorch-cuda12_9-shared-with-deps-release-build:
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
needs: get-label-type
|
||||||
|
runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge"
|
||||||
|
timeout-minutes: 360
|
||||||
|
env:
|
||||||
|
PYTORCH_ROOT: ${{ github.workspace }}/pytorch
|
||||||
|
PACKAGE_TYPE: libtorch
|
||||||
|
# TODO: This is a legacy variable that we eventually want to get rid of in
|
||||||
|
# favor of GPU_ARCH_VERSION
|
||||||
|
DESIRED_CUDA: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
SKIP_ALL_TESTS: 1
|
||||||
|
LIBTORCH_CONFIG: release
|
||||||
|
LIBTORCH_VARIANT: shared-with-deps
|
||||||
|
# This is a dummy value for libtorch to work correctly with our batch scripts
|
||||||
|
# without this value pip does not get installed for some reason
|
||||||
|
DESIRED_PYTHON: "3.10"
|
||||||
|
steps:
|
||||||
|
# NOTE: These environment variables are put here so that they can be applied on every job equally
|
||||||
|
# They are also here because setting them at a workflow level doesn't give us access to the
|
||||||
|
# runner.temp variable, which we need.
|
||||||
|
- name: Populate binary env
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}"
|
||||||
|
echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}"
|
||||||
|
echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}"
|
||||||
|
- name: Display EC2 information
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
set -euo pipefail
|
||||||
|
function get_ec2_metadata() {
|
||||||
|
# Pulled from instance metadata endpoint for EC2
|
||||||
|
# see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html
|
||||||
|
category=$1
|
||||||
|
curl -H "X-aws-ec2-metadata-token: $(curl -s -X PUT "http://169.254.169.254/latest/api/token" -H "X-aws-ec2-metadata-token-ttl-seconds: 30")" -fsSL "http://169.254.169.254/latest/meta-data/${category}"
|
||||||
|
}
|
||||||
|
echo "ami-id: $(get_ec2_metadata ami-id)"
|
||||||
|
echo "instance-id: $(get_ec2_metadata instance-id)"
|
||||||
|
echo "instance-type: $(get_ec2_metadata instance-type)"
|
||||||
|
echo "system info $(uname -a)"
|
||||||
|
- name: "[FB EMPLOYEES] Enable SSH (Click me for login details)"
|
||||||
|
uses: pytorch/test-infra/.github/actions/setup-ssh@main
|
||||||
|
continue-on-error: true
|
||||||
|
with:
|
||||||
|
github-secret: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
- name: Enable git long paths and symlinks on Windows and disable fsmonitor daemon
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
git config --global core.longpaths true
|
||||||
|
git config --global core.symlinks true
|
||||||
|
|
||||||
|
# https://git-scm.com/docs/git-fsmonitor--daemon. The daemon could lock
|
||||||
|
# the directory on Windows and prevent GHA from checking out as reported
|
||||||
|
# in https://github.com/actions/checkout/issues/1018
|
||||||
|
git config --global core.fsmonitor false
|
||||||
|
# Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560
|
||||||
|
- name: Enable long paths on Windows
|
||||||
|
shell: powershell
|
||||||
|
run: |
|
||||||
|
Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1
|
||||||
|
# Since it's just a defensive command, the workflow should continue even the command fails. This step can be
|
||||||
|
# removed once Windows Defender is removed from the AMI
|
||||||
|
- name: Disables Windows Defender scheduled and real-time scanning for files in directories used by PyTorch
|
||||||
|
continue-on-error: true
|
||||||
|
shell: powershell
|
||||||
|
run: |
|
||||||
|
Add-MpPreference -ExclusionPath $(Get-Location).tostring(),$Env:TEMP -ErrorAction Ignore
|
||||||
|
# Let's both exclude the path and disable Windows Defender completely just to be sure
|
||||||
|
# that it doesn't interfere
|
||||||
|
Set-MpPreference -DisableRealtimeMonitoring $True -ErrorAction Ignore
|
||||||
|
- name: Checkout PyTorch
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
|
||||||
|
submodules: recursive
|
||||||
|
path: pytorch
|
||||||
|
show-progress: false
|
||||||
|
- name: Clean PyTorch checkout
|
||||||
|
run: |
|
||||||
|
# Remove any artifacts from the previous checkouts
|
||||||
|
git clean -fxd
|
||||||
|
working-directory: pytorch
|
||||||
|
- name: Populate binary env
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
"${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh"
|
||||||
|
- name: Build PyTorch binary
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
"${PYTORCH_ROOT}/.circleci/scripts/binary_windows_build.sh"
|
||||||
|
- uses: actions/upload-artifact@v4.4.0
|
||||||
|
if: always()
|
||||||
|
with:
|
||||||
|
name: libtorch-cuda12_9-shared-with-deps-release
|
||||||
|
retention-days: 14
|
||||||
|
if-no-files-found: error
|
||||||
|
path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}"
|
||||||
|
- name: Wait until all sessions have drained
|
||||||
|
shell: powershell
|
||||||
|
working-directory: pytorch
|
||||||
|
if: always()
|
||||||
|
timeout-minutes: 120
|
||||||
|
run: |
|
||||||
|
.github\scripts\wait_for_ssh_to_drain.ps1
|
||||||
|
- name: Kill active ssh sessions if still around (Useful if workflow was cancelled)
|
||||||
|
shell: powershell
|
||||||
|
working-directory: pytorch
|
||||||
|
if: always()
|
||||||
|
run: |
|
||||||
|
.github\scripts\kill_active_ssh_sessions.ps1
|
||||||
|
|
||||||
|
libtorch-cuda12_9-shared-with-deps-release-test: # Testing
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
needs:
|
||||||
|
- libtorch-cuda12_9-shared-with-deps-release-build
|
||||||
|
- get-label-type
|
||||||
|
runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.g4dn.xlarge"
|
||||||
|
timeout-minutes: 360
|
||||||
|
env:
|
||||||
|
PYTORCH_ROOT: ${{ github.workspace }}/pytorch
|
||||||
|
PACKAGE_TYPE: libtorch
|
||||||
|
# TODO: This is a legacy variable that we eventually want to get rid of in
|
||||||
|
# favor of GPU_ARCH_VERSION
|
||||||
|
DESIRED_CUDA: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
SKIP_ALL_TESTS: 1
|
||||||
|
LIBTORCH_CONFIG: release
|
||||||
|
LIBTORCH_VARIANT: shared-with-deps
|
||||||
|
# This is a dummy value for libtorch to work correctly with our batch scripts
|
||||||
|
# without this value pip does not get installed for some reason
|
||||||
|
DESIRED_PYTHON: "3.10"
|
||||||
|
steps:
|
||||||
|
- name: Display EC2 information
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
set -euo pipefail
|
||||||
|
function get_ec2_metadata() {
|
||||||
|
# Pulled from instance metadata endpoint for EC2
|
||||||
|
# see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html
|
||||||
|
category=$1
|
||||||
|
curl -H "X-aws-ec2-metadata-token: $(curl -s -X PUT "http://169.254.169.254/latest/api/token" -H "X-aws-ec2-metadata-token-ttl-seconds: 30")" -fsSL "http://169.254.169.254/latest/meta-data/${category}"
|
||||||
|
}
|
||||||
|
echo "ami-id: $(get_ec2_metadata ami-id)"
|
||||||
|
echo "instance-id: $(get_ec2_metadata instance-id)"
|
||||||
|
echo "instance-type: $(get_ec2_metadata instance-type)"
|
||||||
|
echo "system info $(uname -a)"
|
||||||
|
- name: "[FB EMPLOYEES] Enable SSH (Click me for login details)"
|
||||||
|
uses: pytorch/test-infra/.github/actions/setup-ssh@main
|
||||||
|
continue-on-error: true
|
||||||
|
with:
|
||||||
|
github-secret: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
- name: Enable git long paths and symlinks on Windows and disable fsmonitor daemon
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
git config --global core.longpaths true
|
||||||
|
git config --global core.symlinks true
|
||||||
|
|
||||||
|
# https://git-scm.com/docs/git-fsmonitor--daemon. The daemon could lock
|
||||||
|
# the directory on Windows and prevent GHA from checking out as reported
|
||||||
|
# in https://github.com/actions/checkout/issues/1018
|
||||||
|
git config --global core.fsmonitor false
|
||||||
|
# Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560
|
||||||
|
- name: Enable long paths on Windows
|
||||||
|
shell: powershell
|
||||||
|
run: |
|
||||||
|
Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1
|
||||||
|
# Since it's just a defensive command, the workflow should continue even the command fails. This step can be
|
||||||
|
# removed once Windows Defender is removed from the AMI
|
||||||
|
- name: Disables Windows Defender scheduled and real-time scanning for files in directories used by PyTorch
|
||||||
|
continue-on-error: true
|
||||||
|
shell: powershell
|
||||||
|
run: |
|
||||||
|
Add-MpPreference -ExclusionPath $(Get-Location).tostring(),$Env:TEMP -ErrorAction Ignore
|
||||||
|
# Let's both exclude the path and disable Windows Defender completely just to be sure
|
||||||
|
# that it doesn't interfere
|
||||||
|
Set-MpPreference -DisableRealtimeMonitoring $True -ErrorAction Ignore
|
||||||
|
- name: Checkout PyTorch
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
|
||||||
|
submodules: recursive
|
||||||
|
path: pytorch
|
||||||
|
show-progress: false
|
||||||
|
- name: Clean PyTorch checkout
|
||||||
|
run: |
|
||||||
|
# Remove any artifacts from the previous checkouts
|
||||||
|
git clean -fxd
|
||||||
|
working-directory: pytorch
|
||||||
|
# NOTE: These environment variables are put here so that they can be applied on every job equally
|
||||||
|
# They are also here because setting them at a workflow level doesn't give us access to the
|
||||||
|
# runner.temp variable, which we need.
|
||||||
|
- name: Populate binary env
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}"
|
||||||
|
echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}"
|
||||||
|
echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}"
|
||||||
|
- uses: actions/download-artifact@v4.1.7
|
||||||
|
name: Download Build Artifacts
|
||||||
|
with:
|
||||||
|
name: libtorch-cuda12_9-shared-with-deps-release
|
||||||
|
path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}"
|
||||||
|
- name: Populate binary env
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
"${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh"
|
||||||
|
- name: Test PyTorch binary
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
"${PYTORCH_ROOT}/.circleci/scripts/binary_windows_test.sh"
|
||||||
|
- name: Wait until all sessions have drained
|
||||||
|
shell: powershell
|
||||||
|
working-directory: pytorch
|
||||||
|
if: always()
|
||||||
|
timeout-minutes: 120
|
||||||
|
run: |
|
||||||
|
.github\scripts\wait_for_ssh_to_drain.ps1
|
||||||
|
- name: Kill active ssh sessions if still around (Useful if workflow was cancelled)
|
||||||
|
shell: powershell
|
||||||
|
working-directory: pytorch
|
||||||
|
if: always()
|
||||||
|
run: |
|
||||||
|
.github\scripts\kill_active_ssh_sessions.ps1
|
||||||
|
libtorch-cuda12_9-shared-with-deps-release-upload: # Uploading
|
||||||
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
|
permissions:
|
||||||
|
id-token: write
|
||||||
|
contents: read
|
||||||
|
needs: libtorch-cuda12_9-shared-with-deps-release-test
|
||||||
|
with:
|
||||||
|
PYTORCH_ROOT: ${{ github.workspace }}/pytorch
|
||||||
|
PACKAGE_TYPE: libtorch
|
||||||
|
# TODO: This is a legacy variable that we eventually want to get rid of in
|
||||||
|
# favor of GPU_ARCH_VERSION
|
||||||
|
DESIRED_CUDA: cu129
|
||||||
|
GPU_ARCH_VERSION: "12.9"
|
||||||
|
GPU_ARCH_TYPE: cuda
|
||||||
|
LIBTORCH_CONFIG: release
|
||||||
|
LIBTORCH_VARIANT: shared-with-deps
|
||||||
|
# This is a dummy value for libtorch to work correctly with our batch scripts
|
||||||
|
# without this value pip does not get installed for some reason
|
||||||
|
DESIRED_PYTHON: "3.10"
|
||||||
|
build_name: libtorch-cuda12_9-shared-with-deps-release
|
||||||
|
secrets:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
uses: ./.github/workflows/_binary-upload.yml
|
||||||
libtorch-cuda13_0-shared-with-deps-release-build:
|
libtorch-cuda13_0-shared-with-deps-release-build:
|
||||||
if: ${{ github.repository_owner == 'pytorch' }}
|
if: ${{ github.repository_owner == 'pytorch' }}
|
||||||
needs: get-label-type
|
needs: get-label-type
|
||||||
|
|||||||
1666
.github/workflows/generated-windows-binary-wheel-nightly.yml
generated
vendored
1666
.github/workflows/generated-windows-binary-wheel-nightly.yml
generated
vendored
File diff suppressed because it is too large
Load Diff
Reference in New Issue
Block a user