Put serial case back in for DataSet

OP2 · Jan 25, 2024 · 26f2e51 · 26f2e51
1 parent 2b1e79f
commit 26f2e51
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pyop2/types/dataset.py b/pyop2/types/dataset.py
@@ -113,8 +113,12 @@ def lgmap(self):
         indices for this :class:`DataSet`.
         """
         lgmap = PETSc.LGMap()
-        lgmap.create(indices=self.halo.local_to_global_numbering,
-                     bsize=self.cdim, comm=self.comm)
+        if self.comm.size == 1 and self.halo is None:
+            lgmap.create(indices=np.arange(self.size, dtype=dtypes.IntType),
+                         bsize=self.cdim, comm=self.comm)
+        else:
+            lgmap.create(indices=self.halo.local_to_global_numbering,
+                         bsize=self.cdim, comm=self.comm)
         return lgmap
 
     @utils.cached_property