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

Merge branch 'feat-set-num-threads-via-env-variable' into 'master'

Browse Source 
[feat] Set number of threads via env variable `OMP_NUM_THREADS`.

See merge request naaice/naaice_tug_input!1
This commit is contained in:
Max Lübke 2024-06-06 11:55:12 +02:00
commit 20c2917606
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/run.cpp
View File

@ -85,6 +85,11 @@ double run_bench(const bench_input &input, const std::string &output_file) {
tug::Simulation<TugType> sim(grid, boundary); tug::Simulation<TugType> sim(grid, boundary);
if (const char *out = std::getenv("OMP_NUM_THREADS")) {
int ompNumThreads = std::stoi(out, NULL);
sim.setNumberThreads(ompNumThreads);
}
sim.setTimestep(input.timestep); sim.setTimestep(input.timestep);
sim.setIterations(input.iterations); sim.setIterations(input.iterations);