src/ck-ldb/NodeLevelLB.h

   1 /**
   2  * Author Harshitha Menon (gplkrsh2@illinois.edu)
   3 */
   4
   5 #ifndef NODE_LEVEL_LB_H
   6 #define NODE_LEVEL_LB_H
   7
   8 #include "CentralLB.h"
   9 #include "HybridBaseLB.h"
  10 #include "NodeLevelLB.decl.h"
  11
  12 void CreateNodeLevelLB();
  13
  14 class NodeLevelLB : public CBase_NodeLevelLB {
  15   private:
  16     CkVec<CentralLB *>  clbs;
  17     int num_levels;
  18
  19   protected:
  20     virtual bool QueryBalanceNow(int) { return true; };
  21     virtual void work(LDStats* stats);
  22
  23   public:
  24     NodeLevelLB(const CkLBOptions &);
  25     NodeLevelLB(CkMigrateMessage *m):CBase_NodeLevelLB(m) {}
  26 };
  27
  28 #endif /* NODE_LEVEL_LB_H */