Skip to content

Commit

Permalink
ci(torch): Remove parameterization of TORCH_CUDA_ARCH_LIST
Browse files Browse the repository at this point in the history
This is to fit within the 10-parameter limit for reusable workflows.
  • Loading branch information
Eta0 committed Jan 30, 2025
1 parent 25e8a9e commit 77574c2
Showing 1 changed file with 1 addition and 10 deletions.
11 changes: 1 addition & 10 deletions .github/workflows/torch.yml
Original file line number Diff line number Diff line change
Expand Up @@ -25,10 +25,6 @@ on:
cxx11-abi:
required: false
type: string
cuda-arch-support:
required: false
type: string
default: "7.0 7.5 8.0 8.6 8.9 9.0+PTX"
image-name:
required: false
type: string
Expand Down Expand Up @@ -74,11 +70,6 @@ on:
required: false
description: "Build with the CXX11 ABI (1 = enable, 0 = disable)"
type: string
cuda-arch-support:
required: false
description: "Space-separated list of CUDA architectures to support"
type: string
default: "7.0 7.5 8.0 8.6 8.9 9.0+PTX"
image-name:
required: false
description: "Custom name under which to publish the resulting container"
Expand Down Expand Up @@ -106,8 +97,8 @@ jobs:
BUILD_TORCH_VERSION=${{ inputs.torch-version }}
BUILD_TORCH_VISION_VERSION=${{ inputs.torchvision-version }}
BUILD_TORCH_AUDIO_VERSION=${{ inputs.torchaudio-version }}
BUILD_TORCH_CUDA_ARCH_LIST=7.0 7.5 8.0 8.6 8.9 9.0+PTX
${{ inputs.cxx11-abi && format('BUILD_CXX11_ABI={0}', inputs.cxx11-abi) || '' }}
${{ inputs.cuda-arch-support && format('BUILD_TORCH_CUDA_ARCH_LIST={0}', inputs.cuda-arch-support) || '' }}
${{ inputs.triton-version && format('BUILD_TRITON_VERSION={0}', inputs.triton-version) || '' }}
build-extras:
name: Build torch-extras
Expand Down

0 comments on commit 77574c2

Please sign in to comment.