ck-ldb: count in the struct LdStats changed to n_pes
authorAbhinav S Bhatele <bhatele@illinois.edu>
Sun, 24 Oct 2010 00:40:07 +0000 (19:40 -0500)
committerAbhinav S Bhatele <bhatele@illinois.edu>
Sat, 6 Nov 2010 22:50:14 +0000 (17:50 -0500)
15 files changed:
src/ck-ldb/BaseLB.C
src/ck-ldb/BaseLB.h
src/ck-ldb/CentralLB.C
src/ck-ldb/CommLB.C
src/ck-ldb/GreedyAgentLB.C
src/ck-ldb/GreedyCommLB.C
src/ck-ldb/HybridBaseLB.C
src/ck-ldb/MetisLB.C
src/ck-ldb/OrbLB.C
src/ck-ldb/PhasebyArrayLB.C
src/ck-ldb/RandCentLB.C
src/ck-ldb/RecBisectBfLB.C
src/ck-ldb/RefineCommLB.C
src/ck-ldb/RefineKLB.C
src/ck-ldb/RefineLB.C

index 6ce3c28996bc6ca9bd7705fccd019cba1677a8d9..e0af997d53db2b547229776dec752d89f4cd90cf 100644 (file)
@@ -79,9 +79,9 @@ BaseLB::LDStats::LDStats(int c, int complete)
        : n_objs(0), n_migrateobjs(0), n_comm(0), 
           objHash(NULL), complete_flag(complete)
 {
-  count = c;
-  if (count == 0) count = CkNumPes();
-  procs = new ProcStats[count]; 
+  n_pes = c;
+  if (n_pes == 0) n_pes = CkNumPes();
+  procs = new ProcStats[n_pes];
 }
 
 const static unsigned int doublingPrimes[] = {
@@ -251,7 +251,7 @@ void BaseLB::LDStats::computeNonlocalComm(int &nmsgs, int &nbytes)
              senderPE = to_proc[idx];
              CmiAssert(senderPE != -1);
            }
-           CmiAssert(senderPE < count && senderPE >= 0);
+           CmiAssert(senderPE < n_pes && senderPE >= 0);
 
             // find receiver: point-to-point and multicast two cases
            int receiver_type = cdata.receiver.get_type();
@@ -266,7 +266,7 @@ void BaseLB::LDStats::computeNonlocalComm(int &nmsgs, int &nbytes)
                }
                else {
                  receiverPE = to_proc[idx];
-                  CmiAssert(receiverPE < count && receiverPE >= 0);
+                  CmiAssert(receiverPE < n_pes && receiverPE >= 0);
                }
               }
              if(senderPE != receiverPE)
@@ -285,7 +285,7 @@ void BaseLB::LDStats::computeNonlocalComm(int &nmsgs, int &nbytes)
                CmiAssert(idx != -1);
                if (idx == -1) continue;    // receiver has just been removed?
                receiverPE = to_proc[idx];
-               CmiAssert(receiverPE < count && receiverPE >= 0);
+               CmiAssert(receiverPE < n_pes && receiverPE >= 0);
                int exist = 0;
                for (int p=0; p<pes.size(); p++) 
                  if (receiverPE == pes[p]) { exist=1; break; }
