OilfieldToolsNet/opm-simulators
4
0
Fork 0
mirror of https://github.com/OPM/opm-simulators.git synced 2025-02-25 18:55:30 -06:00
Code Issues Packages Projects Releases Wiki Activity

Added default setting for number of threads

Browse Source
This commit is contained in:
babrodtk 2015-09-02 10:19:50 +02:00
parent 1d6f12e870
commit c991ba8188
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
examples/flow.cpp
View File

@ -95,6 +95,10 @@
#include <boost/filesystem.hpp> #include <boost/filesystem.hpp>
#include <boost/algorithm/string.hpp> #include <boost/algorithm/string.hpp>
#ifdef _OPENMP
#include <omp.h>
#endif
#include <memory> #include <memory>
#include <algorithm> #include <algorithm>
#include <cstdlib> #include <cstdlib>
@ -150,6 +154,20 @@ try
std::cout << "**********************************************************************\n\n"; std::cout << "**********************************************************************\n\n";
} }
#ifdef _OPENMP
if (!getenv("OPM_NUM_THREADS")) {
//Default to max(4,#cores) threads,
//not number of cores (unless ENV(OMP_NUM_THREADS) is defined)
int num_cores = omp_get_num_procs();
int num_threads = std::min(4, num_cores);
omp_set_num_threads(num_threads);
}
#pragma omp parallel
if (omp_get_thread_num() == 0){
std::cout << "OpenMP using " << omp_get_num_threads() << " threads.";
}
#endif
// Read parameters, see if a deck was specified on the command line. // Read parameters, see if a deck was specified on the command line.
if ( output_cout ) if ( output_cout )
{ {