src/ck-ldb/TeamLB.h

   1 /**
   2  * \addtogroup CkLdb
   3 */
   4 /*@{*/
   5
   6 #ifndef _TEAMLB_H_
   7 #define _TEAMLB_H_
   8
   9 #include "CentralLB.h"
  10 #include "TeamLB.decl.h"
  11 #include <metis.h>
  12
  13 void CreateTeamLB();
  14 BaseLB * AllocateTeamLB();
  15
  16 class TeamLB : public CBase_TeamLB {
  17   public:
  18     TeamLB(const CkLBOptions &);
  19     TeamLB(CkMigrateMessage *m):CBase_TeamLB(m) { lbname = "TeamLB"; }
  20
  21     void work(LDStats* stats);
  22     void pup(PUP::er &p) {  }
  23
  24   private:
  25     idx_t teamSize;
  26     idx_t numberTeams;
  27
  28     bool QueryBalanceNow(int step) { return true; }
  29 };
  30
  31 #endif /* _TEAMLB_H_ */
  32
  33 /*@}*/