@@ -306,8 +306,8 @@ void BaseLB::LDStats::print()
 {
 #if CMK_LBDB_ON
   int i;
-  CkPrintf("------------- Processor Data: %d -------------\n", count);
-  for(int pe=0; pe < count; pe++) {
+  CkPrintf("------------- Processor Data: %d -------------\n", n_pes);
+  for(int pe=0; pe < n_pes; pe++) {
     struct ProcStats &proc = procs[pe];
 
     CkPrintf(
@@ -362,7 +362,7 @@ double BaseLB::LDStats::computeAverageLoad()
   double total = 0;
   for (i=0; i<n_objs; i++) total += objData[i].wallTime;
                                                                                 
-  for (i=0; i<count; i++)
+  for (i=0; i<n_pes; i++)
     if (procs[i].available == CmiTrue) {
         total += procs[i].bg_walltime;
        numAvail++;
@@ -403,13 +403,13 @@ void BaseLB::LDStats::removeObject(int obj)
 void BaseLB::LDStats::pup(PUP::er &p)
 {
   int i;
-  p(count);  
+  p(n_pes);
   p(n_objs);
   p(n_migrateobjs);
   p(n_comm);
   if (p.isUnpacking()) {
     // user can specify simulated processors other than the real # of procs.
-    int maxpe = count>LBSimulation::simProcs?count:LBSimulation::simProcs;
+    int maxpe = n_pes > LBSimulation::simProcs ? n_pes : LBSimulation::simProcs;
     procs = new ProcStats[maxpe];
     objData.resize(n_objs);
     commData.resize(n_comm);
@@ -421,10 +421,10 @@ void BaseLB::LDStats::pup(PUP::er &p)
   // otherwise load everything
   if (p.isUnpacking() && LBSimulation::procsChanged) {
     ProcStats dummy;
-    for (i=0; i<count; i++) p|dummy; 
+    for (i=0; i<n_pes; i++) p|dummy;
   }
   else
-    for (i=0; i<count; i++) p|procs[i];
+    for (i=0; i<n_pes; i++) p|procs[i];
   for (i=0; i<n_objs; i++) p|objData[i]; 
   for (i=0; i<n_objs; i++) p|from_proc[i]; 
   for (i=0; i<n_objs; i++) p|to_proc[i]; 
@@ -433,18 +433,18 @@ void BaseLB::LDStats::pup(PUP::er &p)
     for (i=0; i<n_objs; i++) to_proc[i] = from_proc[i];
   for (i=0; i<n_comm; i++) p|commData[i];
   if (p.isUnpacking())
-    count = LBSimulation::simProcs;
+    n_pes = LBSimulation::simProcs;
   if (p.isUnpacking()) {
     objHash = NULL;
     if (_lb_args.lbversion() <= 1) 
-      for (i=0; i<count; i++) procs[i].pe = i;
+      for (i=0; i<n_pes; i++) procs[i].pe = i;
   }
 }
 
 int BaseLB::LDStats::useMem() { 
   // calculate the memory usage of this LB (superclass).
-  return sizeof(LDStats) + sizeof(ProcStats)*count + 
-        (sizeof(LDObjData) + 2*sizeof(int)) * n_objs +
+  return sizeof(LDStats) + sizeof(ProcStats) * n_pes +
+        (sizeof(LDObjData) + 2 * sizeof(int)) * n_objs +
         sizeof(LDCommData) * n_comm;
 }
 
index 41db7f0b932385fc8ffd1f36e4458d6532705068..fdc02b7275200a4e619fbe61044a3a7fba0cd477 100644 (file)
@@ -68,15 +68,16 @@ public:
   };
 
   struct LDStats {  // Passed to Strategy
-    ProcStats  *procs;
-    int count; 
-    
-    int   n_objs;
-    int   n_migrateobjs;
+    int n_pes;         // number of pes in the array procs
+    ProcStats *procs;  // array of pes
+
+    int n_objs;                // number of objects in the database
+    int n_migrateobjs; // number of migratable objects
     CkVec<LDObjData> objData;
-    int   n_comm;
+
+    int n_comm;                // number of edges in the vector commData
     CkVec<LDCommData> commData;
-    CkVec<int>  from_proc, to_proc;
+    CkVec<int> from_proc, to_proc;
 
     int *objHash; 
     int  hashSize;
@@ -103,7 +104,7 @@ public:
       deleteCommHash();
     }
     void clearBgLoad() {
-      for (int i=0; i<count; i++) procs[i].clearBgLoad();
+      for (int i=0; i<n_pes; i++) procs[i].clearBgLoad();
     }
     void computeNonlocalComm(int &nmsgs, int &nbytes);
     double computeAverageLoad();
index 9787ba23ab936f625affff85d4ebbb295821e597..3f8d77b44bf244ffe86b38512428565fdefe79a7 100644 (file)
@@ -366,7 +366,7 @@ void CentralLB::MissMigrate(int waitForBarrier)
 // not used when USE_REDUCTION = 1
 void CentralLB::buildStats()
 {
-    statsData->count = stats_msg_count;
+    statsData->n_pes = stats_msg_count;
     // allocate space
     statsData->objData.resize(statsData->n_objs);
     statsData->from_proc.resize(statsData->n_objs);
@@ -507,7 +507,7 @@ void CentralLB::ReceiveStats(CkMarshalledCLBStatsMessage &msg)
  
   if (stats_msg_count == clients) {
        DEBUGF(("[%d] All stats messages received \n",CmiMyPe()));
-    statsData->count = stats_msg_count;
+    statsData->n_pes = stats_msg_count;
     thisProxy[CkMyPe()].LoadBalance();
   }
 #endif
index 74d1cda66d54b9d1e4d80dfb84482d6713ed2dbe..0c6ead750fb002af25030462e02534f2064a0ca8 100644 (file)
@@ -144,7 +144,7 @@ void CommLB::work(LDStats* stats)
   //  CkPrintf("[%d] CommLB strategy\n",CkMyPe());
 
   nobj = stats->n_objs;
-  npe = stats->count;
+  npe = stats->n_pes;
 
   stats->makeCommHash();
 
index fcea08b070f93e223dfc7aa3dda3fdc6338ad61d..4655140eb8e4432425a7a68a956fa40d75f384f5 100644 (file)
@@ -195,7 +195,7 @@ GreedyAgentLB::BuildCpuArray(CentralLB::LDStats* stats,
 void GreedyAgentLB::work(LDStats* stats)
 {
   int  i, obj, heapSize, objCount;
-  int n_pes = stats->count;
+  int n_pes = stats->n_pes;
 
   int *pemap = new int [n_pes];
   HeapData *cpuData = BuildCpuArray(stats, n_pes, &heapSize);
index ce6d704bc2be3acb5cbd81361efd5204b2d65ed6..0d55d5192af1f0011eb421bbb76d13bbe8d6564d 100644 (file)
@@ -142,7 +142,7 @@ void GreedyCommLB::work(LDStats* stats)
     int i;
     
     if (_lb_args.debug()) CkPrintf("In GreedyCommLB strategy\n",CkMyPe());
-    npe = stats->count;
+    npe = stats->n_pes;
     nobj = stats->n_objs;
 
     // nmigobj is calculated as the number of migratable objects
index 776f61ae65d432fc1e9919fc81af35e275fab258..86f5a9ee6af0a5f4a490e9dd935d7dba3fd75dd7 100644 (file)
@@ -312,7 +312,7 @@ void HybridBaseLB::buildStats(int atlevel)
 
   // statsMsgsList
   DEBUGF(("[%d] buildStats for %d nobj:%d\n", CkMyPe(), stats_msg_count, statsData->n_objs));
-  statsData->count = stats_msg_count;
+  statsData->n_pes = stats_msg_count;
   statsData->objData.resize(statsData->n_objs);
   statsData->commData.resize(statsData->n_comm);
   statsData->from_proc.resize(statsData->n_objs);
@@ -397,7 +397,7 @@ CLBStatsMsg * HybridBaseLB::buildCombinedLBStatsMessage(int atlevel)
   cmsg->bg_walltime = 0.0;
   cmsg->bg_cputime = 0.0;
 
-  for (int pe=0; pe<statsData->count; pe++) {
+  for (int pe=0; pe<statsData->n_pes; pe++) {
         struct ProcStats &procStat = statsData->procs[pe];
         cmsg->pe_speed += procStat.pe_speed;           // important
         cmsg->total_walltime += procStat.total_walltime;
index b2178401a86bb2ed26fd55c254a884d6234377ba..6814e0e937d40f14a020c5b2f38c14b69c90fced 100644 (file)
@@ -94,7 +94,7 @@ void MetisLB::work(LDStats* stats)
 
   stats->makeCommHash();
 
-  int n_pes = stats->count;
+  int n_pes = stats->n_pes;
   int numobjs = stats->n_objs;
 
   removeNonMigratable(stats, n_pes);
index d1bc324e516459137c808a3d29ba7fc401692bf3..3199a9ddabe1f8cfd07544f2de133f503dd5dfb9 100644 (file)
@@ -311,7 +311,7 @@ void OrbLB::work(LDStats* stats)
 
   statsData = stats;
 
-  P = stats->count;
+  P = stats->n_pes;
 
   // calculate total number of migratable objects
   nObjs = stats->n_migrateobjs;
index 7541971b455b4b7b3978284d4394d8371bd5a6c6..500774bf1d6e86b566e3cdf62348839ae305136a 100644 (file)
@@ -47,7 +47,7 @@ CmiBool PhasebyArrayLB::QueryBalanceNow(int _step)
 
 void PhasebyArrayLB::copyStats(BaseLB::LDStats *stats,BaseLB::LDStats *tempStats){
        int i;
-       tempStats->count = stats->count;
+       tempStats->n_pes = stats->n_pes;
        tempStats->n_objs = stats->n_objs;
        tempStats->n_comm = stats->n_comm;
        tempStats->n_migrateobjs = stats->n_migrateobjs;
@@ -71,8 +71,8 @@ void PhasebyArrayLB::copyStats(BaseLB::LDStats *stats,BaseLB::LDStats *tempStats
        for(i=0;i<tempStats->n_comm;i++)
                tempStats->commData[i]=stats->commData[i];
        
-       tempStats->procs = new BaseLB::ProcStats[tempStats->count];
-       for(i=0;i<tempStats->count;i++)
+       tempStats->procs = new BaseLB::ProcStats[tempStats->n_pes];
+       for(i=0; i<tempStats->n_pes; i++)
                tempStats->procs[i]=stats->procs[i];
 }
 
index 70526ddfd9f3c425db5485f436c9632ede812c0d..db44a1c7e891844aae332fa6172912ec2e61f54e 100644 (file)
@@ -45,7 +45,7 @@ void RandCentLB::work(LDStats* stats)
 {
   if (_lb_args.debug()) CkPrintf("Calling RandCentLB strategy\n",CkMyPe());
 
-  int proc, n_pes = stats->count;
+  int proc, n_pes = stats->n_pes;
 
   for (proc=0; proc<n_pes; proc++) {
     if (stats->procs[proc].available) break;
index 798a743820a5100bc04bb55e81ffb06162effe88..0d74c9be583a1cd0483cca231f7d2846363ccb7f 100644 (file)
@@ -64,7 +64,7 @@ CmiBool RecBisectBfLB::QueryBalanceNow(int _step)
 void RecBisectBfLB::work(LDStats* stats)
 {
   int i;
-  int numPartitions = stats->count;
+  int numPartitions = stats->n_pes;
 
   PartitionList *partitions;
 
index 499a2ef4e890ff6665cc1c529ae258a75e31db04..6e7da8dbf202f1bea409b9bb5d6f9f19618a2004 100644 (file)
@@ -34,7 +34,7 @@ void RefineCommLB::work(LDStats* stats)
 {
 #if CMK_LBDB_ON
   int obj;
-  int n_pes = stats->count;
+  int n_pes = stats->n_pes;
 
   //  CkPrintf("[%d] RefineLB strategy\n",CkMyPe());
 
index 2c4acb4fdba0e586e86a3f3a4cf7295e26cf83ca..d73e87cde3a3277c67f7611e843212596e15d2f5 100644 (file)
@@ -31,7 +31,7 @@ RefineKLB::RefineKLB(const CkLBOptions &opt): CentralLB(opt)
 void RefineKLB::work(LDStats* stats)
 {
   int obj;
-  int n_pes = stats->count;
+  int n_pes = stats->n_pes;
 
   //  CkPrintf("[%d] RefineKLB strategy\n",CkMyPe());
 
index 2d9eb54f5488ac1d56088ed2ebaf3e2291f98488..b293934c5db9ba2e2d2b6d143be4f1a20b7d55f9 100644 (file)
@@ -28,7 +28,7 @@ RefineLB::RefineLB(const CkLBOptions &opt): CentralLB(opt)
 void RefineLB::work(LDStats* stats)
 {
   int obj;
-  int n_pes = stats->count;
+  int n_pes = stats->n_pes;
 
   //  CkPrintf("[%d] RefineLB strategy\n",CkMyPe());