Home
last modified time | relevance | path

Searched refs:bs2 (Results 1 – 25 of 61) sorted by relevance

123

/petsc/src/mat/impls/baij/seq/
H A Dbaijfact4.c15 PetscInt diag, bs = A->rmap->bs, bs2 = a->bs2, *pj, *v_pivots; in MatILUFactorNumeric_SeqBAIJ_N_inplace() local
28 PetscCall(PetscCalloc1(bs2 * (n + 1), &rtmp)); in MatILUFactorNumeric_SeqBAIJ_N_inplace()
30 PetscCall(PetscMalloc3(bs, &v_work, bs2, &multiplier, bs, &v_pivots)); in MatILUFactorNumeric_SeqBAIJ_N_inplace()
35 for (j = 0; j < nz; j++) PetscCall(PetscArrayzero(rtmp + bs2 * ajtmp[j], bs2)); in MatILUFactorNumeric_SeqBAIJ_N_inplace()
39 v = aa + bs2 * ai[r[i]]; in MatILUFactorNumeric_SeqBAIJ_N_inplace()
40 … for (j = 0; j < nz; j++) PetscCall(PetscArraycpy(rtmp + bs2 * ic[ajtmpold[j]], v + bs2 * j, bs2)); in MatILUFactorNumeric_SeqBAIJ_N_inplace()
43 pc = rtmp + bs2 * row; in MatILUFactorNumeric_SeqBAIJ_N_inplace()
45 for (flg = 0, k = 0; k < bs2; k++) { in MatILUFactorNumeric_SeqBAIJ_N_inplace()
52 pv = ba + bs2 * diag_offset[row]; in MatILUFactorNumeric_SeqBAIJ_N_inplace()
56 pv += bs2; in MatILUFactorNumeric_SeqBAIJ_N_inplace()
[all …]
H A Dbaijfact2.c24 PetscInt bs2 = a->bs2, bs = A->rmap->bs, flg; in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering() local
33 PetscCall(PetscMalloc2(bs2 * n, &rtmp, bs2, &mwork)); in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
34 PetscCall(PetscArrayzero(rtmp, bs2 * n)); in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
41 for (j = 0; j < nz; j++) PetscCall(PetscArrayzero(rtmp + bs2 * bjtmp[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
46 for (j = 0; j < nz; j++) PetscCall(PetscArrayzero(rtmp + bs2 * bjtmp[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
51 v = aa + bs2 * ai[i]; in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
52 for (j = 0; j < nz; j++) PetscCall(PetscArraycpy(rtmp + bs2 * ajtmp[j], v + bs2 * j, bs2)); in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
59 pc = rtmp + bs2 * row; in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
60 for (flg = 0, j = 0; j < bs2; j++) { in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
67 pv = b->a + bs2 * bdiag[row]; in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
[all …]
H A Dbaijfact13.c175 const PetscInt *ajtmp, *bjtmp, *bdiag = b->diag, *pj, bs2 = a->bs2; in MatLUFactorNumeric_SeqBAIJ_3() local
187 PetscCall(PetscMalloc2(bs2 * n, &rtmp, bs2, &mwork)); in MatLUFactorNumeric_SeqBAIJ_3()
188 PetscCall(PetscArrayzero(rtmp, bs2 * n)); in MatLUFactorNumeric_SeqBAIJ_3()
195 for (j = 0; j < nz; j++) PetscCall(PetscArrayzero(rtmp + bs2 * bjtmp[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_3()
200 for (j = 0; j < nz; j++) PetscCall(PetscArrayzero(rtmp + bs2 * bjtmp[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_3()
205 v = aa + bs2 * ai[r[i]]; in MatLUFactorNumeric_SeqBAIJ_3()
206 for (j = 0; j < nz; j++) PetscCall(PetscArraycpy(rtmp + bs2 * ic[ajtmp[j]], v + bs2 * j, bs2)); in MatLUFactorNumeric_SeqBAIJ_3()
213 pc = rtmp + bs2 * row; in MatLUFactorNumeric_SeqBAIJ_3()
214 for (flg = 0, j = 0; j < bs2; j++) { in MatLUFactorNumeric_SeqBAIJ_3()
221 pv = b->a + bs2 * bdiag[row]; in MatLUFactorNumeric_SeqBAIJ_3()
[all …]
H A Dbaijfact9.c315 const PetscInt *ajtmp, *bjtmp, *bdiag = b->diag, *pj, bs2 = a->bs2; in MatLUFactorNumeric_SeqBAIJ_5() local
327 PetscCall(PetscMalloc2(bs2 * n, &rtmp, bs2, &mwork)); in MatLUFactorNumeric_SeqBAIJ_5()
328 PetscCall(PetscArrayzero(rtmp, bs2 * n)); in MatLUFactorNumeric_SeqBAIJ_5()
335 for (j = 0; j < nz; j++) PetscCall(PetscArrayzero(rtmp + bs2 * bjtmp[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_5()
340 for (j = 0; j < nz; j++) PetscCall(PetscArrayzero(rtmp + bs2 * bjtmp[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_5()
345 v = aa + bs2 * ai[r[i]]; in MatLUFactorNumeric_SeqBAIJ_5()
346 for (j = 0; j < nz; j++) PetscCall(PetscArraycpy(rtmp + bs2 * ic[ajtmp[j]], v + bs2 * j, bs2)); in MatLUFactorNumeric_SeqBAIJ_5()
353 pc = rtmp + bs2 * row; in MatLUFactorNumeric_SeqBAIJ_5()
354 for (flg = 0, j = 0; j < bs2; j++) { in MatLUFactorNumeric_SeqBAIJ_5()
361 pv = b->a + bs2 * bdiag[row]; in MatLUFactorNumeric_SeqBAIJ_5()
[all …]
H A Dbaijfact11.c237 const PetscInt *ajtmp, *bjtmp, *bdiag = b->diag, *pj, bs2 = a->bs2; in MatLUFactorNumeric_SeqBAIJ_4() local
255 PetscCall(PetscMalloc2(bs2 * n, &rtmp, bs2, &mwork)); in MatLUFactorNumeric_SeqBAIJ_4()
256 PetscCall(PetscArrayzero(rtmp, bs2 * n)); in MatLUFactorNumeric_SeqBAIJ_4()
263 for (j = 0; j < nz; j++) PetscCall(PetscArrayzero(rtmp + bs2 * bjtmp[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_4()
268 for (j = 0; j < nz; j++) PetscCall(PetscArrayzero(rtmp + bs2 * bjtmp[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_4()
273 v = aa + bs2 * ai[r[i]]; in MatLUFactorNumeric_SeqBAIJ_4()
274 for (j = 0; j < nz; j++) PetscCall(PetscArraycpy(rtmp + bs2 * ic[ajtmp[j]], v + bs2 * j, bs2)); in MatLUFactorNumeric_SeqBAIJ_4()
281 pc = rtmp + bs2 * row; in MatLUFactorNumeric_SeqBAIJ_4()
282 for (flg = 0, j = 0; j < bs2; j++) { in MatLUFactorNumeric_SeqBAIJ_4()
289 pv = b->a + bs2 * bdiag[row]; in MatLUFactorNumeric_SeqBAIJ_4()
[all …]
H A Dbaijsolvtrann.c10 const PetscInt n = a->mbs, bs = A->rmap->bs, bs2 = a->bs2; in MatSolveTranspose_SeqBAIJ_N_inplace() local
35 PetscKernel_w_gets_transA_times_v(bs, ls, aa + bs2 * a->diag[i], t + i * bs); in MatSolveTranspose_SeqBAIJ_N_inplace()
36 v = aa + bs2 * (a->diag[i] + 1); in MatSolveTranspose_SeqBAIJ_N_inplace()
41 v += bs2; in MatSolveTranspose_SeqBAIJ_N_inplace()
47 v = aa + bs2 * ai[i]; in MatSolveTranspose_SeqBAIJ_N_inplace()
52 v += bs2; in MatSolveTranspose_SeqBAIJ_N_inplace()
65 PetscCall(PetscLogFlops(2.0 * (a->bs2) * (a->nz) - A->rmap->bs * A->cmap->n)); in MatSolveTranspose_SeqBAIJ_N_inplace()
76 const PetscInt bs = A->rmap->bs, bs2 = a->bs2; in MatSolveTranspose_SeqBAIJ_N() local
101 PetscKernel_w_gets_transA_times_v(bs, ls, aa + bs2 * diag[i], t + i * bs); in MatSolveTranspose_SeqBAIJ_N()
102 v = aa + bs2 * (diag[i] - 1); in MatSolveTranspose_SeqBAIJ_N()
[all …]
H A Dbaijsolvnat3.c85 const PetscInt bs = A->rmap->bs, bs2 = a->bs2; in MatSolve_SeqBAIJ_3_NaturalOrdering() local
100 v = aa + bs2 * ai[i]; in MatSolve_SeqBAIJ_3_NaturalOrdering()
116 v += bs2; in MatSolve_SeqBAIJ_3_NaturalOrdering()
126 v = aa + bs2 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_3_NaturalOrdering()
143 v += bs2; in MatSolve_SeqBAIJ_3_NaturalOrdering()
153 PetscCall(PetscLogFlops(2.0 * bs2 * (a->nz) - bs * A->cmap->n)); in MatSolve_SeqBAIJ_3_NaturalOrdering()
162 const PetscInt bs = A->rmap->bs, bs2 = a->bs2; in MatForwardSolve_SeqBAIJ_3_NaturalOrdering() local
177 v = aa + bs2 * ai[i]; in MatForwardSolve_SeqBAIJ_3_NaturalOrdering()
193 v += bs2; in MatForwardSolve_SeqBAIJ_3_NaturalOrdering()
203 PetscCall(PetscLogFlops(1.0 * bs2 * (a->nz) - bs * A->cmap->n)); in MatForwardSolve_SeqBAIJ_3_NaturalOrdering()
[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() local
25 PetscCall(PetscMalloc2(bs2 * n, &rtmp, bs2, &mwork)); in MatLUFactorNumeric_SeqBAIJ_2()
26 PetscCall(PetscArrayzero(rtmp, bs2 * n)); in MatLUFactorNumeric_SeqBAIJ_2()
34 for (PetscInt j = 0; j < nz; j++) PetscCall(PetscArrayzero(rtmp + bs2 * bjtmp[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_2()
39 for (PetscInt j = 0; j < nz; j++) PetscCall(PetscArrayzero(rtmp + bs2 * bjtmp[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_2()
44 v = aa + bs2 * ai[r[i]]; in MatLUFactorNumeric_SeqBAIJ_2()
45 …PetscInt j = 0; j < nz; j++) PetscCall(PetscArraycpy(rtmp + bs2 * ic[ajtmp[j]], v + bs2 * j, bs2)); in MatLUFactorNumeric_SeqBAIJ_2()
54 pc = rtmp + bs2 * row; in MatLUFactorNumeric_SeqBAIJ_2()
55 for (PetscInt j = 0; j < bs2; j++) { in MatLUFactorNumeric_SeqBAIJ_2()
62 pv = b->a + bs2 * bdiag[row]; in MatLUFactorNumeric_SeqBAIJ_2()
[all …]
H A Dbaijsolvtrannat2.c63 const PetscInt bs = A->rmap->bs, bs2 = a->bs2; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering() local
74 v = aa + bs2 * diag[i]; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering()
80 v -= bs2; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering()
88 v -= bs2; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering()
96 v = aa + bs2 * ai[i]; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering()
106 v += bs2; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering()
110 PetscCall(PetscLogFlops(2.0 * bs2 * (a->nz) - bs * A->cmap->n)); in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering()
H A Dbaijfact7.c373 const PetscInt *ajtmp, *bjtmp, *bdiag = b->diag, *pj, bs2 = a->bs2; in MatLUFactorNumeric_SeqBAIJ_6() local
385 PetscCall(PetscMalloc2(bs2 * n, &rtmp, bs2, &mwork)); in MatLUFactorNumeric_SeqBAIJ_6()
386 PetscCall(PetscArrayzero(rtmp, bs2 * n)); in MatLUFactorNumeric_SeqBAIJ_6()
393 for (j = 0; j < nz; j++) PetscCall(PetscArrayzero(rtmp + bs2 * bjtmp[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_6()
398 for (j = 0; j < nz; j++) PetscCall(PetscArrayzero(rtmp + bs2 * bjtmp[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_6()
403 v = aa + bs2 * ai[r[i]]; in MatLUFactorNumeric_SeqBAIJ_6()
404 for (j = 0; j < nz; j++) PetscCall(PetscArraycpy(rtmp + bs2 * ic[ajtmp[j]], v + bs2 * j, bs2)); in MatLUFactorNumeric_SeqBAIJ_6()
411 pc = rtmp + bs2 * row; in MatLUFactorNumeric_SeqBAIJ_6()
412 for (flg = 0, j = 0; j < bs2; j++) { in MatLUFactorNumeric_SeqBAIJ_6()
419 pv = b->a + bs2 * bdiag[row]; in MatLUFactorNumeric_SeqBAIJ_6()
[all …]
H A Dbaijsolvtrannat3.c69 const PetscInt bs = A->rmap->bs, bs2 = a->bs2; in MatSolveTranspose_SeqBAIJ_3_NaturalOrdering() local
80 v = aa + bs2 * diag[i]; in MatSolveTranspose_SeqBAIJ_3_NaturalOrdering()
88 v -= bs2; in MatSolveTranspose_SeqBAIJ_3_NaturalOrdering()
97 v -= bs2; in MatSolveTranspose_SeqBAIJ_3_NaturalOrdering()
106 v = aa + bs2 * ai[i]; in MatSolveTranspose_SeqBAIJ_3_NaturalOrdering()
118 v += bs2; in MatSolveTranspose_SeqBAIJ_3_NaturalOrdering()
122 PetscCall(PetscLogFlops(2.0 * bs2 * (a->nz) - bs * A->cmap->n)); in MatSolveTranspose_SeqBAIJ_3_NaturalOrdering()
H A Dbaijsolvtran2.c97 const PetscInt bs = A->rmap->bs, bs2 = a->bs2; in MatSolveTranspose_SeqBAIJ_2() local
123 v = aa + bs2 * diag[i]; in MatSolveTranspose_SeqBAIJ_2()
129 v -= bs2; in MatSolveTranspose_SeqBAIJ_2()
137 v -= bs2; in MatSolveTranspose_SeqBAIJ_2()
145 v = aa + bs2 * ai[i]; in MatSolveTranspose_SeqBAIJ_2()
155 v += bs2; in MatSolveTranspose_SeqBAIJ_2()
171 PetscCall(PetscLogFlops(2.0 * bs2 * (a->nz) - bs * A->cmap->n)); in MatSolveTranspose_SeqBAIJ_2()
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()
205 odiag = v + bs2 * adiag[i]; in MatInvertBlockDiagonal_SeqBAIJ()
206 PetscCall(PetscArraycpy(diag, odiag, bs2)); in MatInvertBlockDiagonal_SeqBAIJ()
209 diag += bs2; 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() local
522 idiag += bs2; in MatSOR_SeqBAIJ()
524 v = aa + bs2 * ai[i]; in MatSOR_SeqBAIJ()
539 idiag += bs2; in MatSOR_SeqBAIJ()
545 PetscCall(PetscLogFlops(1.0 * bs2 * a->nz)); in MatSOR_SeqBAIJ()
[all …]
H A Dbaijsolvtrannat4.c75 const PetscInt bs = A->rmap->bs, bs2 = a->bs2; in MatSolveTranspose_SeqBAIJ_4_NaturalOrdering() local
86 v = aa + bs2 * diag[i]; in MatSolveTranspose_SeqBAIJ_4_NaturalOrdering()
96 v -= bs2; in MatSolveTranspose_SeqBAIJ_4_NaturalOrdering()
106 v -= bs2; in MatSolveTranspose_SeqBAIJ_4_NaturalOrdering()
116 v = aa + bs2 * ai[i]; in MatSolveTranspose_SeqBAIJ_4_NaturalOrdering()
130 v += bs2; in MatSolveTranspose_SeqBAIJ_4_NaturalOrdering()
134 PetscCall(PetscLogFlops(2.0 * bs2 * (a->nz) - bs * A->cmap->n)); in MatSolveTranspose_SeqBAIJ_4_NaturalOrdering()
H A Dbaijsolvnat14.c10 …cInt n = a->mbs, *ai = a->i, *aj = a->j, *adiag = a->diag, *vi, bs = A->rmap->bs, bs2 = a->bs2; in MatSolve_SeqBAIJ_14_NaturalOrdering() local
23 v = aa + bs2 * ai[i]; in MatSolve_SeqBAIJ_14_NaturalOrdering()
65 v = aa + bs2 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_14_NaturalOrdering()
126 PetscCall(PetscLogFlops(2.0 * bs2 * (a->nz) - bs * A->cmap->n)); in MatSolve_SeqBAIJ_14_NaturalOrdering()
136 …cInt n = a->mbs, *ai = a->i, *aj = a->j, *adiag = a->diag, *vi, bs = A->rmap->bs, bs2 = a->bs2; in MatSolve_SeqBAIJ_13_NaturalOrdering() local
149 v = aa + bs2 * ai[i]; in MatSolve_SeqBAIJ_13_NaturalOrdering()
189 v = aa + bs2 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_13_NaturalOrdering()
247 PetscCall(PetscLogFlops(2.0 * bs2 * (a->nz) - bs * A->cmap->n)); in MatSolve_SeqBAIJ_13_NaturalOrdering()
257 …cInt n = a->mbs, *ai = a->i, *aj = a->j, *adiag = a->diag, *vi, bs = A->rmap->bs, bs2 = a->bs2; in MatSolve_SeqBAIJ_12_NaturalOrdering() local
270 v = aa + bs2 * ai[i]; in MatSolve_SeqBAIJ_12_NaturalOrdering()
[all …]
H A Dbaijsolvtran3.c105 const PetscInt bs = A->rmap->bs, bs2 = a->bs2; in MatSolveTranspose_SeqBAIJ_3() local
132 v = aa + bs2 * diag[i]; in MatSolveTranspose_SeqBAIJ_3()
140 v -= bs2; in MatSolveTranspose_SeqBAIJ_3()
149 v -= bs2; in MatSolveTranspose_SeqBAIJ_3()
158 v = aa + bs2 * ai[i]; in MatSolveTranspose_SeqBAIJ_3()
170 v += bs2; in MatSolveTranspose_SeqBAIJ_3()
187 PetscCall(PetscLogFlops(2.0 * bs2 * (a->nz) - bs * A->cmap->n)); in MatSolveTranspose_SeqBAIJ_3()
H A Dbaijfact5.c469 const PetscInt *ajtmp, *bjtmp, *bdiag = b->diag, *pj, bs2 = a->bs2; in MatLUFactorNumeric_SeqBAIJ_7() local
481 PetscCall(PetscMalloc2(bs2 * n, &rtmp, bs2, &mwork)); in MatLUFactorNumeric_SeqBAIJ_7()
482 PetscCall(PetscArrayzero(rtmp, bs2 * n)); in MatLUFactorNumeric_SeqBAIJ_7()
489 for (j = 0; j < nz; j++) PetscCall(PetscArrayzero(rtmp + bs2 * bjtmp[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_7()
494 for (j = 0; j < nz; j++) PetscCall(PetscArrayzero(rtmp + bs2 * bjtmp[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_7()
499 v = aa + bs2 * ai[r[i]]; in MatLUFactorNumeric_SeqBAIJ_7()
500 for (j = 0; j < nz; j++) PetscCall(PetscArraycpy(rtmp + bs2 * ic[ajtmp[j]], v + bs2 * j, bs2)); in MatLUFactorNumeric_SeqBAIJ_7()
507 pc = rtmp + bs2 * row; in MatLUFactorNumeric_SeqBAIJ_7()
508 for (flg = 0, j = 0; j < bs2; j++) { in MatLUFactorNumeric_SeqBAIJ_7()
515 pv = b->a + bs2 * bdiag[row]; in MatLUFactorNumeric_SeqBAIJ_7()
[all …]
H A Dbaijsolvnat11.c10 …cInt n = a->mbs, *ai = a->i, *aj = a->j, *adiag = a->diag, *vi, bs = A->rmap->bs, bs2 = a->bs2; in MatSolve_SeqBAIJ_11_NaturalOrdering() local
23 v = aa + bs2 * ai[i]; in MatSolve_SeqBAIJ_11_NaturalOrdering()
59 v = aa + bs2 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_11_NaturalOrdering()
111 PetscCall(PetscLogFlops(2.0 * bs2 * (a->nz) - bs * A->cmap->n)); in MatSolve_SeqBAIJ_11_NaturalOrdering()
/petsc/src/mat/utils/
H A Dmatstash.c131 PetscInt bs2, oldnmax; in MatStashScatterEnd_Ref() local
147 bs2 = stash->bs * stash->bs; in MatStashScatterEnd_Ref()
148 oldnmax = ((int)(stash->n * 1.1) + 5) * bs2; in MatStashScatterEnd_Ref()
182 PetscInt bs2 = stash->bs * stash->bs; in MatStashGetInfo_Private() local
185 if (nstash) *nstash = stash->n * bs2; in MatStashGetInfo_Private()
221 PetscInt newnmax, bs2 = stash->bs * stash->bs; in MatStashExpand_Private() local
227 if (stash->umax) cnewnmax = stash->umax / bs2; in MatStashExpand_Private()
228 else cnewnmax = DEFAULT_STASH_SIZE / bs2; in MatStashExpand_Private()
230 if (stash->umax > stash->oldnmax) cnewnmax = stash->umax / bs2; in MatStashExpand_Private()
231 else cnewnmax = stash->oldnmax / bs2; in MatStashExpand_Private()
[all …]
H A Dmatstashspace.c4 PetscErrorCode PetscMatStashSpaceGet(PetscInt bs2, PetscInt n, PetscMatStashSpace *space) in PetscMatStashSpaceGet() argument
12 PetscCall(PetscMalloc3(n * bs2, &a->space_head, n, &a->idx, n, &a->idy)); in PetscMatStashSpaceGet()
30 PetscErrorCode PetscMatStashSpaceContiguous(PetscInt bs2, PetscMatStashSpace *space, PetscScalar *v… in PetscMatStashSpaceContiguous() argument
37 PetscCall(PetscArraycpy(val, (*space)->val, (*space)->local_used * bs2)); in PetscMatStashSpaceContiguous()
38 val += bs2 * (*space)->local_used; in PetscMatStashSpaceContiguous()
/petsc/src/mat/impls/sbaij/seq/
H A Dsbaij.c287 PetscInt i, j, bs = A->rmap->bs, k, l, bs2 = a->bs2; in MatView_SeqSBAIJ_ASCII() local
360 if (PetscImaginaryPart(a->a[bs2 * k + l * bs + j]) > 0.0) { in MatView_SeqSBAIJ_ASCII()
361 …, bs * a->j[k] + l, (double)PetscRealPart(a->a[bs2 * k + l * bs + j]), (double)PetscImaginaryPart(… in MatView_SeqSBAIJ_ASCII()
362 } else if (PetscImaginaryPart(a->a[bs2 * k + l * bs + j]) < 0.0) { in MatView_SeqSBAIJ_ASCII()
363 … bs * a->j[k] + l, (double)PetscRealPart(a->a[bs2 * k + l * bs + j]), -(double)PetscImaginaryPart(… in MatView_SeqSBAIJ_ASCII()
365 … " (%" PetscInt_FMT ", %g) ", bs * a->j[k] + l, (double)PetscRealPart(a->a[bs2 * k + l * bs + j]))… in MatView_SeqSBAIJ_ASCII()
368 …Printf(viewer, " (%" PetscInt_FMT ", %g) ", bs * a->j[k] + l, (double)a->a[bs2 * k + l * bs + j])); 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
409 aa = a->a + j * bs2; in MatView_SeqSBAIJ_Draw_Zoom()
425 aa = a->a + j * bs2; in MatView_SeqSBAIJ_Draw_Zoom()
[all …]
H A Dsbaijfact.c43 PetscInt i, mbs = a->mbs, *jutmp, bs = A->rmap->bs, bs2 = a->bs2; in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR() local
184 PetscCall(PetscShmgetAllocateArray((iu[mbs] + 1) * bs2, sizeof(PetscScalar), (void **)&b->a)); in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
565 PetscInt bs = A->rmap->bs, bs2 = a->bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N() local
574 PetscCall(PetscCalloc1(bs2 * mbs, &rtmp)); in MatCholeskyFactorNumeric_SeqSBAIJ_N()
581 PetscCall(PetscMalloc3(bs2, &dk, bs2, &uik, bs, &work)); in MatCholeskyFactorNumeric_SeqSBAIJ_N()
594 PetscCall(PetscMalloc1(bs2 * ai[mbs], &aa)); in MatCholeskyFactorNumeric_SeqSBAIJ_N()
595 PetscCall(PetscArraycpy(aa, a->a, bs2 * ai[mbs])); in MatCholeskyFactorNumeric_SeqSBAIJ_N()
607 for (k1 = 0; k1 < bs2; k1++) { in MatCholeskyFactorNumeric_SeqSBAIJ_N()
608 dk[k1] = aa[k * bs2 + k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_N()
609 aa[k * bs2 + k1] = aa[j * bs2 + k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_N()
[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
82 av += bs2; in MatConvert_SeqSBAIJ_SeqAIJ()
105 av += bs2; in MatConvert_SeqSBAIJ_SeqAIJ()
241 PetscInt bs = A->rmap->bs, bs2 = bs * bs, mbs = m / bs; in MatConvert_SeqSBAIJ_SeqBAIJ() local
288 itmp = bs2 * browstart[i]; in MatConvert_SeqSBAIJ_SeqBAIJ()
289 for (PetscInt k = 0; k < bs2; k++) { in MatConvert_SeqSBAIJ_SeqBAIJ()
300 itmp = bs2 * browstart[*aj]; /* row index */ in MatConvert_SeqSBAIJ_SeqBAIJ()
314 itmp = bs2 * browstart[i]; in MatConvert_SeqSBAIJ_SeqBAIJ()
315 for (PetscInt k = 0; k < bs2; k++) bv[itmp + k] = av[k]; in MatConvert_SeqSBAIJ_SeqBAIJ()
316 av += bs2; in MatConvert_SeqSBAIJ_SeqBAIJ()
[all …]
/petsc/src/mat/impls/baij/mpi/
H A Dbaijov.c642 PetscInt bs = C->rmap->bs, bs2 = c->bs2, rstart = c->rstartbs; local
1235 PetscCall(PetscMalloc1(rbuf2[i][0] * bs2, &rbuf4[i]));
1236 …PetscCallMPI(MPIU_Irecv(rbuf4[i], rbuf2[i][0] * bs2, MPIU_SCALAR, req_source2[i], tag4, comm, r_wa…
1243 if (nrqr) PetscCall(PetscMalloc1(jcnt * bs2, &sbuf_aa[0]));
1244 for (PetscMPIInt i = 1; i < nrqr; i++) sbuf_aa[i] = sbuf_aa[i - 1] + req_size[i - 1] * bs2;
1263 vworkA = PetscSafePointerPlusOffset(a_a, a_i[row] * bs2);
1264 vworkB = PetscSafePointerPlusOffset(b_a, b_i[row] * bs2);
1267 vals = sbuf_aa_i + ct2 * bs2;
1269 … if ((bmap[cworkB[l]]) < cstart) PetscCall(PetscArraycpy(vals + l * bs2, vworkB + l * bs2, bs2));
1273 …or (l = 0; l < nzA; l++) PetscCall(PetscArraycpy(vals + (imark + l) * bs2, vworkA + l * bs2, bs2));
[all …]
/petsc/src/mat/impls/kaij/
H A Dkaij.c863 PetscInt i, j, k, i2, bs, bs2, nz; in MatSOR_SeqKAIJ() local
873 bs2 = bs * bs; in MatSOR_SeqKAIJ()
882 …>sor.w, bs, &kaij->sor.y, m * bs, &kaij->sor.work, m * bs, &kaij->sor.t, m * bs2, &kaij->sor.arr)); in MatSOR_SeqKAIJ()
899 idiag += bs2; in MatSOR_SeqKAIJ()
924 idiag += bs2; in MatSOR_SeqKAIJ()
928 PetscCall(PetscLogFlops(1.0 * bs2 * a->nz)); in MatSOR_SeqKAIJ()
932 idiag = kaij->ibdiag + bs2 * (m - 1); in MatSOR_SeqKAIJ()
937 idiag -= bs2; in MatSOR_SeqKAIJ()
953 PetscCall(PetscArraycpy(arrt, T, bs2)); in MatSOR_SeqKAIJ()
954 for (k = 0; k < bs2; k++) arrt[k] *= v[j]; in MatSOR_SeqKAIJ()
[all …]

123