Account for matrix ownership range

Addresses usnistgov#644
guyer · Sep 24, 2019 · 34244e0 · 34244e0
1 parent 8ce1ab2
commit 34244e0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/fipy/matrices/petscMatrix.py b/fipy/matrices/petscMatrix.py
@@ -250,11 +250,12 @@ def _ijv2csr(self, i, j, v):
         i = numerix.asarray(i)
         j = numerix.asarray(j)
         v = numerix.asarray(v)
-        N = self._shape[1]
+        start_row, end_row = self.matrix.getOwnershipRange()
 
         ix = numerix.lexsort([i, j])
         cols = i[ix]
-        row_ptr = numerix.searchsorted(j[ix], numerix.arange(N + 1))
+        row_ptr = numerix.searchsorted(j[ix], 
+                                       numerix.arange(start_row, end_row + 1))
         vals = v[ix]
         # note: PETSc (at least via pip) only seems to handle 32 bit addressing
         return row_ptr.astype('int32'), cols.astype('int32'), vals