src/ck-ldb/TempAwareGreedyLB.h

   1 /**
   2  * \addtogroup CkLdb
   3 */
   4 /*@{*/
   5
   6 #ifndef _GREEDYLB_H_
   7 #define _GREEDYLB_H_
   8
   9 #include "CentralLB.h"
  10 #include "TempAwareGreedyLB.decl.h"
  11
  12 void CreateTempAwareGreedyLB();
  13 BaseLB * AllocateTempAwareGreedyLB();
  14
  15 class TempAwareGreedyLB : public CBase_TempAwareGreedyLB {
  16
  17 public:
  18   struct HeapData {
  19     double load;
  20     int    pe;
  21     int    id;
  22   };
  23
  24   TempAwareGreedyLB(const CkLBOptions &);
  25   TempAwareGreedyLB(CkMigrateMessage *m) : CBase_TempAwareGreedyLB(m) { lbname = "GreedyLB"; }
  26   void work(LDStats* stats);
  27 private:
  28   class ProcLoadGreater;
  29   class ObjLoadGreater;
  30
  31         enum           HeapCmp {GT = '>', LT = '<'};
  32         void           Heapify(HeapData*, int, int, HeapCmp);
  33         void           HeapSort(HeapData*, int, HeapCmp);
  34         void           BuildHeap(HeapData*, int, HeapCmp);
  35         bool        Compare(double, double, HeapCmp);
  36         HeapData*      BuildCpuArray(BaseLB::LDStats*, int, int *);
  37         HeapData*      BuildObjectArray(BaseLB::LDStats*, int, int *);
  38         bool        QueryBalanceNow(int step);
  39 };
  40
  41 #endif /* _HEAPCENTLB_H_ */
  42
  43 /*@}*/