gcc 5.xx fixes (torch#421)

marsbzp · Jun 4, 2016 · f545703 · f545703
1 parent f8026ec
commit f545703
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/lib/THC/CMakeLists.txt b/lib/THC/CMakeLists.txt
@@ -10,6 +10,15 @@ IF(NOT MAGMA_FOUND)
   FIND_PACKAGE(MAGMA)
 ENDIF()
 
+if(CMAKE_CXX_COMPILER_ID STREQUAL "GNU")
+  if(CMAKE_CXX_COMPILER_VERSION VERSION_GREATER "4.9.3")
+    if(CUDA_VERSION VERSION_LESS "8.0")
+      MESSAGE(STATUS "Found gcc >=5 and CUDA <= 7.5, adding workaround C++ flags")
+      set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -D_FORCE_INLINES")
+    endif(CUDA_VERSION VERSION_LESS "8.0")
+  endif(CMAKE_CXX_COMPILER_VERSION VERSION_GREATER "4.9.3")
+endif(CMAKE_CXX_COMPILER_ID STREQUAL "GNU")
+
 INCLUDE_DIRECTORIES(${CUDA_INCLUDE_DIRS})
 INCLUDE_DIRECTORIES("${CUDA_SDK_ROOT_DIR}/common/inc")