Lines Matching refs:rmap
59 PetscInt j, lda, rstart = A->rmap->rstart, rend = A->rmap->rend, rend2; in MatShift_MPIDense()
77 PetscInt lrow, rstart = A->rmap->rstart, rend = A->rmap->rend; in MatGetRow_MPIDense()
89 PetscInt lrow, rstart = A->rmap->rstart, rend = A->rmap->rend; in MatRestoreRow_MPIDense()
101 PetscInt m = A->rmap->n, rstart = A->rmap->rstart; in MatGetDiagonalBlock_MPIDense()
136 PetscInt i, j, rstart = mat->rmap->rstart, rend = mat->rmap->rend, row; in MatSetValues_MPIDense()
142 PetscCheck(idxm[i] < mat->rmap->N, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row too large"); in MatSetValues_MPIDense()
169 PetscInt i, j, rstart = mat->rmap->rstart, rend = mat->rmap->rend, row; in MatGetValues_MPIDense()
174 PetscCheck(idxm[i] < mat->rmap->N, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row too large"); in MatGetValues_MPIDense()
203 PetscCall(PetscLayoutSetUp(A->rmap)); in MatDenseSetLDA_MPIDense()
206 PetscCall(MatSetSizes(a->A, A->rmap->n, A->cmap->N, A->rmap->n, A->cmap->N)); in MatDenseSetLDA_MPIDense()
387 PetscCall(MatStashScatterBegin_Private(mat, &mat->stash, mat->rmap->range)); in MatAssemblyBegin_MPIDense()
443 PetscCall(PetscLayoutMapLocal(A->rmap, n, rows, &len, &lrows, NULL)); in MatZeroRows_MPIDense()
676 PetscInt lda, len, i, nl, ng, m = A->rmap->n, radd; in MatGetDiagonal_MPIDense()
683 PetscCheck(ng == A->rmap->N, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Nonconforming mat and vec"); in MatGetDiagonal_MPIDense()
685 len = PetscMin(a->A->rmap->n, a->A->cmap->n); in MatGetDiagonal_MPIDense()
686 radd = A->rmap->rstart * m; in MatGetDiagonal_MPIDense()
701 …tate((PetscObject)mat, "Rows=%" PetscInt_FMT ", Cols=%" PetscInt_FMT, mat->rmap->N, mat->cmap->N)); in MatDestroy_MPIDense()
816 …_FMT " nz alloced %" PetscInt_FMT " mem %" PetscInt_FMT " \n", rank, mat->rmap->n, (PetscInt)info.… in MatView_MPIDense_ASCIIorDraworSocket()
837 PetscInt M = mat->rmap->N, N = mat->cmap->N, m, row, i, nz; in MatView_MPIDense_ASCIIorDraworSocket()
855 row = mat->rmap->rstart; in MatView_MPIDense_ASCIIorDraworSocket()
856 m = mdn->A->rmap->n; in MatView_MPIDense_ASCIIorDraworSocket()
965 if (a->A && A->rmap->n == A->cmap->n) PetscCall(MatSetOption(a->A, op, flg)); in MatSetOption_MPIDense()
978 PetscInt i, j, s2a, s3a, s2, s3, m = mdn->A->rmap->n, n = mdn->A->cmap->n, lda; in MatDiagonalScale_MPIDense()
1038 for (i = 0; i < mdn->A->cmap->n * mdn->A->rmap->n; i++) { in MatNorm_MPIDense()
1044 PetscCall(PetscLogFlops(2.0 * mdn->A->cmap->n * mdn->A->rmap->n)); in MatNorm_MPIDense()
1052 for (i = 0; i < mdn->A->rmap->n; i++) { in MatNorm_MPIDense()
1062 PetscCall(PetscLogFlops(A->cmap->n * A->rmap->n)); in MatNorm_MPIDense()
1076 PetscInt M = A->rmap->N, N = A->cmap->N, m, n, *rwork, rstart = A->rmap->rstart; in MatTranspose_MPIDense()
1084 PetscCall(MatSetSizes(B, A->cmap->n, A->rmap->n, N, M)); in MatTranspose_MPIDense()
1089 m = a->A->rmap->n; in MatTranspose_MPIDense()
1117 PetscCall(PetscLayoutSetUp(A->rmap)); in MatSetUp_MPIDense()
1374 PetscCall(PetscLayoutSetUp(mat->rmap)); in MatMPIDenseSetPreallocation_MPIDense()
1378 PetscCall(MatSetSizes(a->A, mat->rmap->n, mat->cmap->N, mat->rmap->n, mat->cmap->N)); in MatMPIDenseSetPreallocation_MPIDense()
1441 PetscInt m = A->rmap->n, N = A->cmap->N, rstart = A->rmap->rstart, i, *rows, *cols, lda; in MatConvert_MPIDense_Elemental()
1449 PetscCall(MatSetSizes(mat_elemental, PETSC_DECIDE, PETSC_DECIDE, A->rmap->N, A->cmap->N)); in MatConvert_MPIDense_Elemental()
1633 prbegin = PetscMax(0, PetscMin(A->rmap->rend, rbegin) - A->rmap->rstart); in MatDenseGetSubMatrix_MPIDense()
1634 prend = PetscMin(A->rmap->n, PetscMax(0, rend - A->rmap->rstart)); in MatDenseGetSubMatrix_MPIDense()
1640 if (rend - rbegin == A->rmap->N) PetscCall(PetscLayoutReference(A->rmap, &a->cmat->rmap)); in MatDenseGetSubMatrix_MPIDense()
1642 PetscCall(PetscLayoutSetLocalSize(a->cmat->rmap, prend - prbegin)); in MatDenseGetSubMatrix_MPIDense()
1643 PetscCall(PetscLayoutSetSize(a->cmat->rmap, rend - rbegin)); in MatDenseGetSubMatrix_MPIDense()
1644 PetscCall(PetscLayoutSetUp(a->cmat->rmap)); in MatDenseGetSubMatrix_MPIDense()
1660 PetscCall(PetscLayoutDestroy(&a->cmat->rmap)); in MatDenseGetSubMatrix_MPIDense()
1661 PetscCall(PetscLayoutCreate(comm, &a->cmat->rmap)); in MatDenseGetSubMatrix_MPIDense()
1662 PetscCall(PetscLayoutSetLocalSize(a->cmat->rmap, prend - prbegin)); in MatDenseGetSubMatrix_MPIDense()
1663 PetscCall(PetscLayoutSetSize(a->cmat->rmap, rend - rbegin)); in MatDenseGetSubMatrix_MPIDense()
1664 PetscCall(PetscLayoutSetUp(a->cmat->rmap)); in MatDenseGetSubMatrix_MPIDense()
1986 PetscCall(MatSetSizes(mat, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N)); in MatDuplicate_MPIDense()
1997 PetscCall(PetscLayoutReference(A->rmap, &mat->rmap)); in MatDuplicate_MPIDense()
2093 if (ranges[1] == C->rmap->N) { in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2109 if (atb_lda != C->rmap->N) { in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2118 if (c_lda != C->rmap->N) { in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2185 …rmap->rstart == B->rmap->rstart && A->rmap->rend == B->rmap->rend, comm, PETSC_ERR_ARG_SIZ, "Matri… in MatTransposeMatMultSymbolic_MPIDense_MPIDense()
2186 A->rmap->rend, B->rmap->rstart, B->rmap->rend); in MatTransposeMatMultSymbolic_MPIDense_MPIDense()
2202 cM = C->rmap->N; in MatTransposeMatMultSymbolic_MPIDense_MPIDense()
2232 PetscCall(MatSetSizes(C, A->rmap->n, B->rmap->n, A->rmap->N, B->rmap->N)); in MatMatTransposeMultSymbolic_MPIDense_MPIDense()
2245 …ows = 0, i = 0; i < size; i++) maxRows = PetscMax(maxRows, B->rmap->range[i + 1] - B->rmap->range[… in MatMatTransposeMultSymbolic_MPIDense_MPIDense()
2250 …PetscCall(PetscMalloc2(B->rmap->n * B->cmap->N, &abt->buf[0], B->rmap->N * B->cmap->N, &abt->buf[1… in MatMatTransposeMultSymbolic_MPIDense_MPIDense()
2252 …for (i = 0; i <= size; i++) PetscCall(PetscMPIIntCast(B->rmap->range[i] * A->cmap->N, &abt->recvdi… in MatMatTransposeMultSymbolic_MPIDense_MPIDense()
2294 bn = B->rmap->n; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2310 PetscCall(PetscBLASIntCast(c->A->rmap->n, &cm)); in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2374 bn = B->rmap->n; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv()
2384 PetscCall(PetscBLASIntCast(c->A->rmap->n, &cm)); in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv()
2457 PetscCall(PetscLayoutGetRanges(B->rmap, &ranges)); in MatMatMultNumeric_MPIDense_MPIDense()
2473 if (be_lda != B->rmap->N) { in MatMatMultNumeric_MPIDense_MPIDense()
2480 if (b_lda != B->rmap->N) { in MatMatMultNumeric_MPIDense_MPIDense()
2509 …scSFBcastBegin(mdn->Mvctx, MPIU_SCALAR, read + i * lda, write + i * ab->Be->rmap->n, MPI_REPLACE)); in MatMatMultNumeric_MPIDense_MPIDense()
2510 …PetscCall(PetscSFBcastEnd(mdn->Mvctx, MPIU_SCALAR, read + i * lda, write + i * ab->Be->rmap->n, MP… in MatMatMultNumeric_MPIDense_MPIDense()
2531 …PetscCheck(A->cmap->rstart == B->rmap->rstart && A->cmap->rend == B->rmap->rend, PetscObjectComm((… in MatMatMultSymbolic_MPIDense_MPIDense()
2532 A->rmap->rstart, A->rmap->rend, B->rmap->rstart, B->rmap->rend); in MatMatMultSymbolic_MPIDense_MPIDense()
2538 PetscCall(MatSetSizes(C, A->rmap->n, B->cmap->n, A->rmap->N, B->cmap->N)); in MatMatMultSymbolic_MPIDense_MPIDense()
2550 PetscCall(MatSetSizes(ab->Ae, PETSC_DECIDE, PETSC_DECIDE, A->rmap->N, A->cmap->N)); in MatMatMultSymbolic_MPIDense_MPIDense()
2556 PetscCall(MatSetSizes(ab->Be, PETSC_DECIDE, PETSC_DECIDE, B->rmap->N, B->cmap->N)); in MatMatMultSymbolic_MPIDense_MPIDense()
2611 …PetscCheck(A->rmap->rstart == B->rmap->rstart && A->rmap->rend == B->rmap->rend, PETSC_COMM_SELF, … in MatProductSetFromOptions_MPIDense_AtB()
2612 A->rmap->rstart, A->rmap->rend, B->rmap->rstart, B->rmap->rend); in MatProductSetFromOptions_MPIDense_AtB()