| /petsc/src/ksp/pc/impls/pbjacobi/ |
| H A D | pbjacobi.c | 10 const MatScalar *diag = jac->diag; in PCApply_PBJacobi() local
19 for (i = 0; i < m; i++) yy[i] = diag[i] * xx[i]; in PCApply_PBJacobi()
25 yy[2 * i] = diag[0] * x0 + diag[2] * x1; in PCApply_PBJacobi()
26 yy[2 * i + 1] = diag[1] * x0 + diag[3] * x1; in PCApply_PBJacobi()
27 diag += 4; in PCApply_PBJacobi()
36 yy[3 * i] = diag[0] * x0 + diag[3] * x1 + diag[6] * x2; in PCApply_PBJacobi()
37 yy[3 * i + 1] = diag[1] * x0 + diag[4] * x1 + diag[7] * x2; in PCApply_PBJacobi()
38 yy[3 * i + 2] = diag[2] * x0 + diag[5] * x1 + diag[8] * x2; in PCApply_PBJacobi()
39 diag += 9; in PCApply_PBJacobi()
49 yy[4 * i] = diag[0] * x0 + diag[4] * x1 + diag[8] * x2 + diag[12] * x3; in PCApply_PBJacobi()
[all …]
|
| /petsc/src/ksp/pc/impls/vpbjacobi/ |
| H A D | vpbjacobi.c | 8 const MatScalar *diag = jac->diag; in PCApply_VPBJacobi() local
23 yy[ncnt] = *diag * xx[ncnt]; in PCApply_VPBJacobi()
28 yy[ncnt] = diag[0] * x0 + diag[2] * x1; in PCApply_VPBJacobi()
29 yy[ncnt + 1] = diag[1] * x0 + diag[3] * x1; in PCApply_VPBJacobi()
35 yy[ncnt] = diag[0] * x0 + diag[3] * x1 + diag[6] * x2; in PCApply_VPBJacobi()
36 yy[ncnt + 1] = diag[1] * x0 + diag[4] * x1 + diag[7] * x2; in PCApply_VPBJacobi()
37 yy[ncnt + 2] = diag[2] * x0 + diag[5] * x1 + diag[8] * x2; in PCApply_VPBJacobi()
44 yy[ncnt] = diag[0] * x0 + diag[4] * x1 + diag[8] * x2 + diag[12] * x3; in PCApply_VPBJacobi()
45 yy[ncnt + 1] = diag[1] * x0 + diag[5] * x1 + diag[9] * x2 + diag[13] * x3; in PCApply_VPBJacobi()
46 yy[ncnt + 2] = diag[2] * x0 + diag[6] * x1 + diag[10] * x2 + diag[14] * x3; in PCApply_VPBJacobi()
[all …]
|
| /petsc/src/mat/impls/sbaij/seq/ |
| H A D | sbaijfact5.c | 13 MatScalar *u, *diag, *rtmp, *rtmp_ptr; in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering() local
56 diag = ba + i * 16; in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
59 uik[0] = -(diag[0] * u[0] + diag[4] * u[1] + diag[8] * u[2] + diag[12] * u[3]); in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
60 uik[1] = -(diag[1] * u[0] + diag[5] * u[1] + diag[9] * u[2] + diag[13] * u[3]); in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
61 uik[2] = -(diag[2] * u[0] + diag[6] * u[1] + diag[10] * u[2] + diag[14] * u[3]); in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
62 uik[3] = -(diag[3] * u[0] + diag[7] * u[1] + diag[11] * u[2] + diag[15] * u[3]); in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
64 uik[4] = -(diag[0] * u[4] + diag[4] * u[5] + diag[8] * u[6] + diag[12] * u[7]); in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
65 uik[5] = -(diag[1] * u[4] + diag[5] * u[5] + diag[9] * u[6] + diag[13] * u[7]); in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
66 uik[6] = -(diag[2] * u[4] + diag[6] * u[5] + diag[10] * u[6] + diag[14] * u[7]); in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
67 uik[7] = -(diag[3] * u[4] + diag[7] * u[5] + diag[11] * u[6] + diag[15] * u[7]); in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
[all …]
|
| H A D | sbaijfact6.c | 12 MatScalar *u, *diag, *rtmp, *rtmp_ptr; in MatCholeskyFactorNumeric_SeqSBAIJ_4() local
94 diag = ba + i * 16; in MatCholeskyFactorNumeric_SeqSBAIJ_4()
97 uik[0] = -(diag[0] * u[0] + diag[4] * u[1] + diag[8] * u[2] + diag[12] * u[3]); in MatCholeskyFactorNumeric_SeqSBAIJ_4()
98 uik[1] = -(diag[1] * u[0] + diag[5] * u[1] + diag[9] * u[2] + diag[13] * u[3]); in MatCholeskyFactorNumeric_SeqSBAIJ_4()
99 uik[2] = -(diag[2] * u[0] + diag[6] * u[1] + diag[10] * u[2] + diag[14] * u[3]); in MatCholeskyFactorNumeric_SeqSBAIJ_4()
100 uik[3] = -(diag[3] * u[0] + diag[7] * u[1] + diag[11] * u[2] + diag[15] * u[3]); in MatCholeskyFactorNumeric_SeqSBAIJ_4()
102 uik[4] = -(diag[0] * u[4] + diag[4] * u[5] + diag[8] * u[6] + diag[12] * u[7]); in MatCholeskyFactorNumeric_SeqSBAIJ_4()
103 uik[5] = -(diag[1] * u[4] + diag[5] * u[5] + diag[9] * u[6] + diag[13] * u[7]); in MatCholeskyFactorNumeric_SeqSBAIJ_4()
104 uik[6] = -(diag[2] * u[4] + diag[6] * u[5] + diag[10] * u[6] + diag[14] * u[7]); in MatCholeskyFactorNumeric_SeqSBAIJ_4()
105 uik[7] = -(diag[3] * u[4] + diag[7] * u[5] + diag[11] * u[6] + diag[15] * u[7]); in MatCholeskyFactorNumeric_SeqSBAIJ_4()
[all …]
|
| H A D | sbaijfact4.c | 13 MatScalar *u, *diag, *rtmp, *rtmp_ptr; in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering() local
55 diag = ba + i * 9; in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
58 uik[0] = -(diag[0] * u[0] + diag[3] * u[1] + diag[6] * u[2]); in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
59 uik[1] = -(diag[1] * u[0] + diag[4] * u[1] + diag[7] * u[2]); in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
60 uik[2] = -(diag[2] * u[0] + diag[5] * u[1] + diag[8] * u[2]); in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
62 uik[3] = -(diag[0] * u[3] + diag[3] * u[4] + diag[6] * u[5]); in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
63 uik[4] = -(diag[1] * u[3] + diag[4] * u[4] + diag[7] * u[5]); in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
64 uik[5] = -(diag[2] * u[3] + diag[5] * u[4] + diag[8] * u[5]); in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
66 uik[6] = -(diag[0] * u[6] + diag[3] * u[7] + diag[6] * u[8]); in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
67 uik[7] = -(diag[1] * u[6] + diag[4] * u[7] + diag[7] * u[8]); in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
[all …]
|
| H A D | sbaijfact3.c | 12 MatScalar *u, *diag, *rtmp, *rtmp_ptr; in MatCholeskyFactorNumeric_SeqSBAIJ_3() local
93 diag = ba + i * 9; in MatCholeskyFactorNumeric_SeqSBAIJ_3()
96 uik[0] = -(diag[0] * u[0] + diag[3] * u[1] + diag[6] * u[2]); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
97 uik[1] = -(diag[1] * u[0] + diag[4] * u[1] + diag[7] * u[2]); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
98 uik[2] = -(diag[2] * u[0] + diag[5] * u[1] + diag[8] * u[2]); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
100 uik[3] = -(diag[0] * u[3] + diag[3] * u[4] + diag[6] * u[5]); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
101 uik[4] = -(diag[1] * u[3] + diag[4] * u[4] + diag[7] * u[5]); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
102 uik[5] = -(diag[2] * u[3] + diag[5] * u[4] + diag[8] * u[5]); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
104 uik[6] = -(diag[0] * u[6] + diag[3] * u[7] + diag[6] * u[8]); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
105 uik[7] = -(diag[1] * u[6] + diag[4] * u[7] + diag[7] * u[8]); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
[all …]
|
| /petsc/src/mat/tests/ |
| H A D | ex304.c | 12 Vec diag; in main() local
27 PetscCall(MatCreateVecs(A, &diag, NULL)); in main()
28 PetscCall(VecSet(diag, 2.0)); in main()
29 PetscCall(MatCreateDiagonal(diag, &B_diag)); in main()
34 PetscCall(MatDiagonalSet(B_aij_diag, diag, INSERT_VALUES)); in main()
37 PetscCall(VecDestroy(&diag)); in main()
52 PetscCall(MatDiagonalGetDiagonal(B_diag, &diag)); in main()
53 PetscCall(MatDiagonalScale(result_diag, NULL, diag)); in main()
54 PetscCall(MatDiagonalRestoreDiagonal(B_diag, &diag)); in main()
63 PetscCall(MatDiagonalGetDiagonal(B_diag, &diag)); in main()
[all …]
|
| H A D | ex261.c | 5 static PetscErrorCode CheckDiagonal(Mat A, Vec diag, PetscScalar dval) in CheckDiagonal() argument
27 PetscCall(VecSetRandom(diag, NULL)); in CheckDiagonal()
28 PetscCall(MatGetDiagonal(A, diag)); in CheckDiagonal()
29 PetscCall(VecViewFromOptions(diag, NULL, "-diag_vec_view")); in CheckDiagonal()
31 PetscCall(VecGetLocalSize(diag, &n)); in CheckDiagonal()
32 PetscCall(VecGetArrayRead(diag, &arr)); in CheckDiagonal()
46 PetscCall(VecRestoreArrayRead(diag, &arr)); in CheckDiagonal()
86 Vec diag; in main() local
97 PetscCall(MatCreateVecs(A, &diag, NULL)); in main()
99 PetscCall(CheckDiagonal(A, diag, 0.0)); in main()
[all …]
|
| H A D | ex81.c | 8 Vec diag; in main() local
66 PetscCall(MatCreateVecs(B, NULL, &diag)); in main()
67 PetscCall(MatGetDiagonal(B, diag)); in main()
68 PetscCall(VecSetValue(diag, 0, 100.0, INSERT_VALUES)); in main()
69 PetscCall(VecAssemblyBegin(diag)); in main()
70 PetscCall(VecAssemblyEnd(diag)); in main()
72 PetscCall(MatDiagonalSet(B, diag, INSERT_VALUES)); in main()
75 PetscCall(VecDestroy(&diag)); in main()
|
| H A D | ex207.c | 8 Vec diag; in main() local
22 PetscCall(MatCreateVecs(A, &diag, NULL)); in main()
23 PetscCall(VecSet(diag, 1.0)); in main()
24 PetscCall(MatDiagonalSet(A, diag, INSERT_VALUES)); in main()
34 PetscCall(VecDestroy(&diag)); in main()
|
| /petsc/src/mat/impls/diagonal/ |
| H A D | diagonal.c | 4 Vec diag; member
20 PetscCall(VecCopy(ctx->inv_diag, ctx->diag)); in MatDiagonalSetUpDiagonal()
21 PetscCall(VecReciprocal(ctx->diag)); in MatDiagonalSetUpDiagonal()
34 PetscCall(VecCopy(ctx->diag, ctx->inv_diag)); in MatDiagonalSetUpInverseDiagonal()
49 PetscCall(VecAXPY(yctx->diag, a, xctx->diag)); in MatAXPY_Diagonal()
71 PetscCall(VecGetArrayRead(mat->diag, &v)); in MatGetRow_Diagonal()
74 PetscCall(VecRestoreArrayRead(mat->diag, &v)); in MatGetRow_Diagonal()
85 PetscCall(VecPointwiseMult(y, ctx->diag, x)); in MatMult_Diagonal()
96 PetscCall(VecPointwiseMult(v3, ctx->diag, v1)); in MatMultAdd_Diagonal()
101 PetscCall(VecPointwiseMult(w, ctx->diag, v1)); in MatMultAdd_Diagonal()
[all …]
|
| /petsc/src/ksp/pc/impls/jacobi/ |
| H A D | jacobi.c | 56 …Vec diag; /* vector containing the reciprocals of the diagonal elements of the matrix… member
144 static PetscErrorCode PCJacobiGetDiagonal_Jacobi(PC pc, Vec diag, Vec diagsqrt) in PCJacobiGetDiagonal_Jacobi() argument
150 …PetscCheck(j->diag || j->diagsqrt, comm, PETSC_ERR_ARG_WRONGSTATE, "Jacobi diagonal has not been c… in PCJacobiGetDiagonal_Jacobi()
151 …PetscCheck(!diag || (diag && j->diag), comm, PETSC_ERR_ARG_WRONGSTATE, "Jacobi diagonal not availa… in PCJacobiGetDiagonal_Jacobi()
154 if (diag) PetscCall(VecCopy(j->diag, diag)); in PCJacobiGetDiagonal_Jacobi()
175 Vec diag, diagsqrt; in PCSetUp_Jacobi() local
200 diag = jac->diag; in PCSetUp_Jacobi()
203 if (diag) { in PCSetUp_Jacobi()
206 PetscCall(VecLockReadPop(diag)); in PCSetUp_Jacobi()
209 PetscCall(MatGetDiagonal(pc->pmat, diag)); in PCSetUp_Jacobi()
[all …]
|
| /petsc/src/mat/impls/cdiagonal/ |
| H A D | cdiagonal.c | 4 PetscScalar diag; member
13 yctx->diag += a * xctx->diag; in MatAXPY_ConstantDiagonal()
23 *equal = (yctx->diag == xctx->diag) ? PETSC_TRUE : PETSC_FALSE; in MatEqual_ConstantDiagonal()
39 (*vals)[0] = ctx->diag; in MatGetRow_ConstantDiagonal()
58 PetscCall(VecAXPBY(v3, ctx->diag, 1.0, v1)); in MatMultAdd_ConstantDiagonal()
60 PetscCall(VecAXPBYPCZ(v3, ctx->diag, 1.0, 0.0, v1, v2)); in MatMultAdd_ConstantDiagonal()
71 PetscCall(VecAXPBY(v3, PetscConj(ctx->diag), 1.0, v1)); in MatMultHermitianTransposeAdd_ConstantDiagonal()
73 PetscCall(VecAXPBYPCZ(v3, PetscConj(ctx->diag), 1.0, 0.0, v1, v2)); in MatMultHermitianTransposeAdd_ConstantDiagonal()
84 *nrm = PetscAbsScalar(ctx->diag); in MatNorm_ConstantDiagonal()
112 bctx->diag = actx->diag; in MatDuplicate_ConstantDiagonal()
[all …]
|
| /petsc/src/mat/impls/baij/seq/ |
| H A D | baijsolvtrannat2.c | 7 const PetscInt *diag = a->diag, *vi, n = a->mbs, *ai = a->i, *aj = a->j; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering_inplace() local
18 v = aa + 4 * diag[i]; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering_inplace()
26 vi = aj + diag[i] + 1; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering_inplace()
27 nz = ai[i + 1] - diag[i] - 1; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering_inplace()
40 v = aa + 4 * diag[i] - 4; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering_inplace()
41 vi = aj + diag[i] - 1; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering_inplace()
42 nz = diag[i] - ai[i]; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering_inplace()
61 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *diag = a->diag; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering() local
74 v = aa + bs2 * diag[i]; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering()
82 vi = aj + diag[i] - 1; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering()
[all …]
|
| H A D | baijsolvtrannat3.c | 6 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *diag = a->diag; in MatSolveTranspose_SeqBAIJ_3_NaturalOrdering_inplace() local
18 v = aa + 9 * diag[i]; in MatSolveTranspose_SeqBAIJ_3_NaturalOrdering_inplace()
28 vi = aj + diag[i] + 1; in MatSolveTranspose_SeqBAIJ_3_NaturalOrdering_inplace()
29 nz = ai[i + 1] - diag[i] - 1; in MatSolveTranspose_SeqBAIJ_3_NaturalOrdering_inplace()
44 v = aa + 9 * diag[i] - 9; in MatSolveTranspose_SeqBAIJ_3_NaturalOrdering_inplace()
45 vi = aj + diag[i] - 1; in MatSolveTranspose_SeqBAIJ_3_NaturalOrdering_inplace()
46 nz = diag[i] - ai[i]; in MatSolveTranspose_SeqBAIJ_3_NaturalOrdering_inplace()
67 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *diag = a->diag; in MatSolveTranspose_SeqBAIJ_3_NaturalOrdering() local
80 v = aa + bs2 * diag[i]; in MatSolveTranspose_SeqBAIJ_3_NaturalOrdering()
90 vi = aj + diag[i] - 1; in MatSolveTranspose_SeqBAIJ_3_NaturalOrdering()
[all …]
|
| H A D | baijsolvtran2.c | 9 const PetscInt *diag = a->diag, n = a->mbs, *vi, *ai = a->i, *aj = a->j; in MatSolveTranspose_SeqBAIJ_2_inplace() local
37 v = aa + 4 * diag[i]; in MatSolveTranspose_SeqBAIJ_2_inplace()
45 vi = aj + diag[i] + 1; in MatSolveTranspose_SeqBAIJ_2_inplace()
46 nz = ai[i + 1] - diag[i] - 1; in MatSolveTranspose_SeqBAIJ_2_inplace()
59 v = aa + 4 * diag[i] - 4; in MatSolveTranspose_SeqBAIJ_2_inplace()
60 vi = aj + diag[i] - 1; in MatSolveTranspose_SeqBAIJ_2_inplace()
61 nz = diag[i] - ai[i]; in MatSolveTranspose_SeqBAIJ_2_inplace()
94 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *diag = a->diag; in MatSolveTranspose_SeqBAIJ_2() local
123 v = aa + bs2 * diag[i]; in MatSolveTranspose_SeqBAIJ_2()
131 vi = aj + diag[i] - 1; in MatSolveTranspose_SeqBAIJ_2()
[all …]
|
| H A D | baijsolvtrannat1.c | 6 const PetscInt *adiag = a->diag, *ai = a->i, *aj = a->j, *vi; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering()
55 const PetscInt *diag = a->diag, n = a->mbs, *vi, *ai = a->i, *aj = a->j; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering_inplace() local
65 v = aa + diag[i]; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering_inplace()
68 vi = aj + diag[i] + 1; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering_inplace()
69 nz = ai[i + 1] - diag[i] - 1; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering_inplace()
75 v = aa + diag[i] - 1; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering_inplace()
76 vi = aj + diag[i] - 1; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering_inplace()
77 nz = diag[i] - ai[i]; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering_inplace()
|
| H A D | baijsolvtrannat4.c | 6 const PetscInt *diag = a->diag, n = a->mbs, *vi, *ai = a->i, *aj = a->j; in MatSolveTranspose_SeqBAIJ_4_NaturalOrdering_inplace() local
18 v = aa + 16 * diag[i]; in MatSolveTranspose_SeqBAIJ_4_NaturalOrdering_inplace()
30 vi = aj + diag[i] + 1; in MatSolveTranspose_SeqBAIJ_4_NaturalOrdering_inplace()
31 nz = ai[i + 1] - diag[i] - 1; in MatSolveTranspose_SeqBAIJ_4_NaturalOrdering_inplace()
48 v = aa + 16 * diag[i] - 16; in MatSolveTranspose_SeqBAIJ_4_NaturalOrdering_inplace()
49 vi = aj + diag[i] - 1; in MatSolveTranspose_SeqBAIJ_4_NaturalOrdering_inplace()
50 nz = diag[i] - ai[i]; in MatSolveTranspose_SeqBAIJ_4_NaturalOrdering_inplace()
73 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *diag = a->diag; in MatSolveTranspose_SeqBAIJ_4_NaturalOrdering() local
86 v = aa + bs2 * diag[i]; in MatSolveTranspose_SeqBAIJ_4_NaturalOrdering()
98 vi = aj + diag[i] - 1; in MatSolveTranspose_SeqBAIJ_4_NaturalOrdering()
[all …]
|
| H A D | baijsolvtrann.c | 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()
37 vi = aj + a->diag[i] + 1; in MatSolveTranspose_SeqBAIJ_N_inplace()
38 nz = ai[i + 1] - a->diag[i] - 1; in MatSolveTranspose_SeqBAIJ_N_inplace()
49 nz = a->diag[i] - ai[i]; in MatSolveTranspose_SeqBAIJ_N_inplace()
74 const PetscInt n = a->mbs, *ai = a->i, *aj = a->j, *vi, *diag = a->diag; 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()
103 vi = aj + diag[i] - 1; in MatSolveTranspose_SeqBAIJ_N()
104 nz = diag[i] - diag[i + 1] - 1; in MatSolveTranspose_SeqBAIJ_N()
|
| H A D | baijsolvtran3.c | 9 const PetscInt *diag = a->diag, n = a->mbs, *vi, *ai = a->i, *aj = a->j; in MatSolveTranspose_SeqBAIJ_3_inplace() local
38 v = aa + 9 * diag[i]; in MatSolveTranspose_SeqBAIJ_3_inplace()
48 vi = aj + diag[i] + 1; in MatSolveTranspose_SeqBAIJ_3_inplace()
49 nz = ai[i + 1] - diag[i] - 1; in MatSolveTranspose_SeqBAIJ_3_inplace()
64 v = aa + 9 * diag[i] - 9; in MatSolveTranspose_SeqBAIJ_3_inplace()
65 vi = aj + diag[i] - 1; in MatSolveTranspose_SeqBAIJ_3_inplace()
66 nz = diag[i] - ai[i]; in MatSolveTranspose_SeqBAIJ_3_inplace()
102 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *diag = a->diag; in MatSolveTranspose_SeqBAIJ_3() local
132 v = aa + bs2 * diag[i]; in MatSolveTranspose_SeqBAIJ_3()
142 vi = aj + diag[i] - 1; in MatSolveTranspose_SeqBAIJ_3()
[all …]
|
| H A D | baijsolvtran1.c | 8 const PetscInt *rout, *cout, *r, *c, *adiag = a->diag, *ai = a->i, *aj = a->j, *vi; in MatSolveTranspose_SeqBAIJ_1()
65 const PetscInt *diag = a->diag, n = a->mbs, *vi, *ai = a->i, *aj = a->j; in MatSolveTranspose_SeqBAIJ_1_inplace() local
86 v = aa + diag[i]; in MatSolveTranspose_SeqBAIJ_1_inplace()
89 vi = aj + diag[i] + 1; in MatSolveTranspose_SeqBAIJ_1_inplace()
90 nz = ai[i + 1] - diag[i] - 1; in MatSolveTranspose_SeqBAIJ_1_inplace()
96 v = aa + diag[i] - 1; in MatSolveTranspose_SeqBAIJ_1_inplace()
97 vi = aj + diag[i] - 1; in MatSolveTranspose_SeqBAIJ_1_inplace()
98 nz = diag[i] - ai[i]; in MatSolveTranspose_SeqBAIJ_1_inplace()
|
| H A D | baijsolvtrannat5.c | 6 const PetscInt *diag = a->diag, n = a->mbs, *vi, *ai = a->i, *aj = a->j; in MatSolveTranspose_SeqBAIJ_5_NaturalOrdering_inplace() local
18 v = aa + 25 * diag[i]; in MatSolveTranspose_SeqBAIJ_5_NaturalOrdering_inplace()
32 vi = aj + diag[i] + 1; in MatSolveTranspose_SeqBAIJ_5_NaturalOrdering_inplace()
33 nz = ai[i + 1] - diag[i] - 1; in MatSolveTranspose_SeqBAIJ_5_NaturalOrdering_inplace()
52 v = aa + 25 * diag[i] - 25; in MatSolveTranspose_SeqBAIJ_5_NaturalOrdering_inplace()
53 vi = aj + diag[i] - 1; in MatSolveTranspose_SeqBAIJ_5_NaturalOrdering_inplace()
54 nz = diag[i] - ai[i]; in MatSolveTranspose_SeqBAIJ_5_NaturalOrdering_inplace()
79 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *diag = a->diag; in MatSolveTranspose_SeqBAIJ_5_NaturalOrdering() local
92 v = aa + bs2 * diag[i]; in MatSolveTranspose_SeqBAIJ_5_NaturalOrdering()
106 vi = aj + diag[i] - 1; in MatSolveTranspose_SeqBAIJ_5_NaturalOrdering()
[all …]
|
| /petsc/src/ksp/ksp/utils/lmvm/tests/ |
| H A D | ex2.c | 9 Mat sym, diag; in main() local
25 PetscCall(MatCreateLMVMDiagBroyden(comm, 1, 1, &diag)); in main()
28 PetscCall(MatLMVMSetHistorySize(diag, m)); in main()
40 PetscCall(MatSetOptionsPrefix(diag, "diag_")); in main()
41 PetscCall(MatSetFromOptions(diag)); in main()
44 PetscCall(MatSetUp(diag)); in main()
47 PetscCall(MatViewFromOptions(diag, NULL, "-view")); in main()
62 PetscCall(MatLMVMUpdate(diag, x, g)); in main()
78 PetscCall(MatLMVMUpdate(diag, x, g)); in main()
82 PetscCall(MatMult(diag, u, v_diag)); in main()
[all …]
|
| /petsc/src/ksp/ksp/tests/ |
| H A D | ex22.c | 10 Vec b, x, f, h, diag, x1, x2; in test_solve() local
21 PetscCall(VecCreate(PETSC_COMM_WORLD, &diag)); in test_solve()
22 PetscCall(VecSetSizes(diag, PETSC_DECIDE, n)); in test_solve()
23 PetscCall(VecSetFromOptions(diag)); in test_solve()
25 PetscCall(VecSet(diag, 1.0 / 10.0)); /* so inverse = diag(10) */ in test_solve()
33 PetscCall(MatDiagonalSet(A11, diag, INSERT_VALUES)); in test_solve()
35 PetscCall(VecDestroy(&diag)); in test_solve()
121 Vec b, x, f, h, diag, x1, x2; in test_solve_matgetvecs() local
133 PetscCall(VecCreate(PETSC_COMM_WORLD, &diag)); in test_solve_matgetvecs()
134 PetscCall(VecSetSizes(diag, PETSC_DECIDE, n)); in test_solve_matgetvecs()
[all …]
|
| /petsc/src/ksp/ksp/tutorials/ |
| H A D | ex15.c | 18 Vec diag; member
246 newctx->diag = 0; in SampleShellPCCreate()
272 Vec diag; in SampleShellPCSetUp() local
276 PetscCall(VecDuplicate(x, &diag)); in SampleShellPCSetUp()
277 PetscCall(MatGetDiagonal(pmat, diag)); in SampleShellPCSetUp()
278 PetscCall(VecReciprocal(diag)); in SampleShellPCSetUp()
280 shell->diag = diag; in SampleShellPCSetUp()
306 PetscCall(VecPointwiseMult(y, x, shell->diag)); in SampleShellPCApply()
323 PetscCall(VecDestroy(&shell->diag)); in SampleShellPCDestroy()
|