File tree Expand file tree Collapse file tree 1 file changed +1
-10
lines changed
Expand file tree Collapse file tree 1 file changed +1
-10
lines changed Original file line number Diff line number Diff line change 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
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"
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
You can’t perform that action at this time.
0 commit comments