Lines Matching refs:cmap
65 rend2 = PetscMin(rend, A->cmap->N); in MatShift_MPIDense()
150 … PetscCheck(idxn[j] < mat->cmap->N, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Column too large"); in MatSetValues_MPIDense()
179 … PetscCheck(idxn[j] < mat->cmap->N, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Column too large"); in MatGetValues_MPIDense()
204 PetscCall(PetscLayoutSetUp(A->cmap)); in MatDenseSetLDA_MPIDense()
206 PetscCall(MatSetSizes(a->A, A->rmap->n, A->cmap->N, A->rmap->n, A->cmap->N)); in MatDenseSetLDA_MPIDense()
685 len = PetscMin(a->A->rmap->n, a->A->cmap->n); in MatGetDiagonal_MPIDense()
701 …tate((PetscObject)mat, "Rows=%" PetscInt_FMT ", Cols=%" PetscInt_FMT, mat->rmap->N, mat->cmap->N)); in MatDestroy_MPIDense()
837 PetscInt M = mat->rmap->N, N = mat->cmap->N, m, row, i, nz; 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()
1048 PetscCall(PetscCalloc1(A->cmap->N, &tmp)); in MatNorm_MPIDense()
1051 for (j = 0; j < mdn->A->cmap->n; j++) { in MatNorm_MPIDense()
1057 …PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, tmp, A->cmap->N, MPIU_REAL, MPIU_SUM, PetscObjectComm((P… in MatNorm_MPIDense()
1058 for (j = 0; j < A->cmap->N; j++) { 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()
1090 n = a->A->cmap->n; in MatTranspose_MPIDense()
1118 PetscCall(PetscLayoutSetUp(A->cmap)); in MatSetUp_MPIDense()
1375 PetscCall(PetscLayoutSetUp(mat->cmap)); in MatMPIDenseSetPreallocation_MPIDense()
1378 PetscCall(MatSetSizes(a->A, mat->rmap->n, mat->cmap->N, mat->rmap->n, mat->cmap->N)); in MatMPIDenseSetPreallocation_MPIDense()
1411 …PetscCall(MatCreateMPIMatConcatenateSeqMat(PetscObjectComm((PetscObject)A), B, A->cmap->n, !C ? MA… in MatConvert_MPIAIJ_MPIDense()
1428 …PetscCall(MatCreateMPIMatConcatenateSeqMat(PetscObjectComm((PetscObject)A), B, A->cmap->n, !C ? MA… in MatConvert_MPIDense_MPIAIJ()
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()
1635 pcbegin = PetscMax(0, PetscMin(A->cmap->rend, cbegin) - A->cmap->rstart); in MatDenseGetSubMatrix_MPIDense()
1636 pcend = PetscMin(A->cmap->n, PetscMax(0, cend - A->cmap->rstart)); in MatDenseGetSubMatrix_MPIDense()
1646 if (cend - cbegin == A->cmap->N) PetscCall(PetscLayoutReference(A->cmap, &a->cmat->cmap)); in MatDenseGetSubMatrix_MPIDense()
1648 PetscCall(PetscLayoutSetLocalSize(a->cmat->cmap, pcend - pcbegin)); in MatDenseGetSubMatrix_MPIDense()
1649 PetscCall(PetscLayoutSetSize(a->cmat->cmap, cend - cbegin)); in MatDenseGetSubMatrix_MPIDense()
1650 PetscCall(PetscLayoutSetUp(a->cmat->cmap)); in MatDenseGetSubMatrix_MPIDense()
1670 if ((cend - cbegin != a->cmat->cmap->N) || (A->cmap->range[1] != A->cmap->N)) { in MatDenseGetSubMatrix_MPIDense()
1671 PetscCall(PetscLayoutDestroy(&a->cmat->cmap)); in MatDenseGetSubMatrix_MPIDense()
1672 PetscCall(PetscLayoutCreate(comm, &a->cmat->cmap)); in MatDenseGetSubMatrix_MPIDense()
1673 PetscCall(PetscLayoutSetLocalSize(a->cmat->cmap, pcend - pcbegin)); in MatDenseGetSubMatrix_MPIDense()
1674 PetscCall(PetscLayoutSetSize(a->cmat->cmap, cend - cbegin)); in MatDenseGetSubMatrix_MPIDense()
1675 PetscCall(PetscLayoutSetUp(a->cmat->cmap)); in MatDenseGetSubMatrix_MPIDense()
1986 PetscCall(MatSetSizes(mat, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N)); in MatDuplicate_MPIDense()
1998 PetscCall(PetscLayoutReference(A->cmap, &mat->cmap)); in MatDuplicate_MPIDense()
2075 PetscInt i, cN = C->cmap->N, proc, k, j, lda; in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2129 if (B->cmap->N > 1) { in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2132 PetscCall(PetscMPIIntCast(B->cmap->N, &mpi_N)); in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2143 if (B->cmap->N > 1) PetscCallMPI(MPI_Type_free(&vector_type)); in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2176 PetscInt cm = A->cmap->n, cM, cN = B->cmap->N; in MatTransposeMatMultSymbolic_MPIDense_MPIDense()
2189 PetscCall(MatSetSizes(C, cm, B->cmap->n, A->cmap->N, B->cmap->N)); in MatTransposeMatMultSymbolic_MPIDense_MPIDense()
2226 …cmap->n == B->cmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Matrix local column dimensions are inc… in MatMatTransposeMultSymbolic_MPIDense_MPIDense()
2246 bufsiz = A->cmap->N * maxRows; 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()
2270 PetscInt i, cK = A->cmap->N, sendsiz, recvsiz, k, j, bn; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2355 PetscInt blda, i, cK = A->cmap->N, k, j, bn; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv()
2385 PetscCall(PetscBLASIntCast(c->A->cmap->n, &cn)); in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv()
2487 if (B->cmap->N > 1) { in MatMatMultNumeric_MPIDense_MPIDense()
2490 PetscCall(PetscMPIIntCast(B->cmap->N, &mpi_N)); in MatMatMultNumeric_MPIDense_MPIDense()
2501 if (B->cmap->N > 1) PetscCallMPI(MPI_Type_free(&vector_type)); in MatMatMultNumeric_MPIDense_MPIDense()
2508 for (PetscInt i = 0; i < C->cmap->N; ++i) { in MatMatMultNumeric_MPIDense_MPIDense()
2531 …PetscCheck(A->cmap->rstart == B->rmap->rstart && A->cmap->rend == B->rmap->rend, PetscObjectComm((… 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()
2570 PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, A->cmap->N, B->cmap->N, NULL, &ab->Be)); in MatMatMultSymbolic_MPIDense_MPIDense()