From e3074f221e063c91a08e9a05ed419b6b8557d5ca Mon Sep 17 00:00:00 2001 From: algol Date: Thu, 8 Feb 2018 16:44:29 +0000 Subject: Added flag for NVCC and ubuntu 16.04 https://groups.google.com/forum/#!msg/caffe-users/Tm3OsZBwN9Q/XKGRKNdmBAAJ --- Core/CMakeLists.txt | 2 ++ 1 file changed, 2 insertions(+) diff --git a/Core/CMakeLists.txt b/Core/CMakeLists.txt index fec49da..dceb828 100644 --- a/Core/CMakeLists.txt +++ b/Core/CMakeLists.txt @@ -123,6 +123,8 @@ endif() find_package(CUDA) if (CUDA_FOUND) + set(CUDA_NVCC_FLAGS "-Xcompiler -fPIC -shared -D_FORCE_INLINES") + message("CUDA FLAGS ${CUDA_NVCC_FLAGS}") CUDA_ADD_LIBRARY(cilregcuda SHARED ${CMAKE_CURRENT_SOURCE_DIR}/regularizers_GPU/Diffus_HO/Diff4th_GPU_kernel.cu ${CMAKE_CURRENT_SOURCE_DIR}/regularizers_GPU/NL_Regul/NLM_GPU_kernel.cu -- cgit v1.2.3