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

This commit is contained in:
Howard Su
2023-06-18 12:29:47 +08:00
committed by GitHub
parent b2416493ab
commit 57cd69460f

View File

@ -492,6 +492,10 @@ if (GGML_SOURCES_CUDA)
message(STATUS "GGML CUDA sources found, configuring CUDA architecture")
set_property(TARGET ggml PROPERTY CUDA_ARCHITECTURES OFF)
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)
endif()