From b1e28fa5110891a6d25a8b90e07b9d8684b6e561 Mon Sep 17 00:00:00 2001 From: jianlins Date: Wed, 11 Mar 2026 22:28:21 -0600 Subject: [PATCH] workflow: add gpu_arch input for customizable CUDA architecture in build script --- .github/workflows/custom_build_rockylinux.yml | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/.github/workflows/custom_build_rockylinux.yml b/.github/workflows/custom_build_rockylinux.yml index 809ee85ba6..5a82b28cee 100644 --- a/.github/workflows/custom_build_rockylinux.yml +++ b/.github/workflows/custom_build_rockylinux.yml @@ -7,6 +7,11 @@ on: description: 'Create new release' required: true type: boolean + gpu_arch: + description: 'nvidia gpu arch numbers (semicolon-separated, e.g. 70;75;80;86)' + required: false + type: string + default: '70;75;80;86;89' release: types: [published] @@ -73,7 +78,8 @@ jobs: -G Ninja \ -DCMAKE_BUILD_TYPE=${BUILD_TYPE} \ -DGGML_CUDA=ON \ - -DCMAKE_CUDA_COMPILER=$(command -v nvcc) + -DCMAKE_CUDA_COMPILER=$(command -v nvcc) \ + -DCMAKE_CUDA_ARCHITECTURES="${{ github.event.inputs.gpu_arch }}" - name: Build shell: bash