Set C++17 only with CUDA>10

openmm · Aug 8, 2023 · 0e16c49 · 0e16c49
1 parent acc626e
commit 0e16c49
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -37,7 +37,11 @@ target_include_directories(${LIBRARY} PRIVATE ${Python3_INCLUDE_DIRS}
                                               src/ani src/pytorch src/schnet)
 target_link_libraries(${LIBRARY} ${TORCH_LIBRARIES} ${Python3_LIBRARIES})
 if(ENABLE_CUDA)
+  if(CMAKE_CUDA_COMPILER_VERSION VERSION_LESS 11)
+    set_property(TARGET ${LIBRARY} PROPERTY CUDA_STANDARD 14)
+  else()
     set_property(TARGET ${LIBRARY} PROPERTY CUDA_STANDARD 17)
+  endif()
     target_compile_definitions(${LIBRARY} PRIVATE ENABLE_CUDA)
 endif(ENABLE_CUDA)