Skip to content
Snippets Groups Projects
Commit a71b9469 authored by Peter W. Draper's avatar Peter W. Draper
Browse files

Merge branch 'mpi_fixes' into 'master'

mpi_fixes

Fixed broken scaling in task/resource graph in engine_repartition, was causing load imbalances across MPI nodes (see #34).

See merge request !22


Former-commit-id: 07836a868ad9b27b874c0a813029021562ea7521
parents 3bad3c60 a34560b0
No related branches found
No related tags found
No related merge requests found
......@@ -443,7 +443,7 @@ void engine_repartition ( struct engine *e ) {
error( "Failed to allreduce the weight scales (%s)." , buff );
}
if (wscale_buff != wscale) {
float scale = wscale / wscale_buff;
float scale = wscale_buff / wscale;
for (k = 0; k < 26 * nr_cells; k++) weights_e[k] *= scale;
for (k = 0; k < nr_cells; k++) weights_v[k] *= scale;
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment