Просмотр исходного кода

cmake : add CUDA_ARCHITECTURES to new target ggml_static (#1917)

Howard Su 2 лет назад
Родитель
Сommit
57cd69460f
1 измененных файлов с 4 добавлено и 0 удалено
  1. 4 0
      CMakeLists.txt

+ 4 - 0
CMakeLists.txt

@@ -492,6 +492,10 @@ if (GGML_SOURCES_CUDA)
     message(STATUS "GGML CUDA sources found, configuring CUDA architecture")
     message(STATUS "GGML CUDA sources found, configuring CUDA architecture")
     set_property(TARGET ggml  PROPERTY CUDA_ARCHITECTURES OFF)
     set_property(TARGET ggml  PROPERTY CUDA_ARCHITECTURES OFF)
     set_property(TARGET ggml  PROPERTY CUDA_SELECT_NVCC_ARCH_FLAGS "Auto")
     set_property(TARGET ggml  PROPERTY CUDA_SELECT_NVCC_ARCH_FLAGS "Auto")
+
+    set_property(TARGET ggml_static PROPERTY CUDA_ARCHITECTURES OFF)
+    set_property(TARGET ggml_static PROPERTY CUDA_SELECT_NVCC_ARCH_FLAGS "Auto")
+
     set_property(TARGET llama PROPERTY CUDA_ARCHITECTURES OFF)
     set_property(TARGET llama PROPERTY CUDA_ARCHITECTURES OFF)
 endif()
 endif()