Skip to content

Commit d6cb770

Browse files
authored
[cmake] fix sm 86 is only for cuda >= 11.1
1 parent c8a669b commit d6cb770

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

cmake/ChooseCudaCC.cmake

+4-1
Original file line numberDiff line numberDiff line change
@@ -80,8 +80,11 @@ function(chooseCudaCC SUPPORTED_CC SUPPORTED_GENCODE_FLAGS)
8080
#
8181
set(CUDA_MIN_CC 20)
8282
set(CUDA_MAX_CC 86)
83-
if(CUDA_VERSION_MAJOR GREATER_EQUAL 11)
83+
if(CUDA_VERSION VERSION_GREATER_EQUAL 11.1)
8484
set(CUDA_MIN_CC 35)
85+
elseif(CUDA_VERSION_MAJOR GREATER_EQUAL 11)
86+
set(CUDA_MIN_CC 35)
87+
set(CUDA_MAX_CC 80)
8588
elseif(CUDA_VERSION_MAJOR GREATER_EQUAL 10)
8689
set(CUDA_MIN_CC 30)
8790
set(CUDA_MAX_CC 75)

0 commit comments

Comments
 (0)