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 <[email protected]>
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()
