Lines Matching refs:nrow
165 PetscInt nrow, m, i; in MatCreateSubMatrices_Htool() local
173 PetscCall(ISGetLocalSize(irow[i], &nrow)); in MatCreateSubMatrices_Htool()
177 …if (scall != MAT_REUSE_MATRIX) PetscCall(MatCreateDense(PETSC_COMM_SELF, nrow, m, nrow, m, nullptr… in MatCreateSubMatrices_Htool()
184 it = std::lower_bound(idxr, idxr + nrow, A->rmap->rstart); in MatCreateSubMatrices_Htool()
185 if (it != idxr + nrow && *it == A->rmap->rstart) { /* rmap->rstart in IS? */ in MatCreateSubMatrices_Htool()
186 …if (std::distance(idxr, it) + A->rmap->n <= nrow) { /* long enough IS to store the local diagonal … in MatCreateSubMatrices_Htool()
198 …scInt k = 0; k < A->rmap->n; ++k) PetscCall(PetscArraycpy(ptr + (m + k) * nrow + m, copy + k * A->… in MatCreateSubMatrices_Htool()
201 … a->wrapper->copy_submatrix(nrow, m, idxr, idxc, ptr); /* vertical block B from above */ in MatCreateSubMatrices_Htool()
205 PetscCall(MatDenseSetLDA(B, nrow)); in MatCreateSubMatrices_Htool()
206 … PetscCall(MatCreateDense(PETSC_COMM_SELF, m, A->rmap->n, m, A->rmap->n, ptr + m * nrow, &BT)); in MatCreateSubMatrices_Htool()
207 PetscCall(MatDenseSetLDA(BT, nrow)); in MatCreateSubMatrices_Htool()
218 a->wrapper->copy_submatrix(m, 1, idxr, idxc + m + k, ptr + (m + k) * nrow); in MatCreateSubMatrices_Htool()
222 if (m + A->rmap->n != nrow) { in MatCreateSubMatrices_Htool()
223 …apper->copy_submatrix(nrow, std::distance(it + A->rmap->n, idxr + nrow), idxr, idxc + m + A->rmap-… in MatCreateSubMatrices_Htool()
226 …e(PETSC_COMM_SELF, A->rmap->n, nrow - (m + A->rmap->n), A->rmap->n, nrow - (m + A->rmap->n), ptr +… in MatCreateSubMatrices_Htool()
227 PetscCall(MatDenseSetLDA(B, nrow)); in MatCreateSubMatrices_Htool()
228 …MatCreateDense(PETSC_COMM_SELF, nrow - (m + A->rmap->n), A->rmap->n, nrow - (m + A->rmap->n), A->r… in MatCreateSubMatrices_Htool()
229 PetscCall(MatDenseSetLDA(BT, nrow)); in MatCreateSubMatrices_Htool()
240 …bmatrix(std::distance(it + A->rmap->n, idxr + nrow), 1, it + A->rmap->n, idxc + m + k, ptr + (m + … in MatCreateSubMatrices_Htool()
250 if (!flg) a->wrapper->copy_submatrix(nrow, m, idxr, idxc, ptr); /* reassemble everything */ in MatCreateSubMatrices_Htool()