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

Tune cuda architectures (#8152)

Browse Source
This commit is contained in:
Rory Mitchell 2022-08-11 22:36:47 +02:00 committed by GitHub
parent 16bca5d4a1
commit f421c26d35
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
cmake/Utils.cmake
View File

@ -91,21 +91,21 @@ function(format_gencode_flags flags out)
# Set up architecture flags
if(NOT flags)
if (CUDA_VERSION VERSION_GREATER_EQUAL "11.1")
set(flags "52;60;61;70;75;80;86")
set(flags "50;60;70;80")
elseif (CUDA_VERSION VERSION_GREATER_EQUAL "11.0")
set(flags "52;60;61;70;75;80")
set(flags "50;60;70;80")
elseif(CUDA_VERSION VERSION_GREATER_EQUAL "10.0")
set(flags "35;50;52;60;61;70;75")
set(flags "35;50;60;70")
elseif(CUDA_VERSION VERSION_GREATER_EQUAL "9.0")
set(flags "35;50;52;60;61;70")
set(flags "35;50;60;70")
else()
set(flags "35;50;52;60;61")
set(flags "35;50;60")
endif()
endif()
if (CMAKE_VERSION VERSION_GREATER_EQUAL "3.18")
cmake_policy(SET CMP0104 NEW)
list(POP_BACK flags latest_arch)
list(GET flags -1 latest_arch)
list(TRANSFORM flags APPEND "-real")
list(APPEND flags ${latest_arch})
set(CMAKE_CUDA_ARCHITECTURES ${flags})