ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor for clarity

Browse Source
This commit is contained in:
Axel Kohlmeyer
2020-12-17 09:34:56 -05:00
parent 5ddeb45c0a
commit 527ffa79dc
2 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/fix_temp_csld.cpp
View File

@ -305,16 +305,17 @@ double FixTempCSLD::compute_scalar()
void FixTempCSLD::write_restart(FILE *fp)
{
int nsize = (98+2+3)*comm->nprocs+2; // pRNG state per proc + nprocs + energy
const int PRNGSIZE = 98+2+3;
int nsize = PRNGSIZE*comm->nprocs+2; // pRNG state per proc + nprocs + energy
double *list = nullptr;
if (comm->me == 0) {
list = new double[nsize];
list[0] = energy;
list[1] = comm->nprocs;
}
double state[103];
double state[PRNGSIZE];
random->get_state(state);
MPI_Gather(state,103,MPI_DOUBLE,list+2,103,MPI_DOUBLE,0,world);
MPI_Gather(state,PRNGSIZE,MPI_DOUBLE,list+2,PRNGSIZE,MPI_DOUBLE,0,world);
if (comm->me == 0) {
int size = nsize * sizeof(double);

7
src/fix_temp_csvr.cpp
View File

@ -338,16 +338,17 @@ double FixTempCSVR::compute_scalar()
void FixTempCSVR::write_restart(FILE *fp)
{
int nsize = (98+2+3)*comm->nprocs+2; // pRNG state per proc + nprocs + energy
const int PRNGSIZE = 98+2+3;
int nsize = PRNGSIZE*comm->nprocs+2; // pRNG state per proc + nprocs + energy
double *list = nullptr;
if (comm->me == 0) {
list = new double[nsize];
list[0] = energy;
list[1] = comm->nprocs;
}
double state[103];
double state[PRNGSIZE];
random->get_state(state);
MPI_Gather(state,103,MPI_DOUBLE,list+2,103,MPI_DOUBLE,0,world);
MPI_Gather(state,PRNGSIZE,MPI_DOUBLE,list+2,PRNGSIZE,MPI_DOUBLE,0,world);
if (comm->me == 0) {
int size = nsize * sizeof(double);