Skip to content

Commit 16f7c18

Browse files
authored
Update backends.yml
1 parent b57c6d8 commit 16f7c18

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

.github/workflows/backends.yml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ jobs:
3131
- build: 'avx512'
3232
defines: '-DGGML_AVX512=ON -DSD_BUILD_SHARED_LIBS=ON'
3333
- build: 'cuda12'
34-
defines: '-DSD_CUDA=ON -DSD_BUILD_SHARED_LIBS=ON'
34+
defines: '-DSD_CUDA=ON -DCMAKE_CUDA_ARCHITECTURES=61,75,86,89,100 -DSD_BUILD_SHARED_LIBS=ON'
3535
- build: 'vulkan'
3636
defines: "-DSD_VULKAN=ON -DSD_BUILD_SHARED_LIBS=ON"
3737

@@ -84,7 +84,7 @@ jobs:
8484
matrix:
8585
include:
8686
- build: 'cuda11'
87-
defines: '-DSD_CUDA=ON -DSD_BUILD_SHARED_LIBS=ON -DCMAKE_CUDA_FLAGS="-allow-unsupported-compiler"'
87+
defines: '-DSD_CUDA=ON -DSD_BUILD_SHARED_LIBS=ON -DCMAKE_CUDA_ARCHITECTURES=61 -DCMAKE_CUDA_FLAGS="-allow-unsupported-compiler"'
8888

8989
steps:
9090
- name: Checkout
@@ -225,9 +225,9 @@ jobs:
225225
- build: 'avx512'
226226
defines: '-DGGML_AVX512=ON -DSD_BUILD_SHARED_LIBS=ON'
227227
- build: 'cuda11'
228-
defines: '-DSD_CUDA=ON -DSD_BUILD_SHARED_LIBS=ON'
228+
defines: '-DSD_CUDA=ON -DCMAKE_CUDA_ARCHITECTURES=61 -DSD_BUILD_SHARED_LIBS=ON'
229229
- build: 'cuda12'
230-
defines: '-DSD_CUDA=ON -DSD_BUILD_SHARED_LIBS=ON'
230+
defines: '-DSD_CUDA=ON -DCMAKE_CUDA_ARCHITECTURES=61,75,86,89,100 -DSD_BUILD_SHARED_LIBS=ON'
231231
- build: 'sycl'
232232
defines: '-DSD_SYCL=ON -DCMAKE_C_COMPILER=icx -DCMAKE_CXX_COMPILER=icpx -DSD_BUILD_SHARED_LIBS=ON'
233233

0 commit comments

Comments
 (0)