NEM_SLICE: Clean up 1D decomp --handle edges correctly

sandialabs · Jul 24, 2024 · bcf3aed · bcf3aed
1 parent 594ec93
commit bcf3aed
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/packages/seacas/applications/nem_slice/elb.h b/packages/seacas/applications/nem_slice/elb.h
@@ -14,7 +14,7 @@
 #include <string>
 #include <vector>
 
-#define ELB_VERSION "5.00"
+#define ELB_VERSION "5.01"
 #define UTIL_NAME   "nem_slice"
 #define ELB_FALSE   0
 #define ELB_TRUE    1

diff --git a/packages/seacas/applications/nem_slice/elb_loadbal.C b/packages/seacas/applications/nem_slice/elb_loadbal.C
@@ -1556,12 +1556,18 @@ namespace {
 	  lb->bor_elems[procc].push_back(ecnt);
 
 	  if (procl != procc) {
+	    if (ecnt == 1) { // At left end...
+	      lb->bor_elems[procl].push_back(ecnt-1);
+	    }
 	    lb->e_cmap_elems[procl].push_back(ecnt-1);
 	    lb->e_cmap_sides[procl].push_back(2);
 	    lb->e_cmap_procs[procl].push_back(procc);
 	    lb->e_cmap_neigh[procl].push_back(ecnt);
 	  }
 	  if (procc != procr) {
+	    if (ecnt == mesh->num_elems-2) { // At right end...
+	      lb->bor_elems[procr].push_back(ecnt+1);
+	    }
 	    lb->e_cmap_elems[procr].push_back(ecnt+1);
 	    lb->e_cmap_sides[procr].push_back(1);
 	    lb->e_cmap_procs[procr].push_back(procc);