Set the proc field to correct value

MalachiTimothyPhillips · Feb 12, 2019 · b3cfa5f · b3cfa5f
1 parent 0168ced
commit b3cfa5f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/src/genmap-impl.h b/src/genmap-impl.h
@@ -35,7 +35,6 @@ struct GenmapElement_private {
   GenmapLong globalId;
   GenmapLong vertices[8];
   GenmapInt proc;
-  GenmapInt procGlobal;
   GenmapInt origin;
 };
 //

diff --git a/src/genmap-rsb.c b/src/genmap-rsb.c
@@ -59,7 +59,6 @@ GenmapInt GenmapSetFiedlerBin(GenmapHandle h) {
       GenmapScalar end = min + (range * (id + 1)) / nbins;
       if(start <= p->fiedler && p->fiedler < end) {
         p->proc = id;
-        p->procGlobal = GenmapCommRank(GenmapGetGlobalComm(h));
         break;
       }
     }
@@ -87,7 +86,6 @@ GenmapInt GenmapSetGlobalIdBin(GenmapHandle h) {
       GenmapScalar end = min + (range * (id + 1)) / nbins;
       if(start <= p->globalId && p->globalId < end) {
         p->proc = id;
-        p->procGlobal = GenmapCommRank(GenmapGetGlobalComm(h));
         break;
       }
     }

diff --git a/src/parRSB.c b/src/parRSB.c
@@ -55,6 +55,11 @@ int parRSB_partMesh(int *part, long long *vtx, int nel, int nve, int *options,
 
   GenmapRSB(h);
 
+  e = GenmapGetElements(h);
+  for(j = 0; j < GenmapGetNLocalElements(h); j++) {
+    e[j].proc = GenmapCommRank(GenmapGetGlobalComm(h));
+  }
+
   GenmapCrystalInit(h, GenmapGetGlobalComm(h));
   GenmapCrystalTransfer(h, GENMAP_ORIGIN);
   GenmapCrystalFinalize(h);
@@ -68,7 +73,7 @@ int parRSB_partMesh(int *part, long long *vtx, int nel, int nve, int *options,
   buffer_free(&buf);
 
   for(i = 0; i < nel; i++) {
-    part[i] = e[i].procGlobal;
+    part[i] = e[i].proc;
   }
 
   if(id == 0 && h->dbgLevel > 0)