max/tug
1
0
Fork 0
mirror of https://git.gfz-potsdam.de/naaice/tug.git synced 2025-12-13 17:38:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

Added OpenMP pragmas to library.

Browse Source
This commit is contained in:
Max Luebke 2022-03-09 13:14:22 +01:00
parent c2211c8a6f
commit 97f31887ea
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/BTCSDiffusion.cpp
View File

@ -14,6 +14,7 @@
#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>
@ -143,6 +144,7 @@ void Diffusion::BTCSDiffusion::simulate2D(
t0_c = calc_t0_c(c, alpha, bc, local_dt, dx); t0_c = calc_t0_c(c, alpha, bc, local_dt, dx);
#pragma omp parallel for schedule(dynamic)
for (int i = 0; i < n_rows; i++) { for (int i = 0; i < n_rows; i++) {
DVectorRowMajor input_field = c.row(i); DVectorRowMajor input_field = c.row(i);
simulate_base(input_field, bc.row(i), alpha.row(i), dx, local_dt, n_cols, simulate_base(input_field, bc.row(i), alpha.row(i), dx, local_dt, n_cols,
@ -155,6 +157,7 @@ void Diffusion::BTCSDiffusion::simulate2D(
t0_c = t0_c =
calc_t0_c(c.transpose(), alpha.transpose(), bc.transpose(), local_dt, dx); calc_t0_c(c.transpose(), alpha.transpose(), bc.transpose(), local_dt, dx);
#pragma omp parallel for schedule(dynamic)
for (int i = 0; i < n_cols; i++) { for (int i = 0; i < n_cols; i++) {
DVectorRowMajor input_field = c.col(i); DVectorRowMajor input_field = c.col(i);
simulate_base(input_field, bc.col(i), alpha.col(i), dx, local_dt, n_rows, simulate_base(input_field, bc.col(i), alpha.col(i), dx, local_dt, n_rows,
@ -186,7 +189,8 @@ auto Diffusion::BTCSDiffusion::calc_t0_c(const DMatrixRowMajor &c,
(y_values[0] - 2 * y_values[1] + y_values[2]) / (dx * dx); (y_values[0] - 2 * y_values[1] + y_values[2]) / (dx * dx);
} }
// then iterate over inlet // then iterate over inlet
#pragma omp parallel for private(y_values) schedule(dynamic)
for (int i = 1; i < n_rows - 1; i++) { for (int i = 1; i < n_rows - 1; i++) {
for (int j = 0; j < n_cols; j++) { for (int j = 0; j < n_cols; j++) {

4
src/CMakeLists.txt
View File

@ -1,3 +1,5 @@
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) target_link_libraries(diffusion Eigen3::Eigen OpenMP::OpenMP_CXX)
target_include_directories(diffusion PUBLIC ${CMAKE_CURRENT_SOURCE_DIR}) target_include_directories(diffusion PUBLIC ${CMAKE_CURRENT_SOURCE_DIR})