src/ck-ldb/TempAwareCommLB.h

   1
   2 #ifndef _COMMAWARELB_H_
   3 #define _COMMAWARELB_H_
   4
   5 #include "CentralLB.h"
   6 #include "TempAwareCommLB.decl.h"
   7
   8 void CreateTempAwareCommLB();
   9 BaseLB * AllocateTempAwareCommLB();
  10
  11 class TempAwareCommLB : public CBase_TempAwareCommLB {
  12 friend void printCurrentTemperature(void *LB, double curWallTime);
  13 public:
  14   struct HeapData {
  15     double load;
  16     int    pe;
  17     int    id;
  18
  19   };
  20         void populateEffectiveFreq(int numProcs);
  21         void convertToInsts();
  22   int procsPerNode,*freqsEffect,*procFreq,*procFreqEffect,*procFreqNewEffect,*procFreqNew,numProcs,coresPerChip,*freqs,numAvailFreqs;
  23   int numChips,*procFreqPtr;
  24         void initStructs(LDStats *s);
  25         void tempControl();
  26         FILE *migFile;
  27         double starting;
  28
  29   float *procTemp,*avgChipTemp;
  30   TempAwareCommLB(const CkLBOptions &);
  31   TempAwareCommLB(CkMigrateMessage *m):CBase_TempAwareCommLB(m) {
  32     lbname = "TempAwareCommLB";
  33   }
  34   void work(LDStats* stats);
  35
  36   class ProcLoadGreater;
  37   class ObjLoadGreater;
  38   class PeCommInfo;
  39   class ObjPeCommInfo;
  40   class ProcCommGreater;
  41
  42 private:
  43
  44         enum           HeapCmp {GT = '>', LT = '<'};
  45         void           Heapify(HeapData*, int, int, HeapCmp);
  46         void           HeapSort(HeapData*, int, HeapCmp);
  47         void           BuildHeap(HeapData*, int, HeapCmp);
  48         bool        Compare(double, double, HeapCmp);
  49         HeapData*      BuildCpuArray(BaseLB::LDStats*, int, int *);
  50         HeapData*      BuildObjectArray(BaseLB::LDStats*, int, int *);
  51         bool        QueryBalanceNow(int step);
  52 };
  53
  54 #endif /* _COMMAWARELB_H_ */
  55
  56 /*@}*/