Lines Matching refs:inA
200 static PetscErrorCode MatMPISELLDiagonalScaleLocalSetUp(Mat inA, Vec scale) in MatMPISELLDiagonalScaleLocalSetUp() argument
202 Mat_MPISELL *ina = (Mat_MPISELL *)inA->data; /*access private part of matrix */ in MatMPISELLDiagonalScaleLocalSetUp()
207 PetscCall(MatGetOwnershipRange(inA, &cstart, &cend)); in MatMPISELLDiagonalScaleLocalSetUp()
209 PetscCall(PetscCalloc1(inA->rmap->mapping->n, &r_rmapd)); in MatMPISELLDiagonalScaleLocalSetUp()
211 for (i = 0; i < inA->rmap->mapping->n; i++) { in MatMPISELLDiagonalScaleLocalSetUp()
212 if (inA->rmap->mapping->indices[i] >= cstart && inA->rmap->mapping->indices[i] < cend) { in MatMPISELLDiagonalScaleLocalSetUp()
214 r_rmapd[i] = inA->rmap->mapping->indices[i] + 1; in MatMPISELLDiagonalScaleLocalSetUp()
219 for (i = 0; i < inA->rmap->mapping->n; i++) { in MatMPISELLDiagonalScaleLocalSetUp()
224 PetscCall(PetscCalloc1(inA->cmap->N, &lindices)); in MatMPISELLDiagonalScaleLocalSetUp()
226 no = inA->rmap->mapping->n - nt; in MatMPISELLDiagonalScaleLocalSetUp()
227 PetscCall(PetscCalloc1(inA->rmap->mapping->n, &r_rmapo)); in MatMPISELLDiagonalScaleLocalSetUp()
229 for (i = 0; i < inA->rmap->mapping->n; i++) { in MatMPISELLDiagonalScaleLocalSetUp()
230 if (lindices[inA->rmap->mapping->indices[i]]) { in MatMPISELLDiagonalScaleLocalSetUp()
232 r_rmapo[i] = lindices[inA->rmap->mapping->indices[i]]; in MatMPISELLDiagonalScaleLocalSetUp()
238 for (i = 0; i < inA->rmap->mapping->n; i++) { in MatMPISELLDiagonalScaleLocalSetUp()