cuda : fix CUDA arch: OFF -> 52;61 (#389)

diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index 22e02590f5756a62d9a8e75c01eda3db1fc57b3f..2a1db244b2446baa419e13bb1d1897a3641f3e7a 100644 (file)
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -255,7 +255,7 @@ endif()
 
 if (GGML_CUDA_SOURCES)
     message(STATUS "GGML CUDA sources found, configuring CUDA architecture")
-    set_property(TARGET ggml  PROPERTY CUDA_ARCHITECTURES OFF)
+    set_property(TARGET ggml  PROPERTY CUDA_ARCHITECTURES "52;61")
     set_property(TARGET ggml  PROPERTY CUDA_SELECT_NVCC_ARCH_FLAGS "Auto")
     target_link_libraries(ggml PUBLIC stdc++)
 endif()