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

git-svn-id: svn://svn.icms.temple.edu/lammps-ro/trunk@8370 f3b2605a-c512-4ea7-a41b-209d697bcdaa

Browse Source
This commit is contained in:
sjplimp
2012-06-21 17:19:47 +00:00
parent 79a7e57328
commit 83416310f2
2 changed files with 28 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
src/fix_balance.cpp
View File

@ -101,6 +101,7 @@ FixBalance::FixBalance(LAMMPS *lmp, int narg, char **arg) :
imbfinal = imbprev = balance->imbalance_nlocal(maxperproc);
itercount = 0;
pending = 0;
}
/* ---------------------------------------------------------------------- */
@ -118,6 +119,7 @@ int FixBalance::setmask()
{
int mask = 0;
mask |= PRE_EXCHANGE;
mask |= PRE_NEIGHBOR;
return mask;
}
@ -133,6 +135,13 @@ void FixBalance::init()
/* ---------------------------------------------------------------------- */
void FixBalance::setup(int vflag)
{
pre_neighbor();
}
/* ---------------------------------------------------------------------- */
void FixBalance::setup_pre_exchange()
{
// insure atoms are in current box & update box via shrink-wrap
@ -186,6 +195,18 @@ void FixBalance::pre_exchange()
if (nevery) next_reneighbor = (update->ntimestep/nevery)*nevery + nevery;
}
/* ----------------------------------------------------------------------
compute final imbalance factor based on nlocal after comm->exchange()
only do this if rebalancing just occured
------------------------------------------------------------------------- */
void FixBalance::pre_neighbor()
{
if (!pending) return;
imbfinal = balance->imbalance_nlocal(maxperproc);
pending = 0;
}
/* ----------------------------------------------------------------------
perform dynamic load balancing
------------------------------------------------------------------------- */
@ -211,7 +232,7 @@ void FixBalance::rebalance()
// if splits moved further than neighboring processor
// move atoms to new processors via irregular()
// only needed if migrate_check() says an atom moves to far,
// else allow comm->exchange that follows in caller to do it
// else allow caller's comm->exchange() to do it
if (domain->triclinic) domain->x2lamda(atom->nlocal);
if (irregular->migrate_check()) irregular->migrate_atoms();
@ -221,9 +242,10 @@ void FixBalance::rebalance()
// check that new sub-domains are valid with KSpace constraints
// if (kspace_flag) force->kspace->check();
// imbfinal = final imbalance based on final nlocal
// pending triggers pre_neighbor() to compute final imbalance factor
// can only be done after atoms migrate in caller's comm->exchange()
imbfinal = balance->imbalance_nlocal(maxperproc);
pending = 1;
}
/* ----------------------------------------------------------------------