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
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

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()