Skip to content

Commit 0392273

Browse files
authored
chore: add compute kernels to Windows CUDA build (#1062)
* Fix syntax for CUDA architecture definitions * Extend CUDA support to GTX 10 Series to RTX 50 Series * update cuda installer step version to install cuda 12.8.1 * Remove unsupported compute capability
1 parent bf1a388 commit 0392273

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

.github/workflows/build.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ jobs:
163163
- build: "avx512"
164164
defines: "-DGGML_NATIVE=OFF -DGGML_AVX512=ON -DGGML_AVX=ON -DGGML_AVX2=ON -DSD_BUILD_SHARED_LIBS=ON"
165165
- build: "cuda12"
166-
defines: "-DSD_CUDA=ON -DSD_BUILD_SHARED_LIBS=ON -DCMAKE_CUDA_ARCHITECTURES=90;89;86;80;75"
166+
defines: "-DSD_CUDA=ON -DSD_BUILD_SHARED_LIBS=ON -DCMAKE_CUDA_ARCHITECTURES='61;70;75;80;86;89;90;100;120'"
167167
- build: 'vulkan'
168168
defines: "-DSD_VULKAN=ON -DSD_BUILD_SHARED_LIBS=ON"
169169
steps:
@@ -176,9 +176,9 @@ jobs:
176176
- name: Install cuda-toolkit
177177
id: cuda-toolkit
178178
if: ${{ matrix.build == 'cuda12' }}
179-
uses: Jimver/[email protected].19
179+
uses: Jimver/[email protected].22
180180
with:
181-
cuda: "12.6.2"
181+
cuda: "12.8.1"
182182
method: "network"
183183
sub-packages: '["nvcc", "cudart", "cublas", "cublas_dev", "thrust", "visual_studio_integration"]'
184184

0 commit comments

Comments
 (0)