diff --git a/src/run.cpp b/src/run.cpp index baf9d44..eb263f8 100644 --- a/src/run.cpp +++ b/src/run.cpp @@ -85,6 +85,11 @@ double run_bench(const bench_input &input, const std::string &output_file) { tug::Simulation 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.setIterations(input.iterations);