Revision 5362a7d076236863b176c4c0a356ab3890c11563 authored by Thomas Quinn on 30 September 2012, 01:55:31 UTC, committed by Tom Quinn on 02 September 2016, 21:43:22 UTC
Change-Id: I2ddfdcf35486094e0bb2cf44a6d19d54e0339292
1 parent 76a59c6
MultistepNodeLB_notopo.h
/*****************************************************************************
* $Source$
* $Author$
* $Date$
* $Revision$
*****************************************************************************/
/**
* \addtogroup CkLdb
*/
/*@{*/
#ifndef _MULTISTEPNODELB_NOTOPO_H_
#define _MULTISTEPNODELB_NOTOPO_H_
#define MCLBMS // multistepping enabled
#define MCLB_ORBSMOOTH // orbsmooth for large steps
#define MCLB_RR // round robin otherwise
#include "Orb3dLBCommon.h"
#include "MultistepNodeLB_notopo.decl.h"
void CreateMultistepLB_notopo();
BaseLB * AllocateMultistepNodeLB_notopo();
/// @brief Multistep load balancer where no processor topology
/// information is used. This first performs orb partition at the node level and
/// assigns the partitioned TreePieces to the PEs belonging to the node. Finally
/// after this assignment, a refinement is done.
class MultistepNodeLB_notopo : public CBase_MultistepNodeLB_notopo, public Orb3dCommon {
private:
int prevMaxPredPe;
void init();
bool QueryBalanceNow(int step);
public:
MultistepNodeLB_notopo(const CkLBOptions &);
MultistepNodeLB_notopo(CkMigrateMessage *m) : CBase_MultistepNodeLB_notopo(m) {
init();
}
void work(BaseLB::LDStats* stats);
void balanceTPs(BaseLB::LDStats* stats);
void balanceTPsNode(BaseLB::LDStats* stats);
public:/* <- Sun CC demands Partition be public for ComputeLoad to access it. */
class Partition {
public:
int refno;
double load; // total load in this set
int origin[3]; // box coordinates
int corner[3];
int count; // number of objects in this partition
int node, mapped;
CkVec<int> bkpes; // background processors
public:
Partition(): refno(0), load(0.0), node(-1), mapped(0) {};
};
private:
struct ComputeLoad {
int id;
int v[3];
double load;
int refno;
double tv;
Partition * partition;
};
struct VecArray {
int v;
int id;
};
enum {XDIR=0, YDIR, ZDIR};
public:
//**************************************
// ORB3DLB functions
//**************************************
//
void work2(BaseLB::LDStats* stats, int count);
void greedy(BaseLB::LDStats* stats, int count);
void pup(PUP::er &p);
};
#endif /* _MultistepNodeLB_notopo */
/*@}*/
Computing file changes ...