Lines Matching refs:bs2
24 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()
71 pv = b->a + bs2 * (bdiag[row + 1] + 1); in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
74 vv = rtmp + bs2 * pj[j]; in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
77 pv += bs2; in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
79 … PetscCall(PetscLogFlops(2.0 * bs2 * bs * (nz + 1) - bs2)); /* flops = 2*bs^3*nz + 2*bs^3 - bs2) */ in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
85 pv = b->a + bs2 * bi[i]; in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
88 for (j = 0; j < nz; j++) PetscCall(PetscArraycpy(pv + bs2 * j, rtmp + bs2 * pj[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
91 pv = b->a + bs2 * bdiag[i]; in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
93 PetscCall(PetscArraycpy(pv, rtmp + bs2 * pj[0], bs2)); in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
98 pv = b->a + bs2 * (bdiag[i + 1] + 1); in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
101 for (j = 0; j < nz; j++) PetscCall(PetscArraycpy(pv + bs2 * j, rtmp + bs2 * pj[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
110 PetscCall(PetscLogFlops(1.333333333333 * bs * bs2 * b->mbs)); /* from inverting diagonal blocks */ in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
123 PetscInt bs = A->rmap->bs, bs2 = a->bs2, *v_pivots, flg; in MatLUFactorNumeric_SeqBAIJ_N() local
133 PetscCall(PetscCalloc1(bs2 * n, &rtmp)); in MatLUFactorNumeric_SeqBAIJ_N()
136 PetscCall(PetscMalloc3(bs, &v_work, bs2, &mwork, bs, &v_pivots)); in MatLUFactorNumeric_SeqBAIJ_N()
143 for (j = 0; j < nz; j++) PetscCall(PetscArrayzero(rtmp + bs2 * bjtmp[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_N()
148 for (j = 0; j < nz; j++) PetscCall(PetscArrayzero(rtmp + bs2 * bjtmp[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_N()
153 v = aa + bs2 * ai[r[i]]; in MatLUFactorNumeric_SeqBAIJ_N()
154 for (j = 0; j < nz; j++) PetscCall(PetscArraycpy(rtmp + bs2 * ic[ajtmp[j]], v + bs2 * j, bs2)); in MatLUFactorNumeric_SeqBAIJ_N()
161 pc = rtmp + bs2 * row; in MatLUFactorNumeric_SeqBAIJ_N()
162 for (flg = 0, j = 0; j < bs2; j++) { in MatLUFactorNumeric_SeqBAIJ_N()
169 pv = b->a + bs2 * bdiag[row]; in MatLUFactorNumeric_SeqBAIJ_N()
172 pv = b->a + bs2 * (bdiag[row + 1] + 1); in MatLUFactorNumeric_SeqBAIJ_N()
174 …j = 0; j < nz; j++) PetscKernel_A_gets_A_minus_B_times_C(bs, rtmp + bs2 * pj[j], pc, pv + bs2 * j); in MatLUFactorNumeric_SeqBAIJ_N()
175 … PetscCall(PetscLogFlops(2.0 * bs2 * bs * (nz + 1) - bs2)); /* flops = 2*bs^3*nz + 2*bs^3 - bs2) */ in MatLUFactorNumeric_SeqBAIJ_N()
181 pv = b->a + bs2 * bi[i]; in MatLUFactorNumeric_SeqBAIJ_N()
184 for (j = 0; j < nz; j++) PetscCall(PetscArraycpy(pv + bs2 * j, rtmp + bs2 * pj[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_N()
187 pv = b->a + bs2 * bdiag[i]; in MatLUFactorNumeric_SeqBAIJ_N()
189 PetscCall(PetscArraycpy(pv, rtmp + bs2 * pj[0], bs2)); in MatLUFactorNumeric_SeqBAIJ_N()
195 pv = b->a + bs2 * (bdiag[i + 1] + 1); in MatLUFactorNumeric_SeqBAIJ_N()
198 for (j = 0; j < nz; j++) PetscCall(PetscArraycpy(pv + bs2 * j, rtmp + bs2 * pj[j], bs2)); in MatLUFactorNumeric_SeqBAIJ_N()
242 PetscCall(PetscLogFlops(1.333333333333 * bs * bs2 * b->mbs)); /* from inverting diagonal blocks */ in MatLUFactorNumeric_SeqBAIJ_N()
255 const PetscInt n = a->mbs, *ai = a->i, *aj, *adiag, bs2 = a->bs2; in MatILUFactorSymbolic_SeqBAIJ_ilu0() local
264 PetscCall(PetscShmgetAllocateArray(bs2 * ai[n], sizeof(PetscScalar), (void **)&b->a)); in MatILUFactorSymbolic_SeqBAIJ_ilu0()
274 if (n > 0) PetscCall(PetscArrayzero(b->a, bs2 * ai[n])); in MatILUFactorSymbolic_SeqBAIJ_ilu0()
327 PetscInt bs = A->rmap->bs, bs2 = a->bs2; in MatILUFactorSymbolic_SeqBAIJ() local
484 PetscCall(PetscShmgetAllocateArray(bs2 * (bdiag[0] + 1), sizeof(PetscScalar), (void **)&b->a)); in MatILUFactorSymbolic_SeqBAIJ()