Spaces:
Running
Running
bobqianic
commited on
cmake : set default CUDA architectures (#1667)
Browse files- 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
|