From b8e36e0771ecd42137b8d5755ad638bb06026cc7 Mon Sep 17 00:00:00 2001 From: Kellen Sunderland Date: Tue, 13 Nov 2018 21:31:16 -0700 Subject: [PATCH] Add Turing and Volta support to arch_name (#13168) --- cmake/FirstClassLangCuda.cmake | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/cmake/FirstClassLangCuda.cmake b/cmake/FirstClassLangCuda.cmake index 4b44077d2c55..f7ffcbbb8201 100644 --- a/cmake/FirstClassLangCuda.cmake +++ b/cmake/FirstClassLangCuda.cmake @@ -185,6 +185,12 @@ function(mshadow_select_nvcc_arch_flags out_variable) elseif(${arch_name} STREQUAL "Pascal") set(arch_bin 6.0 6.1) set(arch_ptx 6.1) + elseif(${arch_name} STREQUAL "Volta") + set(arch_bin 7.0) + set(arch_ptx 7.0) + elseif(${arch_name} STREQUAL "Turing") + set(arch_bin 7.5) + set(arch_ptx 7.5) else() message(SEND_ERROR "Unknown CUDA Architecture Name ${arch_name} in CUDA_SELECT_NVCC_ARCH_FLAGS") endif()