Lines Matching refs:lrows
865 PetscInt *lrows; in MatZeroRows_MPIAIJ() local
871 PetscCall(MatZeroRowsMapLocal_Private(A, N, rows, &len, &lrows)); in MatZeroRows_MPIAIJ()
881 for (r = 0; r < len; ++r) bb[lrows[r]] = diag * xx[lrows[r]]; in MatZeroRows_MPIAIJ()
887 PetscCall(MatZeroRows(mat->A, len, lrows, diag, NULL, NULL)); in MatZeroRows_MPIAIJ()
888 PetscCall(MatZeroRows(mat->B, len, lrows, 0.0, NULL, NULL)); in MatZeroRows_MPIAIJ()
908 PetscCall(MatZeroRows(mat->A, len, lrows, 0.0, NULL, NULL)); in MatZeroRows_MPIAIJ()
909 PetscCall(MatZeroRows(mat->B, len, lrows, 0.0, NULL, NULL)); in MatZeroRows_MPIAIJ()
911 const PetscInt row = lrows[r] + A->rmap->rstart; in MatZeroRows_MPIAIJ()
918 PetscCall(MatZeroRows(mat->A, len, lrows, 0.0, NULL, NULL)); in MatZeroRows_MPIAIJ()
919 PetscCall(MatZeroRows(mat->B, len, lrows, 0.0, NULL, NULL)); in MatZeroRows_MPIAIJ()
921 PetscCall(PetscFree(lrows)); in MatZeroRows_MPIAIJ()
938 PetscInt *lrows, *owners = A->rmap->range; in MatZeroRowsColumns_MPIAIJ() local
951 PetscCall(PetscMalloc1(n, &lrows)); in MatZeroRowsColumns_MPIAIJ()
952 for (r = 0; r < n; ++r) lrows[r] = -1; in MatZeroRowsColumns_MPIAIJ()
966 PetscCall(PetscSFReduceBegin(sf, MPIU_INT, (PetscInt *)rows, lrows, MPI_LOR)); in MatZeroRowsColumns_MPIAIJ()
967 PetscCall(PetscSFReduceEnd(sf, MPIU_INT, (PetscInt *)rows, lrows, MPI_LOR)); in MatZeroRowsColumns_MPIAIJ()
971 if (lrows[r] >= 0) lrows[len++] = r; in MatZeroRowsColumns_MPIAIJ()
973 PetscCall(MatZeroRowsColumns(l->A, len, lrows, diag, x, b)); in MatZeroRowsColumns_MPIAIJ()
978 for (i = 0; i < len; i++) bb[lrows[i]] = 1; in MatZeroRowsColumns_MPIAIJ()
997 …Call(PetscArrayzero(PetscSafePointerPlusOffset(aij_a, ii[lrows[i]]), ii[lrows[i] + 1] - ii[lrows[i… in MatZeroRowsColumns_MPIAIJ()
1041 PetscCall(PetscFree(lrows)); in MatZeroRowsColumns_MPIAIJ()