We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1507bfb commit e164fbcCopy full SHA for e164fbc
1 file changed
CMakeLists.txt
@@ -211,7 +211,9 @@ if(WIN32)
211
212
if(WHISPERCPP_NVIDIA)
213
# Build with CUDA
214
- list(APPEND WHISPER_ADDITIONAL_CMAKE_ARGS -DGGML_CUDA=ON)
+ 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}")
217
elseif(WHISPERCPP_AMD)
218
# Build with hipBLAS
219
if(NOT DEFINED ENV{HIP_PATH})
0 commit comments