Lines Matching refs:rmap
12 …State((PetscObject)mat, "Rows=%" PetscInt_FMT ",Cols=%" PetscInt_FMT, mat->rmap->N, mat->cmap->N)); in MatDestroy_MPIBAIJ()
60 PetscInt i, *idxb = NULL, m = A->rmap->n, bs = A->cmap->bs; in MatGetRowMaxAbs_MPIBAIJ()
147 PetscInt nbs = B->nbs, i, bs = mat->rmap->bs; in MatCreateColmap_MPIBAIJ_Private()
246 PetscInt rstart_orig = mat->rmap->rstart; in MatSetValues_MPIBAIJ()
247 PetscInt rend_orig = mat->rmap->rend, cstart_orig = mat->cmap->rstart; in MatSetValues_MPIBAIJ()
248 PetscInt cend_orig = mat->cmap->rend, bs = mat->rmap->bs; in MatSetValues_MPIBAIJ()
268 …im[i] < mat->rmap->N, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row too large: row %" PetscInt_F… in MatSetValues_MPIBAIJ()
331 PetscInt *aj = a->j, nonew = a->nonew, bs2 = a->bs2, bs = A->rmap->bs; in MatSetValuesBlocked_SeqBAIJ_Inlined()
415 PetscInt cend = baij->cendbs, bs = mat->rmap->bs, bs2 = baij->bs2; in MatSetValuesBlocked_MPIBAIJ()
497 PetscInt rstart_orig = mat->rmap->rstart; in MatSetValues_MPIBAIJ_HT()
498 PetscInt rend_orig = mat->rmap->rend, Nbs = baij->Nbs; in MatSetValues_MPIBAIJ_HT()
499 PetscInt h1, key, size = baij->ht_size, bs = mat->rmap->bs, *HT = baij->ht, idx; in MatSetValues_MPIBAIJ_HT()
508 …im[i] < mat->rmap->N, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row too large: row %" PetscInt_F… in MatSetValues_MPIBAIJ_HT()
563 …PetscInt rend = mat->rmap->rend, stepval, bs = mat->rmap->bs, bs2 = baij->bs2, nbs2 = n … in MatSetValuesBlocked_MPIBAIJ_HT()
655 PetscInt bs = mat->rmap->bs, i, j, bsrstart = mat->rmap->rstart, bsrend = mat->rmap->rend; in MatGetValues_MPIBAIJ()
661 …m[i] < mat->rmap->N, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row too large: row %" PetscInt_FM… in MatGetValues_MPIBAIJ()
693 PetscInt i, j, bs2 = baij->bs2, bs = baij->A->rmap->bs, nz, row, col; in MatNorm_MPIBAIJ()
891 PetscCall(MatStashScatterBegin_Private(mat, &mat->stash, mat->rmap->range)); in MatAssemblyBegin_MPIBAIJ()
1012 PetscInt bs = mat->rmap->bs; in MatView_MPIBAIJ_ASCIIorDraworSocket()
1027 …" nz alloced %" PetscInt_FMT " bs %" PetscInt_FMT " mem %g\n", rank, mat->rmap->n, (PetscInt)info.… in MatView_MPIBAIJ_ASCIIorDraworSocket()
1028 mat->rmap->bs, info.memory)); in MatView_MPIBAIJ_ASCIIorDraworSocket()
1053 … PetscInt M = mat->rmap->N, N = mat->cmap->N, *ai, *aj, col, i, j, k, *rvals, mbs = baij->mbs; in MatView_MPIBAIJ_ASCIIorDraworSocket()
1066 PetscCall(MatMPIBAIJSetPreallocation(A, mat->rmap->bs, 0, NULL, 0, NULL)); in MatView_MPIBAIJ_ASCIIorDraworSocket()
1140 M = mat->rmap->N; in MatView_MPIBAIJ_Binary()
1142 m = mat->rmap->n; in MatView_MPIBAIJ_Binary()
1143 rs = mat->rmap->rstart; in MatView_MPIBAIJ_Binary()
1145 bs = mat->rmap->bs; in MatView_MPIBAIJ_Binary()
1227 …PetscCheck(nt == A->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Incompatible partition of A and … in MatMult_MPIBAIJ()
1284 …PetscCheck(A->rmap->N == A->cmap->N, PETSC_COMM_SELF, PETSC_ERR_SUP, "Supports only square matrix … in MatGetDiagonal_MPIBAIJ()
1303 PetscInt bs = matin->rmap->bs, bs2 = mat->bs2, i, *cworkA, *cworkB, **pcA, **pcB; in MatGetRow_MPIBAIJ()
1304 PetscInt nztot, nzA, nzB, lrow, brstart = matin->rmap->rstart, brend = matin->rmap->rend; in MatGetRow_MPIBAIJ()
1409 info->block_size = (PetscReal)matin->rmap->bs; in MatGetInfo_MPIBAIJ()
1492 if (a->A && A->rmap->n == A->cmap->n) PetscCall(MatSetOption(a->A, op, flg)); in MatSetOption_MPIBAIJ()
1505 PetscInt M = A->rmap->N, N = A->cmap->N, *ai, *aj, i, *rvals, j, k, col; in MatTranspose_MPIBAIJ()
1506 PetscInt bs = A->rmap->bs, mbs = baij->mbs; in MatTranspose_MPIBAIJ()
1513 PetscCall(MatSetSizes(B, A->cmap->n, A->rmap->n, N, M)); in MatTranspose_MPIBAIJ()
1516 PetscCall(MatMPIBAIJSetPreallocation(B, A->rmap->bs, PETSC_DECIDE, NULL, PETSC_DECIDE, NULL)); in MatTranspose_MPIBAIJ()
1636 const PetscInt row = lrows[r] + A->rmap->rstart; in MatZeroRows_MPIBAIJ()
1659 PetscInt *lrows, *owners = A->rmap->range; in MatZeroRowsColumns_MPIBAIJ()
1666 PetscInt bs = A->rmap->bs, bs2 = baij->bs2; in MatZeroRowsColumns_MPIBAIJ()
1670 PetscCall(PetscMPIIntCast(A->rmap->n, &n)); in MatZeroRowsColumns_MPIBAIJ()
1677 …>= 0 && A->rmap->N > idx, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row %" PetscInt_FMT " out of… in MatZeroRowsColumns_MPIBAIJ()
1679 PetscCall(PetscLayoutFindOwner(A->rmap, idx, &p)); in MatZeroRowsColumns_MPIBAIJ()
1722 for (i = 0; i < l->B->rmap->N; ++i) { in MatZeroRowsColumns_MPIBAIJ()
1799 PetscInt bs = Y->rmap->bs, m = Y->rmap->N / bs; in MatAXPYGetPreallocation_MPIBAIJ()
1813 PetscInt bs2 = Y->rmap->bs * Y->rmap->bs; in MatAXPY_MPIBAIJ()
1831 PetscInt *nnz_d, *nnz_o, bs = Y->rmap->bs; in MatAXPY_MPIBAIJ()
1832 PetscCall(PetscMalloc1(yy->A->rmap->N, &nnz_d)); in MatAXPY_MPIBAIJ()
1833 PetscCall(PetscMalloc1(yy->B->rmap->N, &nnz_o)); in MatAXPY_MPIBAIJ()
1836 PetscCall(MatSetSizes(B, Y->rmap->n, Y->cmap->n, Y->rmap->N, Y->cmap->N)); in MatAXPY_MPIBAIJ()
1925 …PetscCall(ISCompressIndicesGeneral(mat->rmap->N, mat->rmap->n, mat->rmap->bs, 1, &isrow, &isrow_ne… in MatCreateSubMatrix_MPIBAIJ_Private()
2094 PetscInt sendcount, i, *rstarts = A->rmap->range, n, cnt, j, bs = A->rmap->bs; in MatGetSeqNonzeroStructure_MPIBAIJ()
2102 PetscCall(PetscMalloc1(A->rmap->N / bs, &lens)); in MatGetSeqNonzeroStructure_MPIBAIJ()
2103 …rmap->rstart / bs; i < A->rmap->rend / bs; i++) lens[i] = ad->i[i - A->rmap->rstart / bs + 1] - ad… in MatGetSeqNonzeroStructure_MPIBAIJ()
2107 PetscCall(PetscMPIIntCast(A->rmap->range[i + 1] / bs - A->rmap->range[i] / bs, &recvcounts[i])); in MatGetSeqNonzeroStructure_MPIBAIJ()
2108 PetscCall(PetscMPIIntCast(A->rmap->range[i] / bs, &displs[i])); in MatGetSeqNonzeroStructure_MPIBAIJ()
2113 PetscCall(MatSetSizes(B, A->rmap->N / bs, A->cmap->N / bs, PETSC_DETERMINE, PETSC_DETERMINE)); in MatGetSeqNonzeroStructure_MPIBAIJ()
2123 n = A->rmap->rend / bs - A->rmap->rstart / bs; in MatGetSeqNonzeroStructure_MPIBAIJ()
2130 if (garray[*b_jsendbuf] > A->rmap->rstart / bs + i) break; in MatGetSeqNonzeroStructure_MPIBAIJ()
2136 … for (j = ad->i[i]; j < ad->i[i + 1]; j++) jsendbuf[cnt++] = A->rmap->rstart / bs + *a_jsendbuf++; in MatGetSeqNonzeroStructure_MPIBAIJ()
2146 for (j = A->rmap->range[i] / bs; j < A->rmap->range[i + 1] / bs; j++) recvcounts[i] += lens[j]; in MatGetSeqNonzeroStructure_MPIBAIJ()
2153 PetscCall(PetscArraycpy(b->ilen, lens, A->rmap->N / bs)); in MatGetSeqNonzeroStructure_MPIBAIJ()
2156 for (i = 1; i <= A->rmap->N / bs; i++) b->i[i] = b->i[i - 1] + lens[i - 1]; in MatGetSeqNonzeroStructure_MPIBAIJ()
2239 PetscInt ib, jb, bs = A->rmap->bs; in MatGetColumnReductions_MPIBAIJ()
2250 for (i = a_aij->i[0]; i < a_aij->i[aij->A->rmap->n / bs]; i++) { in MatGetColumnReductions_MPIBAIJ()
2258 for (i = b_aij->i[0]; i < b_aij->i[aij->B->rmap->n / bs]; i++) { in MatGetColumnReductions_MPIBAIJ()
2267 for (i = a_aij->i[0]; i < a_aij->i[aij->A->rmap->n / bs]; i++) { in MatGetColumnReductions_MPIBAIJ()
2275 for (i = b_aij->i[0]; i < b_aij->i[aij->B->rmap->n / bs]; i++) { in MatGetColumnReductions_MPIBAIJ()
2284 for (i = a_aij->i[0]; i < a_aij->i[aij->A->rmap->n / bs]; i++) { in MatGetColumnReductions_MPIBAIJ()
2293 for (i = b_aij->i[0]; i < b_aij->i[aij->B->rmap->n / bs]; i++) { in MatGetColumnReductions_MPIBAIJ()
2303 for (i = a_aij->i[0]; i < a_aij->i[aij->A->rmap->n / bs]; i++) { in MatGetColumnReductions_MPIBAIJ()
2311 for (i = b_aij->i[0]; i < b_aij->i[aij->B->rmap->n / bs]; i++) { in MatGetColumnReductions_MPIBAIJ()
2320 for (i = a_aij->i[0]; i < a_aij->i[aij->A->rmap->n / bs]; i++) { in MatGetColumnReductions_MPIBAIJ()
2328 for (i = b_aij->i[0]; i < b_aij->i[aij->B->rmap->n / bs]; i++) { in MatGetColumnReductions_MPIBAIJ()
2370 PetscCall(MatMPIBAIJSetPreallocation(Y, Y->rmap->bs, 1, NULL, 0, NULL)); in MatShift_MPIBAIJ()
2373 PetscCall(MatSeqBAIJSetPreallocation(maij->A, Y->rmap->bs, 1, NULL)); in MatShift_MPIBAIJ()
2555 PetscCall(PetscLayoutSetBlockSize(B->rmap, bs)); in MatMPIBAIJSetPreallocationCSR_MPIBAIJ()
2557 PetscCall(PetscLayoutSetUp(B->rmap)); in MatMPIBAIJSetPreallocationCSR_MPIBAIJ()
2559 PetscCall(PetscLayoutGetBlockSize(B->rmap, &bs)); in MatMPIBAIJSetPreallocationCSR_MPIBAIJ()
2560 m = B->rmap->n / bs; in MatMPIBAIJSetPreallocationCSR_MPIBAIJ()
2561 rstart = B->rmap->rstart / bs; in MatMPIBAIJSetPreallocationCSR_MPIBAIJ()
2666 PetscCall(PetscLayoutSetUp(B->rmap)); in MatMPIBAIJSetPreallocation_MPIBAIJ()
2668 PetscCall(PetscLayoutGetBlockSize(B->rmap, &bs)); in MatMPIBAIJSetPreallocation_MPIBAIJ()
2671 …for (i = 0; i < B->rmap->n / bs; i++) PetscCheck(d_nnz[i] >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUT… in MatMPIBAIJSetPreallocation_MPIBAIJ()
2674 …for (i = 0; i < B->rmap->n / bs; i++) PetscCheck(o_nnz[i] >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUT… in MatMPIBAIJSetPreallocation_MPIBAIJ()
2678 b->mbs = B->rmap->n / bs; in MatMPIBAIJSetPreallocation_MPIBAIJ()
2680 b->Mbs = B->rmap->N / bs; in MatMPIBAIJSetPreallocation_MPIBAIJ()
2683 for (i = 0; i <= b->size; i++) b->rangebs[i] = B->rmap->range[i] / bs; in MatMPIBAIJSetPreallocation_MPIBAIJ()
2684 b->rstartbs = B->rmap->rstart / bs; in MatMPIBAIJSetPreallocation_MPIBAIJ()
2685 b->rendbs = B->rmap->rend / bs; in MatMPIBAIJSetPreallocation_MPIBAIJ()
2703 …PetscCall(MatSetSizes(b->B, B->rmap->n, size > 1 ? B->cmap->N : 0, B->rmap->n, size > 1 ? B->cmap-… in MatMPIBAIJSetPreallocation_MPIBAIJ()
2710 PetscCall(MatSetSizes(b->A, B->rmap->n, B->cmap->n, B->rmap->n, B->cmap->n)); in MatMPIBAIJSetPreallocation_MPIBAIJ()
2729 …PetscInt M = B->rmap->n / B->rmap->bs, i, *ii, *jj, cnt, j, k, rstart = B->rmap->rstart / B… in MatConvert_MPIBAIJ_MPIAdj()
2759 …PetscCall(MatCreateMPIAdj(PetscObjectComm((PetscObject)B), M, B->cmap->N / B->rmap->bs, ii, jj, NU… in MatConvert_MPIBAIJ_MPIAdj()
2781 PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N)); in MatConvert_MPIBAIJ_MPIAIJ()
2782 PetscCall(MatSetBlockSizes(B, A->rmap->bs, A->cmap->bs)); in MatConvert_MPIBAIJ_MPIAIJ()
2804 bs = A->rmap->bs; in MatConvert_MPIBAIJ_MPIAIJ()
2805 nnz = bB->i[A->rmap->n]; in MatConvert_MPIBAIJ_MPIAIJ()
3156 PetscCall(MatSetSizes(mat, matin->rmap->n, matin->cmap->n, matin->rmap->N, matin->cmap->N)); in MatDuplicate_MPIBAIJ()
3159 PetscCall(PetscLayoutReference(matin->rmap, &mat->rmap)); in MatDuplicate_MPIBAIJ()
3170 mat->rmap->bs = matin->rmap->bs; in MatDuplicate_MPIBAIJ()
3214 …PetscCall(MatStashCreate_Private(PetscObjectComm((PetscObject)matin), matin->rmap->bs, &mat->bstas… in MatDuplicate_MPIBAIJ()
3249 if (mat->rmap->n < 0 && M == N) mat->rmap->n = mat->cmap->n; in MatLoad_MPIBAIJ_Binary()
3250 if (mat->cmap->n < 0 && M == N) mat->cmap->n = mat->rmap->n; in MatLoad_MPIBAIJ_Binary()
3252 if (mat->rmap->N < 0) mat->rmap->N = M; in MatLoad_MPIBAIJ_Binary()
3254 PetscCall(PetscLayoutSetUp(mat->rmap)); in MatLoad_MPIBAIJ_Binary()
3262 PetscCall(PetscLayoutGetRange(mat->rmap, &rs, NULL)); in MatLoad_MPIBAIJ_Binary()
3428 PetscInt cend = baij->cendbs, bs = mat->rmap->bs, bs2 = baij->bs2; in matmpibaijsetvaluesblocked_()