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

Fix implementation of absmax

Browse Source
This commit is contained in:
Tim Bernhard
2023-04-12 09:26:33 +02:00
parent 7a32152ee5
commit 4e33b1b867
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/compute_reduce.cpp
View File

@ -400,11 +400,13 @@ void ComputeReduce::compute_vector()
if (mode == SUM || mode == SUMSQ || mode == AVEABS) { if (mode == SUM || mode == SUMSQ || mode == AVEABS) {
for (int m = 0; m < nvalues; m++) for (int m = 0; m < nvalues; m++)
MPI_Allreduce(&onevec[m], &vector[m], 1, MPI_DOUBLE, MPI_SUM, world); MPI_Allreduce(&onevec[m], &vector[m], 1, MPI_DOUBLE, MPI_SUM, world);
} else if (mode == MINABS || mode == MAXABS) {
for (int m = 0; m < nvalues; m++)
MPI_Allreduce(&onevec[m], &vector[m], 1, MPI_DOUBLE, this->scalar_reduction_operation, world);
} else if (mode == MINN) { } else if (mode == MINN) {
if (!replace) { if (!replace) {
for (int m = 0; m < nvalues; m++) for (int m = 0; m < nvalues; m++)
MPI_Allreduce(&onevec[m], &vector[m], 1, MPI_DOUBLE, MPI_MIN, world); MPI_Allreduce(&onevec[m], &vector[m], 1, MPI_DOUBLE, this->scalar_reduction_operation, world);
} else { } else {
for (int m = 0; m < nvalues; m++) for (int m = 0; m < nvalues; m++)
@ -421,11 +423,10 @@ void ComputeReduce::compute_vector()
MPI_Bcast(&vector[m], 1, MPI_DOUBLE, owner[replace[m]], world); MPI_Bcast(&vector[m], 1, MPI_DOUBLE, owner[replace[m]], world);
} }
} }
} else if (mode == MAXX) { } else if (mode == MAXX) {
if (!replace) { if (!replace) {
for (int m = 0; m < nvalues; m++) for (int m = 0; m < nvalues; m++)
MPI_Allreduce(&onevec[m], &vector[m], 1, MPI_DOUBLE, MPI_MAX, world); MPI_Allreduce(&onevec[m], &vector[m], 1, MPI_DOUBLE, this->scalar_reduction_operation, world);
} else { } else {
for (int m = 0; m < nvalues; m++) for (int m = 0; m < nvalues; m++)