206 static PetscErrorCode MatMPIAIJDiagonalScaleLocalSetUp(Mat inA, Vec scale)  in MatMPIAIJDiagonalScaleLocalSetUp()  argument
208   Mat_MPIAIJ *ina = (Mat_MPIAIJ *)inA->data; /*access private part of matrix */  in MatMPIAIJDiagonalScaleLocalSetUp()
209   PetscInt    i, j, n, nt, cstart, cend, no, *garray = ina->garray, *lindices, bs = inA->rmap->bs;  in MatMPIAIJDiagonalScaleLocalSetUp()
213   PetscCall(MatGetOwnershipRange(inA, &cstart, &cend));  in MatMPIAIJDiagonalScaleLocalSetUp()
215   PetscCall(PetscCalloc1(inA->rmap->mapping->n, &r_rmapd));  in MatMPIAIJDiagonalScaleLocalSetUp()
217   for (i = 0; i < inA->rmap->mapping->n; i++) {  in MatMPIAIJDiagonalScaleLocalSetUp()
218 … if (inA->rmap->mapping->indices[i] * bs >= cstart && inA->rmap->mapping->indices[i] * bs < cend) {  in MatMPIAIJDiagonalScaleLocalSetUp()
220       r_rmapd[i] = inA->rmap->mapping->indices[i] + 1;  in MatMPIAIJDiagonalScaleLocalSetUp()
225   for (i = 0; i < inA->rmap->mapping->n; i++) {  in MatMPIAIJDiagonalScaleLocalSetUp()
237   PetscCall(PetscCalloc1(inA->cmap->N / bs, &lindices));  in MatMPIAIJDiagonalScaleLocalSetUp()
239   no = inA->rmap->mapping->n - nt;  in MatMPIAIJDiagonalScaleLocalSetUp()
240   PetscCall(PetscCalloc1(inA->rmap->mapping->n, &r_rmapo));  in MatMPIAIJDiagonalScaleLocalSetUp()
242   for (i = 0; i < inA->rmap->mapping->n; i++) {  in MatMPIAIJDiagonalScaleLocalSetUp()
243     if (lindices[inA->rmap->mapping->indices[i]]) {  in MatMPIAIJDiagonalScaleLocalSetUp()
245       r_rmapo[i] = lindices[inA->rmap->mapping->indices[i]];  in MatMPIAIJDiagonalScaleLocalSetUp()
251   for (i = 0; i < inA->rmap->mapping->n; i++) {  in MatMPIAIJDiagonalScaleLocalSetUp()