Lines Matching refs:inA

199 static PetscErrorCode MatMPIBAIJDiagonalScaleLocalSetUp(Mat inA, Vec scale)  in MatMPIBAIJDiagonalScaleLocalSetUp()  argument
201 Mat_MPIBAIJ *ina = (Mat_MPIBAIJ *)inA->data; /*access private part of matrix */ in MatMPIBAIJDiagonalScaleLocalSetUp()
203 PetscInt bs = inA->rmap->bs, i, n, nt, j, cstart, cend, no, *garray = ina->garray, *lindices; in MatMPIBAIJDiagonalScaleLocalSetUp()
207 PetscCall(MatGetOwnershipRange(inA, &cstart, &cend)); in MatMPIBAIJDiagonalScaleLocalSetUp()
209 PetscCall(PetscCalloc1(inA->rmap->mapping->n, &r_rmapd)); in MatMPIBAIJDiagonalScaleLocalSetUp()
211 for (i = 0; i < inA->rmap->mapping->n; i++) { in MatMPIBAIJDiagonalScaleLocalSetUp()
212 … if (inA->rmap->mapping->indices[i] * bs >= cstart && inA->rmap->mapping->indices[i] * bs < cend) { in MatMPIBAIJDiagonalScaleLocalSetUp()
214 r_rmapd[i] = inA->rmap->mapping->indices[i] + 1; in MatMPIBAIJDiagonalScaleLocalSetUp()
219 for (i = 0; i < inA->rmap->mapping->n; i++) { in MatMPIBAIJDiagonalScaleLocalSetUp()
229 no = inA->rmap->mapping->n - nt; in MatMPIBAIJDiagonalScaleLocalSetUp()
230 PetscCall(PetscCalloc1(inA->rmap->mapping->n, &r_rmapo)); in MatMPIBAIJDiagonalScaleLocalSetUp()
232 for (i = 0; i < inA->rmap->mapping->n; i++) { in MatMPIBAIJDiagonalScaleLocalSetUp()
233 if (lindices[inA->rmap->mapping->indices[i]]) { in MatMPIBAIJDiagonalScaleLocalSetUp()
235 r_rmapo[i] = lindices[inA->rmap->mapping->indices[i]]; in MatMPIBAIJDiagonalScaleLocalSetUp()
241 for (i = 0; i < inA->rmap->mapping->n; i++) { in MatMPIBAIJDiagonalScaleLocalSetUp()