Lines Matching refs:bs2
13 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()
67 pv = b->a + bs2 * (bdiag[row + 1] + 1); in MatLUFactorNumeric_SeqBAIJ_2()
82 pv = b->a + bs2 * bi[i]; in MatLUFactorNumeric_SeqBAIJ_2()
85 … for (PetscInt j = 0; j < nz; j++) PetscCall(PetscArraycpy(pv + bs2 * j, rtmp + bs2 * pj[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_2()
88 pv = b->a + bs2 * bdiag[i]; in MatLUFactorNumeric_SeqBAIJ_2()
90 PetscCall(PetscArraycpy(pv, rtmp + bs2 * pj[0], bs2)); in MatLUFactorNumeric_SeqBAIJ_2()
99 pv = b->a + bs2 * (bdiag[i + 1] + 1); in MatLUFactorNumeric_SeqBAIJ_2()
102 … for (PetscInt j = 0; j < nz; j++) PetscCall(PetscArraycpy(pv + bs2 * j, rtmp + bs2 * pj[j], 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() local
134 PetscCall(PetscMalloc2(bs2 * n, &rtmp, bs2, &mwork)); in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
135 PetscCall(PetscArrayzero(rtmp, bs2 * n)); in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
142 for (j = 0; j < nz; j++) PetscCall(PetscArrayzero(rtmp + bs2 * bjtmp[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
147 for (j = 0; j < nz; j++) PetscCall(PetscArrayzero(rtmp + bs2 * bjtmp[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
152 v = aa + bs2 * ai[i]; in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
153 for (j = 0; j < nz; j++) PetscCall(PetscArraycpy(rtmp + bs2 * ajtmp[j], v + bs2 * j, bs2)); in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
160 pc = rtmp + bs2 * row; in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
161 for (flg = 0, j = 0; j < bs2; j++) { in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
168 pv = b->a + bs2 * bdiag[row]; in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
173 pv = b->a + bs2 * (bdiag[row + 1] + 1); in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
188 pv = b->a + bs2 * bi[i]; in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
191 for (j = 0; j < nz; j++) PetscCall(PetscArraycpy(pv + bs2 * j, rtmp + bs2 * pj[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
194 pv = b->a + bs2 * bdiag[i]; in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
196 PetscCall(PetscArraycpy(pv, rtmp + bs2 * pj[0], bs2)); in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
206 pv = b->a + bs2 * (bdiag[i + 1] + 1); in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
209 for (j = 0; j < nz; j++) PetscCall(PetscArraycpy(pv + bs2 * j, rtmp + bs2 * pj[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
1428 PetscInt nz, bs = A->rmap->bs, bs2 = a->bs2; in MatSolve_SeqBAIJ_N_NaturalOrdering() local
1443 v = aa + bs2 * ai[i]; in MatSolve_SeqBAIJ_N_NaturalOrdering()
1450 v += bs2; in MatSolve_SeqBAIJ_N_NaturalOrdering()
1457 v = aa + bs2 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_N_NaturalOrdering()
1463 v += bs2; in MatSolve_SeqBAIJ_N_NaturalOrdering()
1465 PetscKernel_w_gets_A_times_v(bs, ls, aa + bs2 * adiag[i], t + i * bs); /* *inv(diagonal[i]) */ in MatSolve_SeqBAIJ_N_NaturalOrdering()
1471 PetscCall(PetscLogFlops(2.0 * (a->bs2) * (a->nz) - A->rmap->bs * A->cmap->n)); in MatSolve_SeqBAIJ_N_NaturalOrdering()
1481 PetscInt nz, bs = A->rmap->bs, bs2 = a->bs2; in MatSolve_SeqBAIJ_N() local
1500 v = aa + bs2 * ai[i]; in MatSolve_SeqBAIJ_N()
1507 v += bs2; in MatSolve_SeqBAIJ_N()
1514 v = aa + bs2 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_N()
1520 v += bs2; in MatSolve_SeqBAIJ_N()
1529 PetscCall(PetscLogFlops(2.0 * (a->bs2) * (a->nz) - A->rmap->bs * A->cmap->n)); in MatSolve_SeqBAIJ_N()