crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-28 22:26:08 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Generalize source bank reading (don't rely on work_index)

Browse source
This commit is contained in:
Paul Romano 2020-10-23 13:26:02 -05:00
parent 5a60133d5b
commit f2526ec439
2 changed files with 17 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

33
src/state_point.cpp
View file

@ -659,27 +659,26 @@ void read_source_bank(hid_t group_id, std::vector<Particle::Bank>& sites)
// Open the dataset
hid_t dset = H5Dopen(group_id, "source_bank", H5P_DEFAULT);
// Create another data space but for each proc individually
hsize_t dims[] {static_cast<hsize_t>(simulation::work_per_rank)};
hid_t memspace = H5Screate_simple(1, dims, nullptr);
hid_t dspace = H5Dget_space(dset);
hsize_t n_sites;
H5Sget_simple_extent_dims(dspace, &n_sites, nullptr);
// Make sure source bank is big enough
hid_t dspace = H5Dget_space(dset);
hsize_t dims_all;
H5Sget_simple_extent_dims(dspace, &dims_all, nullptr);
if (&sites == &simulation::source_bank) {
if (simulation::work_index[mpi::n_procs] > dims_all) {
fatal_error("Number of source sites in source file is less "
"than number of source particles per generation.");
}
} else {
sites.resize(dims_all);
}
sites.resize(n_sites);
// Determine number of source particles to read from each rank and offset
hsize_t min_sites = n_sites / mpi::n_procs;
hsize_t remainder = n_sites % mpi::n_procs;
hsize_t n_sites_local = (mpi::rank < remainder) ? min_sites + 1 : min_sites;
hsize_t offset = (mpi::rank <= remainder) ?
mpi::rank*(min_sites + 1) :
remainder*(min_sites + 1) + (mpi::rank - remainder)*min_sites;
// Create another data space but for each proc individually
hid_t memspace = H5Screate_simple(1, &n_sites_local, nullptr);
// Select hyperslab for each process
hsize_t start[] {static_cast<hsize_t>(simulation::work_index[mpi::rank])};
H5Sselect_hyperslab(dspace, H5S_SELECT_SET, start, nullptr, dims, nullptr);
H5Sselect_hyperslab(dspace, H5S_SELECT_SET, &offset, nullptr, &n_sites_local, nullptr);
#ifdef PHDF5
// Read data in parallel

2
tests/regression_tests/source_file/results_true.dat
View file

@ -1,2 +1,2 @@
k-combined:
2.939525E-01 6.311780E-03
3.080726E-01 9.554124E-03