crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-27 13:45:36 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add a new variable to store maximum bank size

Browse source
This commit is contained in:
YoungHui Park 2020-07-29 11:33:27 -05:00
parent 9fd222f483
commit 5091418d71
3 changed files with 14 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/simulation.cpp
View file

@ -271,6 +271,7 @@ std::vector<double> k_generation;
std::vector<int64_t> work_index;
int64_t total_surf_banks {0};
int64_t max_bank_size {0};
std::vector<int64_t> surf_src_index;
} // namespace simulation
@ -557,6 +558,9 @@ void calculate_work()
// Number of particles for rank i
int64_t work_i = i < remainder ? min_work + 1 : min_work;
// Set maximum bank size
if (mpi::master) simulation::max_bank_size = work_i;
// Set number of particles
if (mpi::rank == i) simulation::work_per_rank = work_i;
@ -656,6 +660,7 @@ void query_surf_src_size()
for (int i = 1; i < mpi::n_procs + 1; ++i) {
simulation::surf_src_index[i] = simulation::surf_src_index[i - 1] + bank_size[i - 1];
}
simulation::max_bank_size = *std::max_element(bank_size.begin(), bank_size.end());
total = simulation::surf_src_index[mpi::n_procs];
}
#else