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)
12
12
# 61 == Pascal, __dp4a instruction (per-byte integer dot product)
13
13
# 70 == V100, FP16 tensor cores
14
14
# 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.
15
25
if (GGML_NATIVE AND CUDAToolkit_VERSION VERSION_GREATER_EQUAL "11.6" AND CMAKE_VERSION VERSION_GREATER_EQUAL "3.24" )
16
26
set (CMAKE_CUDA_ARCHITECTURES "native" )
17
27
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 " )
19
29
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 " )
21
31
endif ()
22
32
endif ()
23
33
message (STATUS "Using CUDA architectures: ${CMAKE_CUDA_ARCHITECTURES} " )
You can’t perform that action at this time.
0 commit comments