From d7209b21fd5f388728c2fe40c92fcc4900d096e3 Mon Sep 17 00:00:00 2001
From: psychocrypt <psychocrypt@users.noreply.github.com>
Date: Fri, 20 Oct 2017 21:17:08 +0200
Subject: [PATCH] add NVIDIA Volta support

add `sm_70` if CUDA 9+ is avalable
---
 CMakeLists.txt | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 015f7c2..9ae7b26 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -79,6 +79,10 @@ if(CUDA_ENABLE)
         if(NOT CUDA_VERSION VERSION_LESS 8.0)
             list(APPEND DEFAULT_CUDA_ARCH "60" "61" "62")
         endif()
+        # add Volta support for CUDA >= 9.0
+        if(NOT CUDA_VERSION VERSION_LESS 9.0)
+            list(APPEND DEFAULT_CUDA_ARCH "70")
+        endif()
         set(CUDA_ARCH "${DEFAULT_CUDA_ARCH}" CACHE STRING "Set GPU architecture (semicolon separated list, e.g. '-DCUDA_ARCH=20;35;60')")
 
         # validate architectures (only numbers are allowed)