vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

test setting arch to all major

Browse source
This commit is contained in:
Concedo 2024-01-10 17:22:51 +08:00
parent 0526cc5d72
commit d2bf47987a
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
CMakeLists.txt
View file

@ -118,9 +118,9 @@ if (LLAMA_CUBLAS)
else() else()
message("CUDA Toolkit Version: ${CUDAToolkit_VERSION}") message("CUDA Toolkit Version: ${CUDAToolkit_VERSION}")
if(CUDAToolkit_VERSION VERSION_GREATER 12) if(CUDAToolkit_VERSION VERSION_GREATER 12)
set(CMAKE_CUDA_ARCHITECTURES "52;61;70") # lowest CUDA 12 standard + lowest for integer intrinsics set(CMAKE_CUDA_ARCHITECTURES "all-major") # lowest CUDA 12 standard + lowest for integer intrinsics
else() else()
set(CMAKE_CUDA_ARCHITECTURES "37;52;61;70") # lowest CUDA 12 standard + lowest for integer intrinsics set(CMAKE_CUDA_ARCHITECTURES "all-major") # lowest CUDA 12 standard + lowest for integer intrinsics
endif() endif()
endif() endif()
endif() endif()