Home
last modified time | relevance | path

Searched refs:mbs (Results 1 – 25 of 90) sorted by relevance

1234

/petsc/src/mat/impls/sbaij/seq/
H A Dsro.c34 const PetscInt mbs = a->mbs; in MatReorderingSeqSBAIJ() local
37 if (!mbs) PetscFunctionReturn(PETSC_SUCCESS); in MatReorderingSeqSBAIJ()
49 …for (i=0; i<mbs; i++) PetscCheck(rip[i] == riip[i],PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Non-symme… in MatReorderingSeqSBAIJ()
54 PetscCall(PetscMalloc2(mbs+1,&ai, 2*a->i[mbs],&aj)); in MatReorderingSeqSBAIJ()
58 PetscCall(PetscArraycpy(ai,a->i,mbs+1)); in MatReorderingSeqSBAIJ()
59 PetscCall(PetscArraycpy(aj,a->j,a->i[mbs])); in MatReorderingSeqSBAIJ()
68 PetscCall(PetscMalloc1(mbs,&nzr)); in MatReorderingSeqSBAIJ()
69 PetscCall(PetscMalloc1(ai[mbs],&r)); in MatReorderingSeqSBAIJ()
70 for (i=0; i<mbs; i++) nzr[i] = 0; in MatReorderingSeqSBAIJ()
71 for (i=0; i<ai[mbs]; i++) r[i] = 0; in MatReorderingSeqSBAIJ()
[all …]
H A Dsbaijfact.c10 PetscInt mbs = fact->mbs, bs = F->rmap->bs, i, nneg_tmp, npos_tmp, *fi = fact->diag; in MatGetInertia_SeqSBAIJ() local
18 for (i = 0; i < mbs; i++) { in MatGetInertia_SeqSBAIJ()
24 for (i = 0; i < mbs; i++) { in MatGetInertia_SeqSBAIJ()
31 if (nzero) *nzero = mbs - nneg_tmp - npos_tmp; in MatGetInertia_SeqSBAIJ()
43 PetscInt i, mbs = a->mbs, *jutmp, bs = A->rmap->bs, bs2 = a->bs2; in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR() local
69 PetscCall(PetscShmgetAllocateArray(mbs + 1, sizeof(PetscInt), (void **)&iu)); in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
70 umax = (PetscInt)(f * ai[mbs] + 1); in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
71 umax += mbs + 1; in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
73 iu[0] = mbs + 1; in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
74 juidx = mbs + 1; /* index for ju */ in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
[all …]
H A Dsbaijfact2.c13 PetscInt mbs = a->mbs, *ai = a->i, *aj = a->j; in MatSolve_SeqSBAIJ_N_inplace() local
31 for (k = 0; k < mbs; k++) { /* t <- perm(b) */ in MatSolve_SeqSBAIJ_N_inplace()
35 for (k = 0; k < mbs; k++) { in MatSolve_SeqSBAIJ_N_inplace()
55 for (k = mbs - 1; k >= 0; k--) { in MatSolve_SeqSBAIJ_N_inplace()
76 PetscCall(PetscLogFlops(4.0 * bs2 * a->nz - (bs + 2.0 * bs2) * mbs)); in MatSolve_SeqSBAIJ_N_inplace()
92 …ring(const PetscInt *ai, const PetscInt *aj, const MatScalar *aa, PetscInt mbs, PetscInt bs, Petsc… in MatForwardSolve_SeqSBAIJ_N_NaturalOrdering() argument
102 for (k = 0; k < mbs; k++) { in MatForwardSolve_SeqSBAIJ_N_NaturalOrdering()
124 …ring(const PetscInt *ai, const PetscInt *aj, const MatScalar *aa, PetscInt mbs, PetscInt bs, Petsc… in MatBackwardSolve_SeqSBAIJ_N_NaturalOrdering() argument
132 for (k = mbs - 1; k >= 0; k--) { in MatBackwardSolve_SeqSBAIJ_N_NaturalOrdering()
152 const PetscInt mbs = a->mbs, *ai = a->i, *aj = a->j; in MatSolve_SeqSBAIJ_N_NaturalOrdering_inplace() local
[all …]
H A Dsbaijfact3.c9 const PetscInt *ai, *aj, *perm_ptr, mbs = a->mbs, *bi = b->i, *bj = b->j; in MatCholeskyFactorNumeric_SeqSBAIJ_3() local
19 PetscCall(PetscCalloc1(9 * mbs, &rtmp)); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
20 PetscCall(PetscMalloc2(mbs, &il, mbs, &jl)); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
22 for (i = 0; i < mbs; i++) jl[i] = mbs; in MatCholeskyFactorNumeric_SeqSBAIJ_3()
35 PetscCall(PetscMalloc1(9 * ai[mbs], &aa)); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
36 PetscCall(PetscArraycpy(aa, a->a, 9 * ai[mbs])); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
37 PetscCall(PetscMalloc1(ai[mbs], &a2anew)); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
38 PetscCall(PetscArraycpy(a2anew, a->a2anew, ai[mbs])); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
40 for (i = 0; i < mbs; i++) { in MatCholeskyFactorNumeric_SeqSBAIJ_3()
69 for (k = 0; k < mbs; k++) { in MatCholeskyFactorNumeric_SeqSBAIJ_3()
[all …]
H A Daijsbaij.c11 PetscInt bs = A->rmap->bs, bs2 = bs * bs, mbs = A->rmap->N / bs, diagcnt = 0; in MatConvert_SeqSBAIJ_SeqAIJ() local
19 for (i = 0; i < mbs; i++) rowlengths[i * bs] = 0; in MatConvert_SeqSBAIJ_SeqAIJ()
21 for (i = 0; i < mbs; i++) { in MatConvert_SeqSBAIJ_SeqAIJ()
56 for (i = 0; i < mbs; i++) { in MatConvert_SeqSBAIJ_SeqAIJ()
63 …cCheck(bi[mbs] == 2 * a->nz - diagcnt, PETSC_COMM_SELF, PETSC_ERR_PLIB, "bi[mbs]: %" PetscInt_FMT … in MatConvert_SeqSBAIJ_SeqAIJ()
68 for (i = 0; i < mbs; i++) { in MatConvert_SeqSBAIJ_SeqAIJ()
241 PetscInt bs = A->rmap->bs, bs2 = bs * bs, mbs = m / bs; in MatConvert_SeqSBAIJ_SeqBAIJ() local
247 PetscCall(PetscMalloc2(mbs, &browlengths, mbs, &browstart)); in MatConvert_SeqSBAIJ_SeqBAIJ()
248 for (PetscInt i = 0; i < mbs; i++) browlengths[i] = 0; in MatConvert_SeqSBAIJ_SeqBAIJ()
249 for (PetscInt i = 0; i < mbs; i++) { in MatConvert_SeqSBAIJ_SeqBAIJ()
[all …]
H A Dsbaijfact6.c9 const PetscInt *ai, *aj, *perm_ptr, mbs = a->mbs, *bi = b->i, *bj = b->j; in MatCholeskyFactorNumeric_SeqSBAIJ_4() local
20 PetscCall(PetscCalloc1(16 * mbs, &rtmp)); in MatCholeskyFactorNumeric_SeqSBAIJ_4()
21 PetscCall(PetscMalloc2(mbs, &il, mbs, &jl)); in MatCholeskyFactorNumeric_SeqSBAIJ_4()
23 for (i = 0; i < mbs; i++) jl[i] = mbs; in MatCholeskyFactorNumeric_SeqSBAIJ_4()
36 PetscCall(PetscMalloc1(16 * ai[mbs], &aa)); in MatCholeskyFactorNumeric_SeqSBAIJ_4()
37 PetscCall(PetscArraycpy(aa, a->a, 16 * ai[mbs])); in MatCholeskyFactorNumeric_SeqSBAIJ_4()
38 PetscCall(PetscMalloc1(ai[mbs], &a2anew)); in MatCholeskyFactorNumeric_SeqSBAIJ_4()
39 PetscCall(PetscArraycpy(a2anew, a->a2anew, ai[mbs])); in MatCholeskyFactorNumeric_SeqSBAIJ_4()
41 for (i = 0; i < mbs; i++) { in MatCholeskyFactorNumeric_SeqSBAIJ_4()
70 for (k = 0; k < mbs; k++) { in MatCholeskyFactorNumeric_SeqSBAIJ_4()
[all …]
H A Dsbaijfact4.c10 PetscInt i, j, mbs = a->mbs, *bi = b->i, *bj = b->j; in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering() local
20 PetscCall(PetscCalloc1(9 * mbs, &rtmp)); in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
21 PetscCall(PetscMalloc2(mbs, &il, mbs, &jl)); in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
23 for (i = 0; i < mbs; i++) jl[i] = mbs; in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
31 for (k = 0; k < mbs; k++) { in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
48 while (i < mbs) { in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
160 PetscCall(PetscLogFlops(1.3333 * 27 * b->mbs)); /* from inverting diagonal blocks */ in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
H A Dsbaij2.c11 PetscInt brow, i, j, k, l, mbs, n, *nidx, isz, bcol, bcol_max, start, end, *ai, *aj, bs; in MatIncreaseOverlap_SeqSBAIJ() local
17 mbs = a->mbs; in MatIncreaseOverlap_SeqSBAIJ()
21 PetscCall(PetscBTCreate(mbs, &table_out)); in MatIncreaseOverlap_SeqSBAIJ()
22 PetscCall(PetscMalloc1(mbs + 1, &nidx)); in MatIncreaseOverlap_SeqSBAIJ()
23 PetscCall(PetscBTCreate(mbs, &table_in)); in MatIncreaseOverlap_SeqSBAIJ()
27 PetscCall(PetscBTMemzero(mbs, table_out)); in MatIncreaseOverlap_SeqSBAIJ()
37 … PetscCheck(brow < mbs, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "index greater than mat-dim"); in MatIncreaseOverlap_SeqSBAIJ()
49 PetscCall(PetscBTMemzero(mbs, table_in)); in MatIncreaseOverlap_SeqSBAIJ()
53 for (brow = 0; brow < mbs; brow++) { in MatIncreaseOverlap_SeqSBAIJ()
142 …PetscCheck(c->mbs == nrows && c->nbs == ncols && (*B)->rmap->bs == bs, PETSC_COMM_SELF, PETSC_ERR_… in MatCreateSubMatrix_SeqSBAIJ_Private()
[all …]
H A Dsbaij.c41 PetscInt i, j, n = a->mbs, nz = a->i[n], *tia, *tja, bs = A->rmap->bs, k, l, cnt; in MatGetRowIJ_SeqSBAIJ()
322 for (i = 0; i < a->mbs; i++) { /* for row block i */ in MatView_SeqSBAIJ_ASCII()
354 for (i = 0; i < a->mbs; i++) { /* for row block i */ in MatView_SeqSBAIJ_ASCII()
387 PetscInt row, i, j, k, l, mbs = a->mbs, bs = A->rmap->bs, bs2 = a->bs2; in MatView_SeqSBAIJ_Draw_Zoom() local
403 for (i = 0, row = 0; i < mbs; i++, row += bs) { in MatView_SeqSBAIJ_Draw_Zoom()
419 for (i = 0, row = 0; i < mbs; i++, row += bs) { in MatView_SeqSBAIJ_Draw_Zoom()
435 for (i = 0, row = 0; i < mbs; i++, row += bs) { in MatView_SeqSBAIJ_Draw_Zoom()
591 …heck(row < a->mbs, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Block index row too large %" PetscI… in MatSetValuesBlocked_SeqSBAIJ()
648 …MatSeqXAIJReallocateAIJ(A, a->mbs, bs2, nrow, row, col, rmax, aa, ai, aj, rp, ap, imax, nonew, Mat… in MatSetValuesBlocked_SeqSBAIJ()
679 PetscInt mbs = a->mbs, bs2 = a->bs2, rmax = 0; in MatAssemblyEnd_SeqSBAIJ() local
[all …]
H A Dsbaijfact7.c9 const PetscInt *ai, *aj, *perm_ptr, mbs = a->mbs, *bi = b->i, *bj = b->j; in MatCholeskyFactorNumeric_SeqSBAIJ_5() local
19 PetscCall(PetscCalloc1(25 * mbs, &rtmp)); in MatCholeskyFactorNumeric_SeqSBAIJ_5()
20 PetscCall(PetscMalloc2(mbs, &il, mbs, &jl)); in MatCholeskyFactorNumeric_SeqSBAIJ_5()
22 for (i = 0; i < mbs; i++) jl[i] = mbs; in MatCholeskyFactorNumeric_SeqSBAIJ_5()
35 PetscCall(PetscMalloc1(25 * ai[mbs], &aa)); in MatCholeskyFactorNumeric_SeqSBAIJ_5()
36 PetscCall(PetscArraycpy(aa, a->a, 25 * ai[mbs])); in MatCholeskyFactorNumeric_SeqSBAIJ_5()
37 PetscCall(PetscMalloc1(ai[mbs], &a2anew)); in MatCholeskyFactorNumeric_SeqSBAIJ_5()
38 PetscCall(PetscArraycpy(a2anew, a->a2anew, ai[mbs])); in MatCholeskyFactorNumeric_SeqSBAIJ_5()
40 for (i = 0; i < mbs; i++) { in MatCholeskyFactorNumeric_SeqSBAIJ_5()
69 for (k = 0; k < mbs; k++) { in MatCholeskyFactorNumeric_SeqSBAIJ_5()
[all …]
H A Dsbaijfact5.c10 PetscInt i, j, mbs = a->mbs, *bi = b->i, *bj = b->j; in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering() local
21 PetscCall(PetscCalloc1(16 * mbs, &rtmp)); in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
22 PetscCall(PetscMalloc2(mbs, &il, mbs, &jl)); in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
24 for (i = 0; i < mbs; i++) jl[i] = mbs; in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
32 for (k = 0; k < mbs; k++) { in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
49 while (i < mbs) { in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
189 PetscCall(PetscLogFlops(1.3333 * 64 * b->mbs)); /* from inverting diagonal blocks */ in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
H A Dsbaijfact8.c10 PetscInt i, j, mbs = a->mbs, *bi = b->i, *bj = b->j; in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering() local
20 PetscCall(PetscCalloc1(25 * mbs, &rtmp)); in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering()
21 PetscCall(PetscMalloc2(mbs, &il, mbs, &jl)); in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering()
23 for (i = 0; i < mbs; i++) jl[i] = mbs; in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering()
31 for (k = 0; k < mbs; k++) { in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering()
48 while (i < mbs) { in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering()
213 PetscCall(PetscLogFlops(1.3333 * 125 * b->mbs)); /* from inverting diagonal blocks */ in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering()
/petsc/src/mat/utils/
H A Dcompressedrow.c30 …t A, PetscInt nrows, Mat_CompressedRow *compressedrow, PetscInt *ai, PetscInt mbs, PetscReal ratio) in MatCheckCompressedRow() argument
39 nrows = mbs - nrows; in MatCheckCompressedRow()
42 if (nrows < ratio * mbs) { in MatCheckCompressedRow()
45 …alrows %" PetscInt_FMT ") < %g. Do not use CompressedRow routines.\n", nrows, mbs, (double)ratio)); in MatCheckCompressedRow()
49 …num_localrows %" PetscInt_FMT ") > %g. Use CompressedRow routines.\n", nrows, mbs, (double)ratio)); in MatCheckCompressedRow()
52 nrows = mbs - nrows; /* num of non-zero rows */ in MatCheckCompressedRow()
56 for (i = 0; i < mbs; i++) { in MatCheckCompressedRow()
/petsc/src/mat/impls/sbaij/mpi/
H A Dmmsbaij.c11 PetscInt bs = mat->rmap->bs, *stmp, mbs = sbaij->mbs, vec_size, nt; in MatSetUpMultiply_MPISBAIJ() local
29 PetscCall(PetscHMapICreateWithSize(mbs, &gid1_lid1)); in MatSetUpMultiply_MPISBAIJ()
30 for (i = 0; i < B->mbs; i++) { in MatSetUpMultiply_MPISBAIJ()
52 for (i = 0; i < B->mbs; i++) { in MatSetUpMultiply_MPISBAIJ()
70 for (i = 0; i < mbs; i++) { in MatSetUpMultiply_MPISBAIJ()
96 for (i = 0; i < mbs; i++) { in MatSetUpMultiply_MPISBAIJ()
127 …PetscCall(VecCreateMPI(PetscObjectComm((PetscObject)mat), (mbs + ec) * bs, PETSC_DETERMINE, &sbaij… in MatSetUpMultiply_MPISBAIJ()
139 k = sowners[rank] / bs + mbs; in MatSetUpMultiply_MPISBAIJ()
144 k = sowners[rank] / bs + mbs; in MatSetUpMultiply_MPISBAIJ()
156 PetscCall(VecCreateSeqWithArray(PETSC_COMM_SELF, 1, bs * mbs, ptr, &sbaij->slvec1a)); in MatSetUpMultiply_MPISBAIJ()
[all …]
/petsc/src/mat/impls/aij/seq/
H A Dfdaij.c64 PetscInt i, j, nrows, nbcols, brows = c->brows, bcols = c->bcols, mbs = c->m, nis = c->ncolors; in MatFDColoringSetUpBlocked_AIJ_Private() local
68 if (brows < 1 || brows > mbs) brows = mbs; in MatFDColoringSetUpBlocked_AIJ_Private()
94 if (row_end > mbs) row_end = mbs; in MatFDColoringSetUpBlocked_AIJ_Private()
96 while (row_end <= mbs) { /* loop over block rows */ in MatFDColoringSetUpBlocked_AIJ_Private()
105 Jentry_new[nz_new].row = Jentry[nz].row + j * mbs; /* index in dy-array */ in MatFDColoringSetUpBlocked_AIJ_Private()
114 if (row_end == mbs) break; in MatFDColoringSetUpBlocked_AIJ_Private()
116 if (row_end > mbs) row_end = mbs; in MatFDColoringSetUpBlocked_AIJ_Private()
139 if (row_end > mbs) row_end = mbs; in MatFDColoringSetUpBlocked_AIJ_Private()
141 while (row_end <= mbs) { /* loop over block rows */ in MatFDColoringSetUpBlocked_AIJ_Private()
150 Jentry2_new[nz_new].row = Jentry2[nz].row + j * mbs; /* index in dy-array */ in MatFDColoringSetUpBlocked_AIJ_Private()
[all …]
/petsc/src/ksp/pc/impls/pbjacobi/cuda/
H A Dpbjacobi_cuda.cu6 __global__ static void MatMultBatched(PetscInt bs, PetscInt mbs, const PetscScalar *A, const PetscS… in MatMultBatched() argument
13 for (; row < bs * mbs; row += gridSize) { in MatMultBatched()
40 const PetscInt bs = jac->bs, mbs = jac->mbs; in PCApplyOrTranspose_PBJacobi_CUDA() local
51 …XgemvStridedBatched(handle, op, bs, bs, &alpha, A, bs, bs * bs, xx, 1, bs, &beta, yy, 1, bs, mbs)); in PCApplyOrTranspose_PBJacobi_CUDA()
53 PetscInt gridSize = PetscMin((bs * mbs + 255) / 256, 2147483647); /* <= 2^31-1 */ in PCApplyOrTranspose_PBJacobi_CUDA()
54 …MatMultBatched<<<gridSize, 256>>>(bs, mbs, A, xx, yy, op == CUBLAS_OP_T ? PETSC_TRUE : PETSC_FALSE… in PCApplyOrTranspose_PBJacobi_CUDA()
59 PetscCall(PetscLogGpuFlops(bs * bs * mbs * 2)); in PCApplyOrTranspose_PBJacobi_CUDA()
94 size = sizeof(PetscScalar) * jac->bs * jac->bs * jac->mbs; in PCSetUp_PBJacobi_CUDA()
/petsc/src/mat/impls/baij/seq/
H A Dbaij2.c22 m = a->mbs; in MatIncreaseOverlap_SeqBAIJ()
105 …PetscCheck(c->mbs == nrows && c->nbs == ncols && (*B)->rmap->bs == bs, PETSC_COMM_SELF, PETSC_ERR_… in MatCreateSubMatrix_SeqBAIJ_Private()
106 PetscCall(PetscArraycmp(c->ilen, lens, c->mbs, &flag)); in MatCreateSubMatrix_SeqBAIJ_Private()
108 PetscCall(PetscArrayzero(c->ilen, c->mbs)); in MatCreateSubMatrix_SeqBAIJ_Private()
242 PetscInt mbs, i, n; in MatMult_SeqBAIJ_1() local
251 mbs = a->compressedrow.nrows; in MatMult_SeqBAIJ_1()
254 PetscCall(PetscArrayzero(z, a->mbs)); in MatMult_SeqBAIJ_1()
256 mbs = a->mbs; in MatMult_SeqBAIJ_1()
260 for (i = 0; i < mbs; i++) { in MatMult_SeqBAIJ_1()
288 PetscInt mbs, i, *idx, *ii, j, n, *ridx = NULL; in MatMult_SeqBAIJ_2() local
[all …]
H A Dbaijfact.c13 const PetscInt n = a->mbs, *ai = a->i, *aj = a->j, *bi = b->i, *bj = b->j, bs2 = a->bs2; in MatLUFactorNumeric_SeqBAIJ_2()
122 const PetscInt n = a->mbs, *ai = a->i, *aj = a->j, *bi = b->i, *bj = b->j, bs2 = a->bs2; in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
229 PetscInt i, j, n = a->mbs, *bi = b->i, *bj = b->j; in MatILUFactorNumeric_SeqBAIJ_2_inplace()
331 PetscCall(PetscLogFlops(1.333333333333 * 8 * b->mbs)); /* from inverting diagonal blocks */ in MatILUFactorNumeric_SeqBAIJ_2_inplace()
340 PetscInt i, j, n = a->mbs, *bi = b->i, *bj = b->j; in MatILUFactorNumeric_SeqBAIJ_2_NaturalOrdering_inplace()
442 PetscCall(PetscLogFlops(1.333333333333 * 8 * b->mbs)); /* from inverting diagonal blocks */ in MatILUFactorNumeric_SeqBAIJ_2_NaturalOrdering_inplace()
455 const PetscInt n = a->mbs, *ai = a->i, *aj = a->j, *bi = b->i, *bj = b->j, *bdiag = b->diag; in MatLUFactorNumeric_SeqBAIJ_1()
618 PetscInt i, j, n = a->mbs, *bi = b->i, *bj = b->j; in MatILUFactorNumeric_SeqBAIJ_1_inplace()
753 PetscInt i, j, mbs = a->mbs, bs = A->rmap->bs, *bi = b->i, *bj = b->j, *bcol; in MatCholeskyFactorNumeric_SeqBAIJ_N() local
776 PetscCall(PetscMalloc3(mbs, &rtmp, mbs, &il, mbs, &jl)); in MatCholeskyFactorNumeric_SeqBAIJ_N()
[all …]
H A Dbaij.c101 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()
[all …]
/petsc/src/mat/tests/
H A Dex141.c9 PetscInt i, bs = 2, mbs, m, block, d_nz = 6, col[3]; in main() local
32 mbs = 8; in main()
33 PetscCall(PetscOptionsGetInt(NULL, NULL, "-mbs", &mbs, NULL)); in main()
34 m = mbs * bs; in main()
43 for (block = 0; block < mbs; block++) { in main()
75 for (block = 0; block < mbs - 1; block++) { in main()
H A Dex77.c10 PetscInt n, mbs = 16, bs = 1, nz = 3, prob = 2, i, j, col[3], row, Ii, J, n1; in main() local
25 PetscCall(PetscOptionsGetInt(NULL, NULL, "-mbs", &mbs, NULL)); in main()
27 n = mbs * bs; in main()
231 PetscCall(PetscMalloc1(mbs, &nip_ptr)); in main()
233 PetscCall(PetscArraycpy(nip_ptr, ip_ptr, mbs)); in main()
235 nip_ptr[1] = nip_ptr[mbs - 2]; in main()
236 nip_ptr[mbs - 2] = i; in main()
238 nip_ptr[0] = nip_ptr[mbs - 1]; in main()
239 nip_ptr[mbs - 1] = i; in main()
241 PetscCall(ISCreateGeneral(PETSC_COMM_SELF, mbs, nip_ptr, PETSC_COPY_VALUES, &nip)); in main()
H A Dex55.c9 PetscInt i, j, ntypes, bs, mbs, m, block, d_nz = 6, o_nz = 3, col[3], row, verbose = 0; in main() local
61 mbs = 8; in main()
62 PetscCall(PetscOptionsGetInt(NULL, NULL, "-mbs", &mbs, NULL)); in main()
65 m = mbs * bs; in main()
67 for (block = 0; block < mbs; block++) { in main()
94 for (i = 0; i < (mbs - 1) * bs; i++) { in main()
/petsc/src/ksp/pc/impls/pbjacobi/kokkos/
H A Dpbjacobi_kok.kokkos.cxx38 const PetscInt bs = jac->bs, mbs = jac->mbs, bs2 = bs * bs; in PCApplyOrTranspose_PBJacobi_Kokkos() local
48 …label, Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, bs * mbs), KOKKOS_LAMBDA(PetscInt … in PCApplyOrTranspose_PBJacobi_Kokkos()
67 PetscCall(PetscLogGpuFlops(bs * bs * mbs * 2)); /* FMA on entries in all blocks */ in PCApplyOrTranspose_PBJacobi_Kokkos()
89 len = jac->bs * jac->bs * jac->mbs; in PCSetUp_PBJacobi_Kokkos()
/petsc/src/mat/impls/baij/mpi/
H A Dmmbaij.c26 PetscCall(PetscHMapICreateWithSize(B->mbs, &gid1_lid1)); in MatSetUpMultiply_MPIBAIJ()
27 for (i = 0; i < B->mbs; i++) { in MatSetUpMultiply_MPIBAIJ()
52 for (i = 0; i < B->mbs; i++) { in MatSetUpMultiply_MPIBAIJ()
68 for (i = 0; i < B->mbs; i++) { in MatSetUpMultiply_MPIBAIJ()
86 for (i = 0; i < B->mbs; i++) { in MatSetUpMultiply_MPIBAIJ()
133 PetscInt i, j, mbs, n = A->cmap->N, col, *garray = baij->garray; in MatDisAssemble_MPIBAIJ() local
154 mbs = Bbaij->mbs; in MatDisAssemble_MPIBAIJ()
158 PetscCall(PetscMalloc1(mbs, &nz)); in MatDisAssemble_MPIBAIJ()
159 for (i = 0; i < mbs; i++) nz[i] = Bbaij->i[i + 1] - Bbaij->i[i]; in MatDisAssemble_MPIBAIJ()
174 for (i = 0; i < mbs; i++) { in MatDisAssemble_MPIBAIJ()
/petsc/src/mat/impls/shell/
H A Dshellcnv.c10 PetscInt cst, cen, Nbs, mbs, nbs, rbs, cbs; in MatConvert_Shell() local
26 mbs = m / rbs; in MatConvert_Shell()
30 PetscCall(PetscMalloc4(mbs, &dnnz, mbs, &onnz, mbs, &dnnzu, mbs, &onnzu)); in MatConvert_Shell()
31 for (i = 0; i < mbs; i++) { in MatConvert_Shell()

1234