Skip to content

Commit bf0cd2b

Browse files
committed
CMake changes
1 parent 80f3cf0 commit bf0cd2b

File tree

1 file changed

+12
-8
lines changed

1 file changed

+12
-8
lines changed

CMakeLists.txt

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ option(BUILD_C_BINDINGS "Build C bindings" ON)
4343
option(BUILD_PYTHON_BINDINGS "Build Python bindings" ON)
4444
option(BUILD_MATLAB_BINDINGS "Build Matlab bindings" ON)
4545
option(BUILD_CUDA_LIB "Build CUDA library" OFF)
46+
option(USE_OPENMP "Use OpenMP multi-threading" ON)
4647
option(USE_MPI "Use MPI" OFF)
4748

4849
set(NVCC_COMPILER_BINDIR "" CACHE PATH "Directory where nvcc should look for C++ compiler. This is passed to nvcc through the --compiler-bindir option.")
@@ -95,13 +96,15 @@ if (NOT GTEST_FOUND)
9596
endif()
9697

9798

98-
find_package(OpenMP)
99-
if(OPENMP_FOUND)
100-
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}")
101-
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}")
102-
set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} ${OpenMP_EXE_LINKER_FLAGS}")
103-
else()
104-
message(WARNING "OpenMP NOT found")
99+
if (USE_OPENMP)
100+
find_package(OpenMP)
101+
if(OPENMP_FOUND)
102+
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}")
103+
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}")
104+
set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} ${OpenMP_EXE_LINKER_FLAGS}")
105+
else()
106+
message(WARNING "OpenMP NOT found")
107+
endif()
105108
endif()
106109

107110

@@ -128,7 +131,7 @@ if (WIN32)
128131
#add_definitions( "/W1" )
129132
add_definitions("-D_CRT_SECURE_NO_WARNINGS -D_SCL_SECURE_NO_WARNINGS /bigobj")
130133
else(WIN32)
131-
add_definitions( "-Wall" )
134+
add_definitions( "-Wall -Wno-unknown-pragmas" )
132135
endif(WIN32)
133136

134137
add_subdirectory( cmake )
@@ -181,4 +184,5 @@ message(STATUS "Building C bindings: ${BUILD_C_BINDINGS}")
181184
message(STATUS "Building python bindings: ${BUILD_PYTHON_BINDINGS}")
182185
message(STATUS "Building matlab bindings: ${BUILD_MATLAB_BINDINGS}")
183186
message(STATUS "Building CUDA library: ${BUILD_CUDA_LIB}")
187+
message(STATUS "Using OpenMP support: ${USE_OPENMP}")
184188
message(STATUS "Using MPI support: ${USE_MPI}")

0 commit comments

Comments
 (0)