Lines Matching refs:nr

249 static PetscErrorCode MatZeroRowsColumns_Local_Shell(Mat mat, PetscInt nr, PetscInt rows[], PetscIn…  in MatZeroRowsColumns_Local_Shell()  argument
265 PetscCall(PetscMalloc1(nr, &idxs)); in MatZeroRowsColumns_Local_Shell()
266 for (i = 0; i < nr; i++) idxs[i] = rows[i] + rst; in MatZeroRowsColumns_Local_Shell()
267 PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)mat), nr, idxs, PETSC_OWN_POINTER, &is1)); in MatZeroRowsColumns_Local_Shell()
287 PetscCall(ISGetLocalSize(shell->zrows, &nr)); in MatZeroRowsColumns_Local_Shell()
288 PetscCall(PetscMalloc1(nr, &gidxs)); in MatZeroRowsColumns_Local_Shell()
289 for (i = 0, cum = 0; i < nr; i++) { in MatZeroRowsColumns_Local_Shell()
319 PetscInt nr, *lrows; in MatZeroRows_Shell() local
355 PetscCall(PetscLayoutMapLocal(mat->rmap, n, rows, &nr, &lrows, NULL)); in MatZeroRows_Shell()
356 PetscCall(MatZeroRowsColumns_Local_Shell(mat, nr, lrows, 0, NULL, diag, PETSC_FALSE)); in MatZeroRows_Shell()
366 PetscInt nr, nc; in MatZeroRowsColumns_Shell() local
376 for (i = 0, nr = 0; i < n; i++) in MatZeroRowsColumns_Shell()
377 if (rowscols[i] < mat->rmap->N) grows[nr++] = rowscols[i]; in MatZeroRowsColumns_Shell()
389 PetscCall(VecSetValues(bt, nr, grows, vals, INSERT_VALUES)); /* bt = [-A12*x2,0] */ in MatZeroRowsColumns_Shell()
393 PetscCall(VecSetValues(bt, nr, grows, vals, INSERT_VALUES)); /* b = [b1 - A12*x2, 0] */ in MatZeroRowsColumns_Shell()
414 PetscCall(PetscLayoutMapLocal(mat->rmap, n, rowscols, &nr, &lrows, NULL)); in MatZeroRowsColumns_Shell()
417 nc = nr; in MatZeroRowsColumns_Shell()
428 PetscCall(MatZeroRowsColumns_Local_Shell(mat, nr, lrows, nc, lcols, diag, PETSC_TRUE)); in MatZeroRowsColumns_Shell()