Add Turing and Volta support to arch_name (apache#13168)

azai91 · Dec 1, 2018 · 680f633 · 680f633
1 parent 8cb426e
commit 680f633
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()