Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[breaking] Remove CUDA sm_35, add sm_86 (#7182)

Browse Source
This commit is contained in:
Jiaming Yuan
2021-08-25 16:04:23 +08:00
committed by GitHub
parent d04312b9c0
commit 9c64618cb6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cmake/Utils.cmake
View File

@@ -91,7 +91,7 @@ function(format_gencode_flags flags out)
# Set up architecture flags
if(NOT flags)
if (CUDA_VERSION VERSION_GREATER_EQUAL "11.0")
set(flags "35;50;52;60;61;70;75;80")
set(flags "50;52;60;61;70;75;80;86")
elseif(CUDA_VERSION VERSION_GREATER_EQUAL "10.0")
set(flags "35;50;52;60;61;70;75")
elseif(CUDA_VERSION VERSION_GREATER_EQUAL "9.0")