add nvcc flags [ci skip]

f127bd3a · rusty1s · 1ef1a335 · f127bd3a
Commit f127bd3a authored Jul 16, 2020 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

CMakeLists.txt CMakeLists.txt +2 -0

No files found.
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -8,6 +8,8 @@ option(WITH_CUDA "Enable CUDA support" OFF)
 if(WITH_CUDA)
  enable_language(CUDA)
  add_definitions(-D__CUDA_NO_HALF_OPERATORS__)
+  add_definitions(-DWITH_CUDA)
+  set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -arch=sm_35 --expt-relaxed-constexpr")
 endif()
 find_package(Python3 COMPONENTS Development)