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

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

Browse Source
This commit is contained in:
DannyPuhan 2024-06-06 09:45:42 +02:00
parent 5f333fcb77
commit 4692a246d5
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);