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

ggerganov · Jul 14, 2023 · 90cb1ef · 90cb1ef
1 parent 5bef75b
commit 90cb1ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()