File tree Expand file tree Collapse file tree 1 file changed +12
-2
lines changed Expand file tree Collapse file tree 1 file changed +12
-2
lines changed Original file line number Diff line number Diff line change @@ -12,12 +12,22 @@ if (CUDAToolkit_FOUND)
1212 # 61 == Pascal, __dp4a instruction (per-byte integer dot product)
1313 # 70 == V100, FP16 tensor cores
1414 # 75 == Turing, int8 tensor cores
15+ # 80 == Ampere, asynchronous data loading, faster tensor core instructions
16+ # 86 == RTX 3000
17+ # 89 == RTX 4000
18+ #
19+ # XX-virtual == compile CUDA code as PTX, do JIT compilation to binary code on first run
20+ # XX-real == compile CUDA code as device code for this specific architecture
21+ # no suffix == compile as both PTX and device code
22+ #
23+ # The default behavior for a non-native is to build virtual architectures as needed to cover all features needed
24+ # for best performance and to also build real architectures for the most commonly used GPUs.
1525 if (GGML_NATIVE AND CUDAToolkit_VERSION VERSION_GREATER_EQUAL "11.6" AND CMAKE_VERSION VERSION_GREATER_EQUAL "3.24" )
1626 set (CMAKE_CUDA_ARCHITECTURES "native" )
1727 elseif (GGML_CUDA_F16 OR GGML_CUDA_DMMV_F16)
18- set (CMAKE_CUDA_ARCHITECTURES "60;61;70;75;80" )
28+ set (CMAKE_CUDA_ARCHITECTURES "60-virtual ;61-virtual ;70-virtual ;75-virtual ;80-virtual;86-real;89-real " )
1929 else ()
20- set (CMAKE_CUDA_ARCHITECTURES "50;61;70;75;80" )
30+ set (CMAKE_CUDA_ARCHITECTURES "50-virtual ;61-virtual ;70-virtual ;75-virtual ;80-virtual;86-real;89-real " )
2131 endif ()
2232 endif ()
2333 message (STATUS "Using CUDA architectures: ${CMAKE_CUDA_ARCHITECTURES} " )
You can’t perform that action at this time.
0 commit comments