Skip to content

Commit e164fbc

Browse files
committed
Explicitly set list of CUDA architectures to stop 120 being built
Curse you Blackwell!!
1 parent 1507bfb commit e164fbc

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

CMakeLists.txt

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -211,7 +211,9 @@ if(WIN32)
211211
212212
if(WHISPERCPP_NVIDIA)
213213
# Build with CUDA
214-
list(APPEND WHISPER_ADDITIONAL_CMAKE_ARGS -DGGML_CUDA=ON)
214+
list(APPEND CUDA_TARGETS 50-virtual 60-virtual 61 70-virtual 75 80 86 89 120a-real)
215+
list(APPEND WHISPER_ADDITIONAL_CMAKE_ARGS -DGGML_CUDA=ON
216+
"-DCMAKE_CUDA_ARCHITECTURES=${CUDA_TARGETS}")
215217
elseif(WHISPERCPP_AMD)
216218
# Build with hipBLAS
217219
if(NOT DEFINED ENV{HIP_PATH})

0 commit comments

Comments
 (0)