Home
last modified time | relevance | path

Searched refs:dk (Results 1 – 14 of 14) sorted by relevance

/petsc/src/mat/impls/sbaij/seq/
H A Dsbaijfact6.c11 MatScalar *ba = b->a, *aa, *ap, *dk, *uik; in MatCholeskyFactorNumeric_SeqSBAIJ_4() local
25 PetscCall(PetscMalloc2(16, &dk, 16, &uik)); in MatCholeskyFactorNumeric_SeqSBAIJ_4()
50 dk[k1] = aa[k * 16 + k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_4()
52 aa[j * 16 + k1] = dk[k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_4()
59 for (k = 0; k < 16; k++) dk[k] = ap[k]; /* dk <- j-th block of aa */ in MatCholeskyFactorNumeric_SeqSBAIJ_4()
61 for (k1 = 0; k1 < 4; k1++) *ap++ = dk[k + 4 * k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_4()
84 PetscCall(PetscArraycpy(dk, rtmp + k * 16, 16)); in MatCholeskyFactorNumeric_SeqSBAIJ_4()
118 dk[0] += uik[0] * u[0] + uik[1] * u[1] + uik[2] * u[2] + uik[3] * u[3]; in MatCholeskyFactorNumeric_SeqSBAIJ_4()
119 dk[1] += uik[4] * u[0] + uik[5] * u[1] + uik[6] * u[2] + uik[7] * u[3]; in MatCholeskyFactorNumeric_SeqSBAIJ_4()
120 dk[2] += uik[8] * u[0] + uik[9] * u[1] + uik[10] * u[2] + uik[11] * u[3]; in MatCholeskyFactorNumeric_SeqSBAIJ_4()
[all …]
H A Dsbaijfact7.c11 MatScalar *ba = b->a, *aa, *ap, *dk, *uik; in MatCholeskyFactorNumeric_SeqSBAIJ_5() local
24 PetscCall(PetscMalloc2(25, &dk, 25, &uik)); in MatCholeskyFactorNumeric_SeqSBAIJ_5()
49 dk[k1] = aa[k * 25 + k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_5()
51 aa[j * 25 + k1] = dk[k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_5()
58 for (k = 0; k < 25; k++) dk[k] = ap[k]; /* dk <- j-th block of aa */ in MatCholeskyFactorNumeric_SeqSBAIJ_5()
60 for (k1 = 0; k1 < 5; k1++) *ap++ = dk[k + 5 * k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_5()
83 PetscCall(PetscArraycpy(dk, rtmp + k * 25, 25)); in MatCholeskyFactorNumeric_SeqSBAIJ_5()
127 dk[0] += uik[0] * u[0] + uik[1] * u[1] + uik[2] * u[2] + uik[3] * u[3] + uik[4] * u[4]; in MatCholeskyFactorNumeric_SeqSBAIJ_5()
128 dk[1] += uik[5] * u[0] + uik[6] * u[1] + uik[7] * u[2] + uik[8] * u[3] + uik[9] * u[4]; in MatCholeskyFactorNumeric_SeqSBAIJ_5()
129 dk[2] += uik[10] * u[0] + uik[11] * u[1] + uik[12] * u[2] + uik[13] * u[3] + uik[14] * u[4]; in MatCholeskyFactorNumeric_SeqSBAIJ_5()
[all …]
H A Dsbaijfact8.c12 MatScalar *ba = b->a, *aa, *ap, *dk, *uik; in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering() local
25 PetscCall(PetscMalloc2(25, &dk, 25, &uik)); in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering()
45 PetscCall(PetscArraycpy(dk, rtmp + k * 25, 25)); in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering()
89 dk[0] += uik[0] * u[0] + uik[1] * u[1] + uik[2] * u[2] + uik[3] * u[3] + uik[4] * u[4]; in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering()
90 dk[1] += uik[5] * u[0] + uik[6] * u[1] + uik[7] * u[2] + uik[8] * u[3] + uik[9] * u[4]; in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering()
91 dk[2] += uik[10] * u[0] + uik[11] * u[1] + uik[12] * u[2] + uik[13] * u[3] + uik[14] * u[4]; in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering()
92 dk[3] += uik[15] * u[0] + uik[16] * u[1] + uik[17] * u[2] + uik[18] * u[3] + uik[19] * u[4]; in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering()
93 dk[4] += uik[20] * u[0] + uik[21] * u[1] + uik[22] * u[2] + uik[23] * u[3] + uik[24] * u[4]; in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering()
95 dk[5] += uik[0] * u[5] + uik[1] * u[6] + uik[2] * u[7] + uik[3] * u[8] + uik[4] * u[9]; in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering()
96 dk[6] += uik[5] * u[5] + uik[6] * u[6] + uik[7] * u[7] + uik[8] * u[8] + uik[9] * u[9]; in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering()
[all …]
H A Dsbaijfact5.c12 MatScalar *ba = b->a, *aa, *ap, *dk, *uik; in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering() local
26 PetscCall(PetscMalloc2(16, &dk, 16, &uik)); in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
46 PetscCall(PetscArraycpy(dk, rtmp + k * 16, 16)); in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
80 dk[0] += uik[0] * u[0] + uik[1] * u[1] + uik[2] * u[2] + uik[3] * u[3]; in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
81 dk[1] += uik[4] * u[0] + uik[5] * u[1] + uik[6] * u[2] + uik[7] * u[3]; in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
82 dk[2] += uik[8] * u[0] + uik[9] * u[1] + uik[10] * u[2] + uik[11] * u[3]; in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
83 dk[3] += uik[12] * u[0] + uik[13] * u[1] + uik[14] * u[2] + uik[15] * u[3]; in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
85 dk[4] += uik[0] * u[4] + uik[1] * u[5] + uik[2] * u[6] + uik[3] * u[7]; in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
86 dk[5] += uik[4] * u[4] + uik[5] * u[5] + uik[6] * u[6] + uik[7] * u[7]; in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
87 dk[6] += uik[8] * u[4] + uik[9] * u[5] + uik[10] * u[6] + uik[11] * u[7]; in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
[all …]
H A Dsbaijfact3.c11 MatScalar *ba = b->a, *aa, *ap, *dk, *uik; in MatCholeskyFactorNumeric_SeqSBAIJ_3() local
24 PetscCall(PetscMalloc2(9, &dk, 9, &uik)); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
49 dk[k1] = aa[k * 9 + k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_3()
51 aa[j * 9 + k1] = dk[k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_3()
58 for (k = 0; k < 9; k++) dk[k] = ap[k]; /* dk <- j-th block of aa */ in MatCholeskyFactorNumeric_SeqSBAIJ_3()
60 for (k1 = 0; k1 < 3; k1++) *ap++ = dk[k + 3 * k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_3()
83 PetscCall(PetscArraycpy(dk, rtmp + k * 9, 9)); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
109 dk[0] += uik[0] * u[0] + uik[1] * u[1] + uik[2] * u[2]; in MatCholeskyFactorNumeric_SeqSBAIJ_3()
110 dk[1] += uik[3] * u[0] + uik[4] * u[1] + uik[5] * u[2]; in MatCholeskyFactorNumeric_SeqSBAIJ_3()
111 dk[2] += uik[6] * u[0] + uik[7] * u[1] + uik[8] * u[2]; in MatCholeskyFactorNumeric_SeqSBAIJ_3()
[all …]
H A Dsbaijfact4.c12 MatScalar *ba = b->a, *aa, *ap, *dk, *uik; in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering() local
25 PetscCall(PetscMalloc2(9, &dk, 9, &uik)); in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
45 PetscCall(PetscArraycpy(dk, rtmp + k * 9, 9)); in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
71 dk[0] += uik[0] * u[0] + uik[1] * u[1] + uik[2] * u[2]; in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
72 dk[1] += uik[3] * u[0] + uik[4] * u[1] + uik[5] * u[2]; in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
73 dk[2] += uik[6] * u[0] + uik[7] * u[1] + uik[8] * u[2]; in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
75 dk[3] += uik[0] * u[3] + uik[1] * u[4] + uik[2] * u[5]; in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
76 dk[4] += uik[3] * u[3] + uik[4] * u[4] + uik[5] * u[5]; in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
77 dk[5] += uik[6] * u[3] + uik[7] * u[4] + uik[8] * u[5]; in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
79 dk[6] += uik[0] * u[6] + uik[1] * u[7] + uik[2] * u[8]; in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
[all …]
H A Dsbaijfact9.c11 MatScalar *ba = b->a, *aa, *ap, *dk, *uik; in MatCholeskyFactorNumeric_SeqSBAIJ_6() local
26 PetscCall(PetscMalloc2(36, &dk, 36, &uik)); in MatCholeskyFactorNumeric_SeqSBAIJ_6()
51 dk[k1] = aa[k * 36 + k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_6()
53 aa[j * 36 + k1] = dk[k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_6()
60 for (k = 0; k < 36; k++) dk[k] = ap[k]; /* dk <- j-th block of aa */ in MatCholeskyFactorNumeric_SeqSBAIJ_6()
62 for (k1 = 0; k1 < 6; k1++) *ap++ = dk[k + 6 * k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_6()
85 PetscCall(PetscArraycpy(dk, w + k * 36, 36)); in MatCholeskyFactorNumeric_SeqSBAIJ_6()
178 dk[0] += uik[0] * u0 + uik[1] * u1 + uik[2] * u2 + uik[3] * u3 + uik[4] * u4 + uik[5] * u5; in MatCholeskyFactorNumeric_SeqSBAIJ_6()
179 dk[1] += uik[6] * u0 + uik[7] * u1 + uik[8] * u2 + uik[9] * u3 + uik[10] * u4 + uik[11] * u5; in MatCholeskyFactorNumeric_SeqSBAIJ_6()
180dk[2] += uik[12] * u0 + uik[13] * u1 + uik[14] * u2 + uik[15] * u3 + uik[16] * u4 + uik[17] * u5; in MatCholeskyFactorNumeric_SeqSBAIJ_6()
[all …]
H A Dsbaijfact12.c12 MatScalar *ba = b->a, *aa, *ap, *dk, *uik; in MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering() local
28 PetscCall(PetscMalloc2(49, &dk, 49, &uik)); in MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering()
49 PetscCall(PetscArraycpy(dk, w + k * 49, 49)); in MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering()
169dk[0] += uik[0] * u0 + uik[1] * u1 + uik[2] * u2 + uik[3] * u3 + uik[4] * u4 + uik[5] * u5 + uik[6… in MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering()
170dk[1] += uik[7] * u0 + uik[8] * u1 + uik[9] * u2 + uik[10] * u3 + uik[11] * u4 + uik[12] * u5 + ui… in MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering()
171dk[2] += uik[14] * u0 + uik[15] * u1 + uik[16] * u2 + uik[17] * u3 + uik[18] * u4 + uik[19] * u5 +… in MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering()
172dk[3] += uik[21] * u0 + uik[22] * u1 + uik[23] * u2 + uik[24] * u3 + uik[25] * u4 + uik[26] * u5 +… in MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering()
173dk[4] += uik[28] * u0 + uik[29] * u1 + uik[30] * u2 + uik[31] * u3 + uik[32] * u4 + uik[33] * u5 +… in MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering()
174dk[5] += uik[35] * u0 + uik[36] * u1 + uik[37] * u2 + uik[38] * u3 + uik[39] * u4 + uik[40] * u5 +… in MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering()
175dk[6] += uik[42] * u0 + uik[43] * u1 + uik[44] * u2 + uik[45] * u3 + uik[46] * u4 + uik[47] * u5 +… in MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering()
[all …]
H A Dsbaijfact10.c12 MatScalar *ba = b->a, *aa, *ap, *dk, *uik; in MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering() local
33 PetscCall(PetscMalloc2(36, &dk, 36, &uik)); in MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering()
53 PetscCall(PetscArraycpy(dk, w + k * 36, 36)); in MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering()
183 dk[0] += m0 * u0 + m1 * u1 + m2 * u2 + m3 * u3 + m4 * u4 + m5 * u5; in MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering()
184 dk[1] += m6 * u0 + m7 * u1 + m8 * u2 + m9 * u3 + m10 * u4 + m11 * u5; in MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering()
185 dk[2] += m12 * u0 + m13 * u1 + m14 * u2 + m15 * u3 + m16 * u4 + m17 * u5; in MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering()
186 dk[3] += m18 * u0 + m19 * u1 + m20 * u2 + m21 * u3 + m22 * u4 + m23 * u5; in MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering()
187 dk[4] += m24 * u0 + m25 * u1 + m26 * u2 + m27 * u3 + m28 * u4 + m29 * u5; in MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering()
188 dk[5] += m30 * u0 + m31 * u1 + m32 * u2 + m33 * u3 + m34 * u4 + m35 * u5; in MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering()
190 dk[6] += m0 * u6 + m1 * u7 + m2 * u8 + m3 * u9 + m4 * u10 + m5 * u11; in MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering()
[all …]
H A Dsbaijfact11.c11 MatScalar *ba = b->a, *aa, *ap, *dk, *uik; in MatCholeskyFactorNumeric_SeqSBAIJ_7() local
27 PetscCall(PetscMalloc2(49, &dk, 49, &uik)); in MatCholeskyFactorNumeric_SeqSBAIJ_7()
52 dk[k1] = aa[k * 49 + k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_7()
54 aa[j * 49 + k1] = dk[k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_7()
61 for (k = 0; k < 49; k++) dk[k] = ap[k]; /* dk <- j-th block of aa */ in MatCholeskyFactorNumeric_SeqSBAIJ_7()
63 for (k1 = 0; k1 < 7; k1++) *ap++ = dk[k + 7 * k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_7()
86 PetscCall(PetscArraycpy(dk, w + k * 49, 49)); in MatCholeskyFactorNumeric_SeqSBAIJ_7()
206dk[0] += uik[0] * u0 + uik[1] * u1 + uik[2] * u2 + uik[3] * u3 + uik[4] * u4 + uik[5] * u5 + uik[6… in MatCholeskyFactorNumeric_SeqSBAIJ_7()
207dk[1] += uik[7] * u0 + uik[8] * u1 + uik[9] * u2 + uik[10] * u3 + uik[11] * u4 + uik[12] * u5 + ui… in MatCholeskyFactorNumeric_SeqSBAIJ_7()
208dk[2] += uik[14] * u0 + uik[15] * u1 + uik[16] * u2 + uik[17] * u3 + uik[18] * u4 + uik[19] * u5 +… in MatCholeskyFactorNumeric_SeqSBAIJ_7()
[all …]
H A Dsbaijfact.c566 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()
[all …]
/petsc/src/mat/impls/baij/seq/
H A Dbaijfact.c756 MatScalar *rtmp, *ba = b->a, *bval, *aa = a->a, dk, uikdi; in MatCholeskyFactorNumeric_SeqBAIJ_N() local
805 dk = rtmp[k]; in MatCholeskyFactorNumeric_SeqBAIJ_N()
814 dk += uikdi * ba[ili]; in MatCholeskyFactorNumeric_SeqBAIJ_N()
845 sctx.pv = dk; in MatCholeskyFactorNumeric_SeqBAIJ_N()
848 dk = sctx.pv; in MatCholeskyFactorNumeric_SeqBAIJ_N()
851 ba[bi[k]] = 1.0 / dk; /* U(k,k) */ in MatCholeskyFactorNumeric_SeqBAIJ_N()
893 MatScalar *rtmp, *ba = b->a, *aa = a->a, dk, uikdi, *aval, *bval; in MatCholeskyFactorNumeric_SeqBAIJ_N_NaturalOrdering() local
931 dk = rtmp[k]; in MatCholeskyFactorNumeric_SeqBAIJ_N_NaturalOrdering()
939 dk += uikdi * ba[ili]; in MatCholeskyFactorNumeric_SeqBAIJ_N_NaturalOrdering()
972 sctx.pv = dk; in MatCholeskyFactorNumeric_SeqBAIJ_N_NaturalOrdering()
[all …]
/petsc/src/mat/impls/aij/seq/
H A Daijfact.c1713 MatScalar *rtmp, *ba = b->a, *bval, dk, uikdi; in MatCholeskyFactorNumeric_SeqAIJ() local
1780 dk = rtmp[k]; in MatCholeskyFactorNumeric_SeqAIJ()
1789 dk += uikdi * ba[ili]; /* update diag[k] */ in MatCholeskyFactorNumeric_SeqAIJ()
1825 sctx.pv = dk; in MatCholeskyFactorNumeric_SeqAIJ()
1828 dk = sctx.pv; in MatCholeskyFactorNumeric_SeqAIJ()
1830 ba[bdiag[k]] = 1.0 / dk; /* U(k,k) */ in MatCholeskyFactorNumeric_SeqAIJ()
1880 MatScalar *rtmp, *ba = b->a, *bval, dk, uikdi; in MatCholeskyFactorNumeric_SeqAIJ_inplace() local
1944 dk = rtmp[k]; in MatCholeskyFactorNumeric_SeqAIJ_inplace()
1953 dk += uikdi * ba[ili]; in MatCholeskyFactorNumeric_SeqAIJ_inplace()
1979 sctx.pv = dk; in MatCholeskyFactorNumeric_SeqAIJ_inplace()
[all …]
/petsc/doc/manual/
H A Dtao.md1867 `-tao_bncg_type <gd,fr,pr,prp,hs,dy,hz,dk,kd,ssml_bfgs,ssml_dfp,ssml_brdn>`,