bobqianic commited on
Commit
0969db5
·
unverified ·
1 Parent(s): 282c3a3

cmake : set default CUDA architectures (#1667)

Browse files
Files changed (1) hide show
  1. CMakeLists.txt +7 -1
CMakeLists.txt CHANGED
@@ -521,7 +521,13 @@ endif()
521
 
522
  if (GGML_SOURCES_CUDA)
523
  message(STATUS "GGML CUDA sources found, configuring CUDA architecture")
524
- set_property(TARGET whisper PROPERTY CUDA_ARCHITECTURES OFF)
 
 
 
 
 
 
525
  set_property(TARGET whisper PROPERTY CUDA_SELECT_NVCC_ARCH_FLAGS "Auto")
526
  endif()
527
 
 
521
 
522
  if (GGML_SOURCES_CUDA)
523
  message(STATUS "GGML CUDA sources found, configuring CUDA architecture")
524
+ # Only configure gmml CUDA architectures is not globally set
525
+ if (NOT DEFINED GGML_CUDA_ARCHITECTURES)
526
+ # Not overriden by user, so set defaults
527
+ set(GGML_CUDA_ARCHITECTURES 52 61 70)
528
+ endif()
529
+ message(STATUS "GGML Configuring CUDA architectures ${GGML_CUDA_ARCHITECTURES}")
530
+ set_property(TARGET whisper PROPERTY CUDA_ARCHITECTURES ${GGML_CUDA_ARCHITECTURES})
531
  set_property(TARGET whisper PROPERTY CUDA_SELECT_NVCC_ARCH_FLAGS "Auto")
532
  endif()
533