Lines Matching refs:dk
566 MatScalar *ba = b->a, *aa, *ap, *dk, *uik; in MatCholeskyFactorNumeric_SeqSBAIJ_N() local
581 PetscCall(PetscMalloc3(bs2, &dk, bs2, &uik, bs, &work)); in MatCholeskyFactorNumeric_SeqSBAIJ_N()
608 dk[k1] = aa[k * bs2 + k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_N()
610 aa[j * bs2 + k1] = dk[k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_N()
616 for (k = 0; k < bs2; k++) dk[k] = ap[k]; /* dk <- j-th block of aa */ in MatCholeskyFactorNumeric_SeqSBAIJ_N()
618 for (k1 = 0; k1 < bs; k1++) *ap++ = dk[k + bs * k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_N()
640 PetscCall(PetscArraycpy(dk, rtmp + k * bs2, bs2)); in MatCholeskyFactorNumeric_SeqSBAIJ_N()
656 PetscKernel_A_gets_A_plus_Btranspose_times_C(bs, dk, uik, u); in MatCholeskyFactorNumeric_SeqSBAIJ_N()
687 PetscCall(PetscArraycpy(diag, dk, bs2)); in MatCholeskyFactorNumeric_SeqSBAIJ_N()
715 PetscCall(PetscFree3(dk, uik, work)); in MatCholeskyFactorNumeric_SeqSBAIJ_N()
739 MatScalar *ba = b->a, *aa, *ap, *dk, *uik; in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering() local
751 PetscCall(PetscMalloc3(bs2, &dk, bs2, &uik, bs, &work)); in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
772 PetscCall(PetscArraycpy(dk, rtmp + k * bs2, bs2)); in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
788 PetscKernel_A_gets_A_plus_Btranspose_times_C(bs, dk, uik, u); in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
819 PetscCall(PetscArraycpy(diag, dk, bs2)); in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
847 PetscCall(PetscFree3(dk, uik, work)); in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
873 MatScalar *u, *diag, *rtmp, *rtmp_ptr, dk[4], uik[4]; in MatCholeskyFactorNumeric_SeqSBAIJ_2() local
918 dk[k1] = aa[k * 4 + k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_2()
920 aa[j * 4 + k1] = dk[k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_2()
926 dk[1] = ap[1]; /* swap ap[1] and ap[2] */ in MatCholeskyFactorNumeric_SeqSBAIJ_2()
928 ap[2] = dk[1]; in MatCholeskyFactorNumeric_SeqSBAIJ_2()
948 PetscCall(PetscArraycpy(dk, rtmp + k * 4, 4)); in MatCholeskyFactorNumeric_SeqSBAIJ_2()
966 dk[0] += uik[0] * u[0] + uik[1] * u[1]; in MatCholeskyFactorNumeric_SeqSBAIJ_2()
967 dk[1] += uik[2] * u[0] + uik[3] * u[1]; in MatCholeskyFactorNumeric_SeqSBAIJ_2()
968 dk[2] += uik[0] * u[2] + uik[1] * u[3]; in MatCholeskyFactorNumeric_SeqSBAIJ_2()
969 dk[3] += uik[2] * u[2] + uik[3] * u[3]; in MatCholeskyFactorNumeric_SeqSBAIJ_2()
1004 PetscCall(PetscArraycpy(diag, dk, 4)); in MatCholeskyFactorNumeric_SeqSBAIJ_2()
1051 MatScalar *ba = b->a, *aa, *ap, dk[8], uik[8]; in MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering() local
1090 PetscCall(PetscArraycpy(dk, rtmp + k * 4, 4)); in MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering()
1108 dk[0] += uik[0] * u[0] + uik[1] * u[1]; in MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering()
1109 dk[1] += uik[2] * u[0] + uik[3] * u[1]; in MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering()
1110 dk[2] += uik[0] * u[2] + uik[1] * u[3]; in MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering()
1111 dk[3] += uik[2] * u[2] + uik[3] * u[3]; in MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering()
1146 PetscCall(PetscArraycpy(diag, dk, 4)); in MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering()
1196 MatScalar *rtmp, *ba = b->a, *bval, *aa, dk, uikdi; in MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace() local
1253 dk = rtmp[k]; in MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace()
1262 dk += uikdi * ba[ili]; in MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace()
1295 sctx.pv = dk; in MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace()
1298 dk = sctx.pv; in MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace()
1301 ba[bi[k]] = 1.0 / dk; /* U(k,k) */ in MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace()
1353 MatScalar *rtmp, *ba = b->a, *bval, *aa = a->a, dk, uikdi; in MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering() local
1419 dk = rtmp[k]; in MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering()
1428 dk += uikdi * ba[ili]; /* update diag[k] */ in MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering()
1463 sctx.pv = dk; in MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering()
1466 dk = sctx.pv; in MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering()
1468 ba[bdiag[k]] = 1.0 / dk; /* U(k,k) */ in MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering()
1505 MatScalar *rtmp, *ba = b->a, *aa = a->a, dk, uikdi, *aval, *bval; in MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace() local
1548 dk = rtmp[k]; in MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace()
1556 dk += uikdi * ba[ili]; in MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace()
1591 sctx.pv = dk; in MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace()
1594 dk = sctx.pv; in MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace()
1597 ba[bi[k]] = 1.0 / dk; in MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace()