diff --git a/.github/templates/windows_binary_build_workflow.yml.j2 b/.github/templates/windows_binary_build_workflow.yml.j2 index d5aca578b90..06646f449fc 100644 --- a/.github/templates/windows_binary_build_workflow.yml.j2 +++ b/.github/templates/windows_binary_build_workflow.yml.j2 @@ -53,10 +53,20 @@ env: !{{ common.concurrency(build_environment) }} jobs: + get-label-type: + name: get-label-type + uses: ./.github/workflows/_runner-determinator.yml + with: + triggering_actor: ${{ github.triggering_actor }} + issue_owner: ${{ github.event.pull_request.user.login || github.event.issue.user.login }} + curr_branch: ${{ github.head_ref || github.ref_name }} + curr_ref_type: ${{ github.ref_type }} + {%- for config in build_configs %} !{{ config["build_name"] }}-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: !{{ common.timeout_minutes }} !{{ upload.binary_env(config, True) }} {%- if config.pytorch_extra_install_requirements is defined and config.pytorch_extra_install_requirements|d('')|length > 0 %} @@ -85,15 +95,17 @@ jobs: !{{ common.wait_and_kill_ssh_windows('pytorch') }} !{{ config["build_name"] }}-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: !{{ config["build_name"] }}-build + needs: + - !{{ config["build_name"] }}-build + - get-label-type {%- if config["gpu_arch_type"] == "cuda" %} {%- if branches == "nightly" %} - runs-on: windows.8xlarge.nvidia.gpu + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" {%- else %} - runs-on: windows.8xlarge.nvidia.gpu.nonephemeral + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu.nonephemeral" {%- endif %} {%- else %} - runs-on: windows.4xlarge.nonephemeral + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" {%- endif %} timeout-minutes: !{{ common.timeout_minutes }} !{{ upload.binary_env(config, True) }} diff --git a/.github/workflows/generated-windows-binary-conda-nightly.yml b/.github/workflows/generated-windows-binary-conda-nightly.yml index da0dd504c75..dac8a32e14a 100644 --- a/.github/workflows/generated-windows-binary-conda-nightly.yml +++ b/.github/workflows/generated-windows-binary-conda-nightly.yml @@ -32,9 +32,18 @@ concurrency: cancel-in-progress: true jobs: + get-label-type: + name: get-label-type + uses: ./.github/workflows/_runner-determinator.yml + with: + triggering_actor: ${{ github.triggering_actor }} + issue_owner: ${{ github.event.pull_request.user.login || github.event.issue.user.login }} + curr_branch: ${{ github.head_ref || github.ref_name }} + curr_ref_type: ${{ github.ref_type }} conda-py3_9-cpu-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -145,8 +154,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 conda-py3_9-cpu-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: conda-py3_9-cpu-build - runs-on: windows.4xlarge.nonephemeral + needs: + - conda-py3_9-cpu-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -276,7 +287,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml conda-py3_9-cuda11_8-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -388,8 +400,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 conda-py3_9-cuda11_8-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: conda-py3_9-cuda11_8-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - conda-py3_9-cuda11_8-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -521,7 +535,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml conda-py3_9-cuda12_1-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -633,8 +648,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 conda-py3_9-cuda12_1-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: conda-py3_9-cuda12_1-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - conda-py3_9-cuda12_1-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -766,7 +783,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml conda-py3_9-cuda12_4-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -878,8 +896,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 conda-py3_9-cuda12_4-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: conda-py3_9-cuda12_4-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - conda-py3_9-cuda12_4-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -1011,7 +1031,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml conda-py3_10-cpu-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -1122,8 +1143,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 conda-py3_10-cpu-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: conda-py3_10-cpu-build - runs-on: windows.4xlarge.nonephemeral + needs: + - conda-py3_10-cpu-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -1253,7 +1276,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml conda-py3_10-cuda11_8-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -1365,8 +1389,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 conda-py3_10-cuda11_8-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: conda-py3_10-cuda11_8-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - conda-py3_10-cuda11_8-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -1498,7 +1524,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml conda-py3_10-cuda12_1-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -1610,8 +1637,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 conda-py3_10-cuda12_1-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: conda-py3_10-cuda12_1-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - conda-py3_10-cuda12_1-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -1743,7 +1772,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml conda-py3_10-cuda12_4-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -1855,8 +1885,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 conda-py3_10-cuda12_4-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: conda-py3_10-cuda12_4-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - conda-py3_10-cuda12_4-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -1988,7 +2020,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml conda-py3_11-cpu-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -2099,8 +2132,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 conda-py3_11-cpu-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: conda-py3_11-cpu-build - runs-on: windows.4xlarge.nonephemeral + needs: + - conda-py3_11-cpu-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -2230,7 +2265,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml conda-py3_11-cuda11_8-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -2342,8 +2378,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 conda-py3_11-cuda11_8-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: conda-py3_11-cuda11_8-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - conda-py3_11-cuda11_8-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -2475,7 +2513,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml conda-py3_11-cuda12_1-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -2587,8 +2626,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 conda-py3_11-cuda12_1-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: conda-py3_11-cuda12_1-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - conda-py3_11-cuda12_1-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -2720,7 +2761,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml conda-py3_11-cuda12_4-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -2832,8 +2874,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 conda-py3_11-cuda12_4-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: conda-py3_11-cuda12_4-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - conda-py3_11-cuda12_4-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -2965,7 +3009,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml conda-py3_12-cpu-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -3076,8 +3121,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 conda-py3_12-cpu-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: conda-py3_12-cpu-build - runs-on: windows.4xlarge.nonephemeral + needs: + - conda-py3_12-cpu-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -3207,7 +3254,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml conda-py3_12-cuda11_8-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -3319,8 +3367,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 conda-py3_12-cuda11_8-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: conda-py3_12-cuda11_8-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - conda-py3_12-cuda11_8-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -3452,7 +3502,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml conda-py3_12-cuda12_1-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -3564,8 +3615,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 conda-py3_12-cuda12_1-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: conda-py3_12-cuda12_1-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - conda-py3_12-cuda12_1-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -3697,7 +3750,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml conda-py3_12-cuda12_4-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -3809,8 +3863,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 conda-py3_12-cuda12_4-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: conda-py3_12-cuda12_4-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - conda-py3_12-cuda12_4-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch diff --git a/.github/workflows/generated-windows-binary-libtorch-debug-main.yml b/.github/workflows/generated-windows-binary-libtorch-debug-main.yml index 8ac413be0d6..6debd8b9a2d 100644 --- a/.github/workflows/generated-windows-binary-libtorch-debug-main.yml +++ b/.github/workflows/generated-windows-binary-libtorch-debug-main.yml @@ -25,9 +25,18 @@ concurrency: cancel-in-progress: true jobs: + get-label-type: + name: get-label-type + uses: ./.github/workflows/_runner-determinator.yml + with: + triggering_actor: ${{ github.triggering_actor }} + issue_owner: ${{ github.event.pull_request.user.login || github.event.issue.user.login }} + curr_branch: ${{ github.head_ref || github.ref_name }} + curr_ref_type: ${{ github.ref_type }} libtorch-cpu-shared-with-deps-debug-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -142,8 +151,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 libtorch-cpu-shared-with-deps-debug-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: libtorch-cpu-shared-with-deps-debug-build - runs-on: windows.4xlarge.nonephemeral + needs: + - libtorch-cpu-shared-with-deps-debug-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch diff --git a/.github/workflows/generated-windows-binary-libtorch-debug-nightly.yml b/.github/workflows/generated-windows-binary-libtorch-debug-nightly.yml index 60ba5955692..8f1576d3956 100644 --- a/.github/workflows/generated-windows-binary-libtorch-debug-nightly.yml +++ b/.github/workflows/generated-windows-binary-libtorch-debug-nightly.yml @@ -32,9 +32,18 @@ concurrency: cancel-in-progress: true jobs: + get-label-type: + name: get-label-type + uses: ./.github/workflows/_runner-determinator.yml + with: + triggering_actor: ${{ github.triggering_actor }} + issue_owner: ${{ github.event.pull_request.user.login || github.event.issue.user.login }} + curr_branch: ${{ github.head_ref || github.ref_name }} + curr_ref_type: ${{ github.ref_type }} libtorch-cpu-shared-with-deps-debug-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -149,8 +158,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 libtorch-cpu-shared-with-deps-debug-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: libtorch-cpu-shared-with-deps-debug-build - runs-on: windows.4xlarge.nonephemeral + needs: + - libtorch-cpu-shared-with-deps-debug-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -288,7 +299,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml libtorch-cuda11_8-shared-with-deps-debug-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -404,8 +416,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 libtorch-cuda11_8-shared-with-deps-debug-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: libtorch-cuda11_8-shared-with-deps-debug-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - libtorch-cuda11_8-shared-with-deps-debug-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -545,7 +559,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml libtorch-cuda12_1-shared-with-deps-debug-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -661,8 +676,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 libtorch-cuda12_1-shared-with-deps-debug-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: libtorch-cuda12_1-shared-with-deps-debug-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - libtorch-cuda12_1-shared-with-deps-debug-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -802,7 +819,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml libtorch-cuda12_4-shared-with-deps-debug-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -918,8 +936,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 libtorch-cuda12_4-shared-with-deps-debug-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: libtorch-cuda12_4-shared-with-deps-debug-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - libtorch-cuda12_4-shared-with-deps-debug-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch diff --git a/.github/workflows/generated-windows-binary-libtorch-release-main.yml b/.github/workflows/generated-windows-binary-libtorch-release-main.yml index ab00cdc8919..b4ae0bc0130 100644 --- a/.github/workflows/generated-windows-binary-libtorch-release-main.yml +++ b/.github/workflows/generated-windows-binary-libtorch-release-main.yml @@ -25,9 +25,18 @@ concurrency: cancel-in-progress: true jobs: + get-label-type: + name: get-label-type + uses: ./.github/workflows/_runner-determinator.yml + with: + triggering_actor: ${{ github.triggering_actor }} + issue_owner: ${{ github.event.pull_request.user.login || github.event.issue.user.login }} + curr_branch: ${{ github.head_ref || github.ref_name }} + curr_ref_type: ${{ github.ref_type }} libtorch-cpu-shared-with-deps-release-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -142,8 +151,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 libtorch-cpu-shared-with-deps-release-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: libtorch-cpu-shared-with-deps-release-build - runs-on: windows.4xlarge.nonephemeral + needs: + - libtorch-cpu-shared-with-deps-release-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch diff --git a/.github/workflows/generated-windows-binary-libtorch-release-nightly.yml b/.github/workflows/generated-windows-binary-libtorch-release-nightly.yml index 842de97a1fb..c2d1b7009ff 100644 --- a/.github/workflows/generated-windows-binary-libtorch-release-nightly.yml +++ b/.github/workflows/generated-windows-binary-libtorch-release-nightly.yml @@ -32,9 +32,18 @@ concurrency: cancel-in-progress: true jobs: + get-label-type: + name: get-label-type + uses: ./.github/workflows/_runner-determinator.yml + with: + triggering_actor: ${{ github.triggering_actor }} + issue_owner: ${{ github.event.pull_request.user.login || github.event.issue.user.login }} + curr_branch: ${{ github.head_ref || github.ref_name }} + curr_ref_type: ${{ github.ref_type }} libtorch-cpu-shared-with-deps-release-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -149,8 +158,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 libtorch-cpu-shared-with-deps-release-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: libtorch-cpu-shared-with-deps-release-build - runs-on: windows.4xlarge.nonephemeral + needs: + - libtorch-cpu-shared-with-deps-release-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -288,7 +299,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml libtorch-cuda11_8-shared-with-deps-release-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -404,8 +416,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 libtorch-cuda11_8-shared-with-deps-release-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: libtorch-cuda11_8-shared-with-deps-release-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - libtorch-cuda11_8-shared-with-deps-release-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -545,7 +559,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml libtorch-cuda12_1-shared-with-deps-release-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -661,8 +676,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 libtorch-cuda12_1-shared-with-deps-release-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: libtorch-cuda12_1-shared-with-deps-release-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - libtorch-cuda12_1-shared-with-deps-release-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -802,7 +819,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml libtorch-cuda12_4-shared-with-deps-release-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -918,8 +936,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 libtorch-cuda12_4-shared-with-deps-release-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: libtorch-cuda12_4-shared-with-deps-release-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - libtorch-cuda12_4-shared-with-deps-release-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch diff --git a/.github/workflows/generated-windows-binary-wheel-nightly.yml b/.github/workflows/generated-windows-binary-wheel-nightly.yml index 98bdf0c6776..7088e10b5c1 100644 --- a/.github/workflows/generated-windows-binary-wheel-nightly.yml +++ b/.github/workflows/generated-windows-binary-wheel-nightly.yml @@ -32,9 +32,18 @@ concurrency: cancel-in-progress: true jobs: + get-label-type: + name: get-label-type + uses: ./.github/workflows/_runner-determinator.yml + with: + triggering_actor: ${{ github.triggering_actor }} + issue_owner: ${{ github.event.pull_request.user.login || github.event.issue.user.login }} + curr_branch: ${{ github.head_ref || github.ref_name }} + curr_ref_type: ${{ github.ref_type }} wheel-py3_9-cpu-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -146,8 +155,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 wheel-py3_9-cpu-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: wheel-py3_9-cpu-build - runs-on: windows.4xlarge.nonephemeral + needs: + - wheel-py3_9-cpu-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -277,7 +288,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml wheel-py3_9-cuda11_8-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -390,8 +402,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 wheel-py3_9-cuda11_8-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: wheel-py3_9-cuda11_8-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - wheel-py3_9-cuda11_8-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -523,7 +537,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml wheel-py3_9-cuda12_1-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -636,8 +651,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 wheel-py3_9-cuda12_1-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: wheel-py3_9-cuda12_1-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - wheel-py3_9-cuda12_1-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -769,7 +786,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml wheel-py3_9-cuda12_4-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -882,8 +900,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 wheel-py3_9-cuda12_4-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: wheel-py3_9-cuda12_4-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - wheel-py3_9-cuda12_4-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -1015,7 +1035,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml wheel-py3_10-cpu-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -1127,8 +1148,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 wheel-py3_10-cpu-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: wheel-py3_10-cpu-build - runs-on: windows.4xlarge.nonephemeral + needs: + - wheel-py3_10-cpu-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -1258,7 +1281,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml wheel-py3_10-cuda11_8-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -1371,8 +1395,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 wheel-py3_10-cuda11_8-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: wheel-py3_10-cuda11_8-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - wheel-py3_10-cuda11_8-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -1504,7 +1530,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml wheel-py3_10-cuda12_1-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -1617,8 +1644,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 wheel-py3_10-cuda12_1-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: wheel-py3_10-cuda12_1-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - wheel-py3_10-cuda12_1-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -1750,7 +1779,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml wheel-py3_10-cuda12_4-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -1863,8 +1893,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 wheel-py3_10-cuda12_4-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: wheel-py3_10-cuda12_4-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - wheel-py3_10-cuda12_4-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -1996,7 +2028,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml wheel-py3_11-cpu-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -2108,8 +2141,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 wheel-py3_11-cpu-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: wheel-py3_11-cpu-build - runs-on: windows.4xlarge.nonephemeral + needs: + - wheel-py3_11-cpu-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -2239,7 +2274,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml wheel-py3_11-cuda11_8-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -2352,8 +2388,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 wheel-py3_11-cuda11_8-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: wheel-py3_11-cuda11_8-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - wheel-py3_11-cuda11_8-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -2485,7 +2523,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml wheel-py3_11-cuda12_1-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -2598,8 +2637,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 wheel-py3_11-cuda12_1-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: wheel-py3_11-cuda12_1-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - wheel-py3_11-cuda12_1-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -2731,7 +2772,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml wheel-py3_11-cuda12_4-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -2844,8 +2886,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 wheel-py3_11-cuda12_4-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: wheel-py3_11-cuda12_4-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - wheel-py3_11-cuda12_4-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -2977,7 +3021,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml wheel-py3_12-cpu-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -3089,8 +3134,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 wheel-py3_12-cpu-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: wheel-py3_12-cpu-build - runs-on: windows.4xlarge.nonephemeral + needs: + - wheel-py3_12-cpu-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -3220,7 +3267,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml wheel-py3_12-cuda11_8-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -3333,8 +3381,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 wheel-py3_12-cuda11_8-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: wheel-py3_12-cuda11_8-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - wheel-py3_12-cuda11_8-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -3466,7 +3516,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml wheel-py3_12-cuda12_1-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -3579,8 +3630,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 wheel-py3_12-cuda12_1-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: wheel-py3_12-cuda12_1-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - wheel-py3_12-cuda12_1-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -3712,7 +3765,8 @@ jobs: uses: ./.github/workflows/_binary-upload.yml wheel-py3_12-cuda12_4-build: if: ${{ github.repository_owner == 'pytorch' }} - runs-on: windows.4xlarge.nonephemeral + needs: get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch @@ -3825,8 +3879,10 @@ jobs: .github\scripts\kill_active_ssh_sessions.ps1 wheel-py3_12-cuda12_4-test: # Testing if: ${{ github.repository_owner == 'pytorch' }} - needs: wheel-py3_12-cuda12_4-build - runs-on: windows.8xlarge.nvidia.gpu + needs: + - wheel-py3_12-cuda12_4-build + - get-label-type + runs-on: "${{ needs.get-label-type.outputs.label-type }}windows.8xlarge.nvidia.gpu" timeout-minutes: 240 env: PYTORCH_ROOT: ${{ github.workspace }}/pytorch