Skip to content

Commit 77574c2

Browse files
committed
ci(torch): Remove parameterization of TORCH_CUDA_ARCH_LIST
This is to fit within the 10-parameter limit for reusable workflows.
1 parent 25e8a9e commit 77574c2

File tree

1 file changed

+1
-10
lines changed

1 file changed

+1
-10
lines changed

.github/workflows/torch.yml

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -25,10 +25,6 @@ on:
2525
cxx11-abi:
2626
required: false
2727
type: string
28-
cuda-arch-support:
29-
required: false
30-
type: string
31-
default: "7.0 7.5 8.0 8.6 8.9 9.0+PTX"
3228
image-name:
3329
required: false
3430
type: string
@@ -74,11 +70,6 @@ on:
7470
required: false
7571
description: "Build with the CXX11 ABI (1 = enable, 0 = disable)"
7672
type: string
77-
cuda-arch-support:
78-
required: false
79-
description: "Space-separated list of CUDA architectures to support"
80-
type: string
81-
default: "7.0 7.5 8.0 8.6 8.9 9.0+PTX"
8273
image-name:
8374
required: false
8475
description: "Custom name under which to publish the resulting container"
@@ -106,8 +97,8 @@ jobs:
10697
BUILD_TORCH_VERSION=${{ inputs.torch-version }}
10798
BUILD_TORCH_VISION_VERSION=${{ inputs.torchvision-version }}
10899
BUILD_TORCH_AUDIO_VERSION=${{ inputs.torchaudio-version }}
100+
BUILD_TORCH_CUDA_ARCH_LIST=7.0 7.5 8.0 8.6 8.9 9.0+PTX
109101
${{ inputs.cxx11-abi && format('BUILD_CXX11_ABI={0}', inputs.cxx11-abi) || '' }}
110-
${{ inputs.cuda-arch-support && format('BUILD_TORCH_CUDA_ARCH_LIST={0}', inputs.cuda-arch-support) || '' }}
111102
${{ inputs.triton-version && format('BUILD_TRITON_VERSION={0}', inputs.triton-version) || '' }}
112103
build-extras:
113104
name: Build torch-extras

0 commit comments

Comments
 (0)