diff options
| author | algol <dkazanc@hotmail.com> | 2018-02-08 16:44:29 +0000 |
|---|---|---|
| committer | Edoardo Pasca <edo.paskino@gmail.com> | 2018-02-09 16:24:47 +0000 |
| commit | e3074f221e063c91a08e9a05ed419b6b8557d5ca (patch) | |
| tree | cd7e660efe9f45df8479e47e261ef5d011c924e4 | |
| parent | eb2633a1cd839a3cc9af23b95c0257648ad7e4d5 (diff) | |
| download | regularization-e3074f221e063c91a08e9a05ed419b6b8557d5ca.tar.gz regularization-e3074f221e063c91a08e9a05ed419b6b8557d5ca.tar.bz2 regularization-e3074f221e063c91a08e9a05ed419b6b8557d5ca.tar.xz regularization-e3074f221e063c91a08e9a05ed419b6b8557d5ca.zip | |
Added flag for NVCC and ubuntu 16.04
https://groups.google.com/forum/#!msg/caffe-users/Tm3OsZBwN9Q/XKGRKNdmBAAJ
| -rw-r--r-- | Core/CMakeLists.txt | 2 |
1 files changed, 2 insertions, 0 deletions
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 |