diff options
| author | Nikolay Bogoychev <nheart@gmail.com> | 2021-05-04 14:36:10 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-05-04 14:36:10 +0300 |
| commit | 379212b75c67b04962f084b7190f024b12dd0068 (patch) | |
| tree | dbf1d7fda82eec86c8364fb09f60f55d5d0042f5 /CMakeLists.txt | |
| parent | 36b4b69d7bbbe5e58cef4499011bef29feebf8b3 (diff) | |
Enable compute86 where supported (#863)
* Enable compute86 where supported
Diffstat (limited to 'CMakeLists.txt')
| -rw-r--r-- | CMakeLists.txt | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index 79c8585e..119bc01f 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -325,6 +325,16 @@ if(CUDA_FOUND) option(COMPILE_AMPERE "Compile GPU version with SM80 support" ON) LIST(APPEND COMPUTE -Wno-deprecated-gpu-targets) endif() + if(CUDA_VERSION VERSION_EQUAL "11.1" OR CUDA_VERSION VERSION_GREATER "11.1") + option(COMPILE_KEPLER "Compile GPU version with SM35 support" OFF) # deprecated for CUDA 11 + option(COMPILE_MAXWELL "Compile GPU version with SM50 support" OFF) # deprecated for CUDA 11 + option(COMPILE_PASCAL "Compile GPU version with SM60 support" ON) + option(COMPILE_VOLTA "Compile GPU version with SM70 support" ON) + option(COMPILE_TURING "Compile GPU version with SM75 support" ON) + option(COMPILE_AMPERE "Compile GPU version with SM80 support" ON) + option(COMPILE_AMPERE_RTX "Compile GPU version with SM86 support" ON) + LIST(APPEND COMPUTE -Wno-deprecated-gpu-targets) + endif() if(COMPILE_KEPLER) message(STATUS "Compiling code for Kepler GPUs") @@ -354,6 +364,12 @@ if(CUDA_FOUND) LIST(APPEND COMPUTE -gencode=arch=compute_80,code=sm_80; -gencode=arch=compute_80,code=compute_80) # Ampere GPUs endif(COMPILE_AMPERE) endif() + if(CUDA_VERSION VERSION_EQUAL "11.1" OR CUDA_VERSION VERSION_GREATER "11.1") + if(COMPILE_AMPERE_RTX) + message(STATUS "Compiling code for Ampere RTX GPUs") + LIST(APPEND COMPUTE -gencode=arch=compute_86,code=sm_86; -gencode=arch=compute_86,code=compute_86) # Ampere RTX GPUs + endif(COMPILE_AMPERE_RTX) + endif() if(USE_STATIC_LIBS) set(EXT_LIBS ${EXT_LIBS} ${CUDA_curand_LIBRARY} ${CUDA_CUBLAS_LIBRARIES} ${CUDA_cusparse_LIBRARY}) |