Add Turing and Volta support to arch_name (#13168)

apache · Nov 14, 2018 · b8e36e0 · b8e36e0
1 parent f5ba267
commit b8e36e0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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()