[incubator-mxnet] branch master updated: Add Turing and Volta support to arch_name (#13168)

[zh...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

zhasheng pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/master by this push:
     new b8e36e0  Add Turing and Volta support to arch_name (#13168)
b8e36e0 is described below

commit b8e36e0771ecd42137b8d5755ad638bb06026cc7
Author: Kellen Sunderland <ke...@gmail.com>
AuthorDate: Tue Nov 13 21:31:16 2018 -0700

    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 4b44077..f7ffcbb 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()