Lines Matching refs:mbs
101 PetscInt i, bs = A->rmap->bs, mbs = a->mbs, ipvt[5], bs2 = bs * bs, *v_pivots; in MatInvertBlockDiagonal_SeqBAIJ() local
115 if (!a->idiag) PetscCall(PetscMalloc1(bs2 * mbs, &a->idiag)); in MatInvertBlockDiagonal_SeqBAIJ()
121 for (i = 0; i < mbs; i++) { in MatInvertBlockDiagonal_SeqBAIJ()
138 for (i = 0; i < mbs; i++) { in MatInvertBlockDiagonal_SeqBAIJ()
150 for (i = 0; i < mbs; i++) { in MatInvertBlockDiagonal_SeqBAIJ()
167 for (i = 0; i < mbs; i++) { in MatInvertBlockDiagonal_SeqBAIJ()
176 for (i = 0; i < mbs; i++) { in MatInvertBlockDiagonal_SeqBAIJ()
185 for (i = 0; i < mbs; i++) { in MatInvertBlockDiagonal_SeqBAIJ()
194 for (i = 0; i < mbs; i++) { in MatInvertBlockDiagonal_SeqBAIJ()
204 for (i = 0; i < mbs; i++) { in MatInvertBlockDiagonal_SeqBAIJ()
224 PetscInt m = a->mbs, i, i2, nz, bs = A->rmap->bs, bs2 = bs * bs, k, j, idx, it; in MatSOR_SeqBAIJ()
549 idiag = a->idiag + bs2 * (a->mbs - 1); in MatSOR_SeqBAIJ()
1060 idiag = a->idiag + bs2 * (a->mbs - 1); in MatSOR_SeqBAIJ()
1416 PetscInt i, j, n = a->mbs, nz = a->i[n], *tia, *tja, bs = A->rmap->bs, k, l, cnt; in MatGetRowIJ_SeqBAIJ()
1647 …PetscInt i, j, k, *aj = a->j, *ai = a->i, bs = A->rmap->bs, mbs = a->mbs, nbs = a->nbs, *atfil… in MatTranspose_SeqBAIJ() local
1655 for (i = 0; i < ai[mbs]; i++) atfill[aj[i]] += 1; /* count num of non-zeros in row aj[i] */ in MatTranspose_SeqBAIJ()
1678 for (i = 0; i < mbs; i++) { in MatTranspose_SeqBAIJ()
1715 PetscCall(PetscArraycmp(a->i, b->i, a->mbs + 1, flg)); in MatCompare_SeqBAIJ_Private()
1775 for (cnt = 0, i = 0; i < A->mbs; i++) in MatView_SeqBAIJ_Binary()
1782 for (cnt = 0, i = 0; i < A->mbs; i++) in MatView_SeqBAIJ_Binary()
1792 for (cnt = 0, i = 0; i < A->mbs; i++) in MatView_SeqBAIJ_Binary()
1812 for (i = 0; i < a->mbs; i++) { in MatView_SeqBAIJ_ASCII_structonly()
1847 for (i = 0; i < a->mbs; i++) { in MatView_SeqBAIJ_ASCII()
1871 for (i = 0; i < a->mbs; i++) { in MatView_SeqBAIJ_ASCII()
1903 PetscInt row, i, j, k, l, mbs = a->mbs, bs = A->rmap->bs, bs2 = a->bs2; in MatView_SeqBAIJ_Draw_Zoom() local
1921 for (i = 0, row = 0; i < mbs; i++, row += bs) { in MatView_SeqBAIJ_Draw_Zoom()
1937 for (i = 0, row = 0; i < mbs; i++, row += bs) { in MatView_SeqBAIJ_Draw_Zoom()
1953 for (i = 0, row = 0; i < mbs; i++, row += bs) { in MatView_SeqBAIJ_Draw_Zoom()
1983 for (i = 0, row = 0; i < mbs; i++, row += bs) { in MatView_SeqBAIJ_Draw_Zoom()
2124 …heck(row < a->mbs, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Block row index too large %" PetscI… in MatSetValuesBlocked_SeqBAIJ()
2184 …MatSeqXAIJReallocateAIJ_structure_only(A, a->mbs, bs2, nrow, row, col, rmax, ai, aj, rp, imax, non… in MatSetValuesBlocked_SeqBAIJ()
2186 …MatSeqXAIJReallocateAIJ(A, a->mbs, bs2, nrow, row, col, rmax, aa, ai, aj, rp, ap, imax, nonew, Mat… in MatSetValuesBlocked_SeqBAIJ()
2219 PetscInt mbs = a->mbs, bs2 = a->bs2, rmax = 0; in MatAssemblyEnd_SeqBAIJ() local
2227 for (i = 1; i < mbs; i++) { in MatAssemblyEnd_SeqBAIJ()
2240 if (mbs) { in MatAssemblyEnd_SeqBAIJ()
2241 fshift += imax[mbs - 1] - ailen[mbs - 1]; in MatAssemblyEnd_SeqBAIJ()
2242 ai[mbs] = ai[mbs - 1] + ailen[mbs - 1]; in MatAssemblyEnd_SeqBAIJ()
2250 for (i = 0; i < mbs; i++) { in MatAssemblyEnd_SeqBAIJ()
2255 a->nz = ai[mbs]; in MatAssemblyEnd_SeqBAIJ()
2269 …e_only) PetscCall(MatCheckCompressedRow(A, a->nonzerorowcnt, &a->compressedrow, a->i, mbs, ratio)); in MatAssemblyEnd_SeqBAIJ()
2501 …MatSeqXAIJReallocateAIJ_structure_only(A, a->mbs, bs2, nrow, brow, bcol, rmax, ai, aj, rp, imax, n… in MatSetValues_SeqBAIJ()
2503 …MatSeqXAIJReallocateAIJ(A, a->mbs, bs2, nrow, brow, bcol, rmax, aa, ai, aj, rp, ap, imax, nonew, M… in MatSetValues_SeqBAIJ()
2565 PetscCall(PetscArraycpy(baij->ilen, baij->imax, baij->mbs)); in MatSeqBAIJSetColumnIndices_SeqBAIJ()
2602 PetscInt i, j, n, row, bs, *ai, *aj, mbs; in MatGetRowMaxAbs_SeqBAIJ() local
2614 mbs = a->mbs; in MatGetRowMaxAbs_SeqBAIJ()
2620 for (i = 0; i < mbs; i++) { in MatGetRowMaxAbs_SeqBAIJ()
2646 PetscInt i, j, n, row, bs, *ai, mbs; in MatGetRowSumAbs_SeqBAIJ() local
2657 mbs = a->mbs; in MatGetRowSumAbs_SeqBAIJ()
2663 for (i = 0; i < mbs; i++) { in MatGetRowSumAbs_SeqBAIJ()
2689 … PetscInt ambs = a->mbs, bmbs = b->mbs, abs = A->rmap->bs, bbs = B->rmap->bs, bs2 = abs * abs; in MatCopy_SeqBAIJ()
2719 PetscInt bs = Y->rmap->bs, mbs = Y->rmap->N / bs; in MatAXPYGetPreallocation_SeqBAIJ() local
2725 PetscCall(MatAXPYGetPreallocation_SeqX_private(mbs, x->i, x->j, y->i, y->j, nnz)); in MatAXPYGetPreallocation_SeqBAIJ()
2737 … PetscBool e = x->nz == y->nz && x->mbs == y->mbs && bs == X->rmap->bs ? PETSC_TRUE : PETSC_FALSE; in MatAXPY_SeqBAIJ()
2739 PetscCall(PetscArraycmp(x->i, y->i, x->mbs + 1, &e)); in MatAXPY_SeqBAIJ()
2741 PetscCall(PetscArraycmp(x->j, y->j, x->i[x->mbs], &e)); in MatAXPY_SeqBAIJ()
2777 PetscInt i, nz = a->bs2 * a->i[a->mbs]; in MatConjugate_SeqBAIJ()
2789 PetscInt i, nz = a->bs2 * a->i[a->mbs]; in MatRealPart_SeqBAIJ()
2805 PetscInt i, nz = a->bs2 * a->i[a->mbs]; in MatImaginaryPart_SeqBAIJ()
2870 PetscInt i, *collengths, *cia, *cja, n = a->nbs, m = a->mbs; in MatGetColumnIJ_SeqBAIJ_Color()
2926 PetscInt mbs = a->mbs, bs2 = a->bs2, rmax = 0; in MatEliminateZeros_SeqBAIJ() local
2933 for (i = 1; i <= mbs; i++) { in MatEliminateZeros_SeqBAIJ()
2954 if (mbs) { in MatEliminateZeros_SeqBAIJ()
2955 ai[mbs] -= fshift; in MatEliminateZeros_SeqBAIJ()
2956 a->nz = ai[mbs]; in MatEliminateZeros_SeqBAIJ()
3116 PetscInt nz = aij->i[aij->mbs] * aij->bs2; in MatStoreValues_SeqBAIJ()
3132 PetscInt nz = aij->i[aij->mbs] * aij->bs2; in MatRetrieveValues_SeqBAIJ()
3149 PetscInt i, mbs, nbs, bs2; in MatSeqBAIJSetPreallocation_SeqBAIJ() local
3176 mbs = B->rmap->n / bs; in MatSeqBAIJSetPreallocation_SeqBAIJ()
3180 …PetscCheck(mbs * bs == B->rmap->n && nbs * bs == B->cmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "… in MatSeqBAIJSetPreallocation_SeqBAIJ()
3185 for (i = 0; i < mbs; i++) { in MatSeqBAIJSetPreallocation_SeqBAIJ()
3313 b->mbs = mbs; in MatSeqBAIJSetPreallocation_SeqBAIJ()
3317 PetscCall(PetscMalloc2(mbs, &b->imax, mbs, &b->ilen)); in MatSeqBAIJSetPreallocation_SeqBAIJ()
3322 for (i = 0; i < mbs; i++) b->ilen[i] = 0; in MatSeqBAIJSetPreallocation_SeqBAIJ()
3327 for (i = 0; i < mbs; i++) b->imax[i] = nz; in MatSeqBAIJSetPreallocation_SeqBAIJ()
3328 PetscCall(PetscIntMultError(nz, mbs, &nz)); in MatSeqBAIJSetPreallocation_SeqBAIJ()
3331 for (i = 0; i < mbs; i++) { in MatSeqBAIJSetPreallocation_SeqBAIJ()
3355 for (i = 1; i < mbs + 1; i++) b->i[i] = b->i[i - 1] + b->imax[i - 1]; in MatSeqBAIJSetPreallocation_SeqBAIJ()
3362 b->mbs = mbs; in MatSeqBAIJSetPreallocation_SeqBAIJ()
3531 PetscInt i, mbs = a->mbs, nz = a->nz, bs2 = a->bs2; in MatDuplicateNoCreate_SeqBAIJ() local
3535 PetscCheck(a->i[mbs] == nz, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Corrupt matrix"); in MatDuplicateNoCreate_SeqBAIJ()
3542 PetscCall(PetscMalloc2(mbs, &c->imax, mbs, &c->ilen)); in MatDuplicateNoCreate_SeqBAIJ()
3543 for (i = 0; i < mbs; i++) { in MatDuplicateNoCreate_SeqBAIJ()
3569 PetscCall(PetscShmgetAllocateArray(mbs + 1, sizeof(PetscInt), (void **)&c->i)); in MatDuplicateNoCreate_SeqBAIJ()
3573 PetscCall(PetscArraycpy(c->i, a->i, mbs + 1)); in MatDuplicateNoCreate_SeqBAIJ()
3574 if (mbs > 0) { in MatDuplicateNoCreate_SeqBAIJ()
3594 c->mbs = a->mbs; in MatDuplicateNoCreate_SeqBAIJ()
3635 PetscInt header[4], M, N, nz, bs, m, n, mbs, nbs, rows, cols, sum, i, j, k; in MatLoad_SeqBAIJ_Binary() local
3667 mbs = m / bs; in MatLoad_SeqBAIJ_Binary()
3689 PetscCall(PetscCalloc1(mbs, &nnz)); in MatLoad_SeqBAIJ_Binary()
3691 for (i = 0; i < mbs; i++) { in MatLoad_SeqBAIJ_Binary()