max/TugJulia
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added optional usage of OpenMP

Browse Source
This commit is contained in:
Max Luebke 2022-03-25 11:28:00 +01:00
parent abc0a5713b
commit dd96bb9bf1
3 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
CMakeLists.txt
View File

@ -6,6 +6,9 @@ project(Diffusion CXX)
set(CMAKE_CXX_STANDARD 14) set(CMAKE_CXX_STANDARD 14)
find_package(Eigen3 REQUIRED NO_MODULE) find_package(Eigen3 REQUIRED NO_MODULE)
find_package(OpenMP)
option(USE_OPENMP "Compile with OpenMP support" ON)
add_subdirectory(app) add_subdirectory(app)
add_subdirectory(src) add_subdirectory(src)

7
src/BTCSDiffusion.cpp
View File

@ -14,11 +14,16 @@
#include <cstddef> #include <cstddef>
#include <iomanip> #include <iomanip>
#include <iterator> #include <iterator>
#include <omp.h>
#include <ostream> #include <ostream>
#include <tuple> #include <tuple>
#include <vector> #include <vector>
#ifdef _OPENMP
#include <omp.h>
#else
#define omp_get_thread_num() 0
#endif
#include <iostream> #include <iostream>
constexpr int BTCS_MAX_DEP_PER_CELL = 3; constexpr int BTCS_MAX_DEP_PER_CELL = 3;

8
src/CMakeLists.txt
View File

@ -1,5 +1,11 @@
find_package(OpenMP) find_package(OpenMP)
add_library(diffusion OBJECT BTCSDiffusion.cpp BTCSDiffusion.hpp) add_library(diffusion OBJECT BTCSDiffusion.cpp BTCSDiffusion.hpp)
target_link_libraries(diffusion Eigen3::Eigen OpenMP::OpenMP_CXX)
target_link_libraries(diffusion Eigen3::Eigen)
if(USE_OPENMP AND OpenMP_CXX_FOUND)
target_link_libraries(diffusion OpenMP::OpenMP_CXX)
endif()
target_include_directories(diffusion PUBLIC ${CMAKE_CURRENT_SOURCE_DIR}) target_include_directories(diffusion PUBLIC ${CMAKE_CURRENT_SOURCE_DIR})