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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats 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 # Set up architecture flags
if(NOT flags) if(NOT flags)
if (CUDA_VERSION VERSION_GREATER_EQUAL "11.0") 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") elseif(CUDA_VERSION VERSION_GREATER_EQUAL "10.0")
set(flags "35;50;52;60;61;70;75") set(flags "35;50;52;60;61;70;75")
elseif(CUDA_VERSION VERSION_GREATER_EQUAL "9.0") elseif(CUDA_VERSION VERSION_GREATER_EQUAL "9.0")