Home
last modified time | relevance | path

Searched refs:yy (Results 1 – 25 of 87) sorted by relevance

1234

/petsc/src/ksp/pc/impls/pbjacobi/
H A Dpbjacobi.c11 PetscScalar *yy, x0, x1, x2, x3, x4, x5, x6; in PCApply_PBJacobi() local
16 PetscCall(VecGetArray(y, &yy)); in PCApply_PBJacobi()
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()
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()
49 yy[4 * i] = diag[0] * x0 + diag[4] * x1 + diag[8] * x2 + diag[12] * x3; in PCApply_PBJacobi()
50 yy[4 * i + 1] = diag[1] * x0 + diag[5] * x1 + diag[9] * x2 + diag[13] * x3; in PCApply_PBJacobi()
[all …]
/petsc/src/ksp/pc/impls/vpbjacobi/
H A Dvpbjacobi.c11 PetscScalar *yy, x0, x1, x2, x3, x4, x5, x6; in PCApply_VPBJacobi() local
18 PetscCall(VecGetArray(y, &yy)); in PCApply_VPBJacobi()
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()
[all …]
/petsc/src/vec/vec/impls/seq/
H A Ddvec2.c16 Vec *yy = (Vec *)yin; in VecMDot_Seq() local
22 PetscCall(VecGetArrayRead(yy[0], &yy0)); in VecMDot_Seq()
23 PetscCall(VecGetArrayRead(yy[1], &yy1)); in VecMDot_Seq()
24 PetscCall(VecGetArrayRead(yy[2], &yy2)); in VecMDot_Seq()
26 PetscCall(VecRestoreArrayRead(yy[0], &yy0)); in VecMDot_Seq()
27 PetscCall(VecRestoreArrayRead(yy[1], &yy1)); in VecMDot_Seq()
28 PetscCall(VecRestoreArrayRead(yy[2], &yy2)); in VecMDot_Seq()
34 PetscCall(VecGetArrayRead(yy[0], &yy0)); in VecMDot_Seq()
35 PetscCall(VecGetArrayRead(yy[1], &yy1)); in VecMDot_Seq()
37 PetscCall(VecRestoreArrayRead(yy[0], &yy0)); in VecMDot_Seq()
[all …]
H A Dbvec1.c103 PetscScalar *yy; in VecAXPBY_Seq() local
106 PetscCall(VecGetArray(yin, &yy)); in VecAXPBY_Seq()
108 for (PetscInt i = 0; i < n; ++i) yy[i] = a * xx[i]; in VecAXPBY_Seq()
111 for (PetscInt i = 0; i < n; ++i) yy[i] = a * xx[i] + b * yy[i]; in VecAXPBY_Seq()
115 PetscCall(VecRestoreArray(yin, &yy)); in VecAXPBY_Seq()
123 const PetscScalar *yy, *xx; in VecAXPBYPCZ_Seq() local
129 PetscCall(VecGetArrayRead(yin, &yy)); in VecAXPBYPCZ_Seq()
132 for (PetscInt i = 0; i < n; ++i) zz[i] = xx[i] + beta * yy[i] + gamma * zz[i]; in VecAXPBYPCZ_Seq()
134 for (PetscInt i = 0; i < n; ++i) zz[i] = alpha * xx[i] + beta * yy[i] + zz[i]; in VecAXPBYPCZ_Seq()
136 for (PetscInt i = 0; i < n; ++i) zz[i] = alpha * xx[i] + beta * yy[i]; in VecAXPBYPCZ_Seq()
[all …]
/petsc/src/ts/tutorials/phasefield/
H A Dpotentials.c11 PetscReal x, hx = .1 / Mx, pause, xx[3], yy[3]; in main() local
36 yy[0] = (1. - x * x) * (1. - x * x); in main()
37 yy[1] = (1. - x * x); in main()
38 yy[2] = -(1. - x) * PetscLogReal(1. - x); in main()
39 PetscCall(PetscDrawLGAddPoint(lg, xx, yy)); in main()
56 yy[0] = x * x * x - x; in main()
57 yy[1] = -x; in main()
58 yy[2] = 1.0 + PetscLogReal(1. - x); in main()
59 PetscCall(PetscDrawLGAddPoint(lg, xx, yy)); in main()
H A Dbiharmonic.c462 …PetscReal x, y, hx, pause, sx, len, max, xx[4], yy[4], xx_netforce, yy_netforce, yup, ydown, y2… in MyMonitor() local
512 …if (ctx->degenerate) yy[0] = PetscRealPart(.25 * (1. - u[i] * u[i]) * ctx->kappa * (u[i - 1] - u[i… in MyMonitor()
513 … else yy[0] = PetscRealPart(.25 * ctx->kappa * (u[i - 1] - u[i + 1]) * (u[i - 1] - u[i + 1]) * sx); in MyMonitor()
518 yy[1] = .25 * PetscRealPart((1. - u[i] * u[i]) * (1. - u[i] * u[i])); in MyMonitor()
521 yy[1] = .5 * PetscRealPart(1. - u[i] * u[i]); in MyMonitor()
524 yy[1] = .25 * PetscRealPart((1. - u[i] * u[i]) * (1. - u[i] * u[i])); in MyMonitor()
525 …if (PetscRealPart(u[i]) < -1.0 + 2.0 * tol) yy[2] = .5 * theta * (2.0 * tol * PetscLogReal(tol) + … in MyMonitor()
526 …else if (PetscRealPart(u[i]) > 1.0 - 2.0 * tol) yy[2] = .5 * theta * (PetscRealPart(1.0 + u[i]) * … in MyMonitor()
527 …else yy[2] = .5 * theta * (PetscRealPart(1.0 + u[i]) * PetscLogReal(PetscRealPart(1.0 + u[i]) / 2.… in MyMonitor()
530 yy[1] = .5 * theta_c * PetscRealPart(1.0 - u[i] * u[i]); in MyMonitor()
[all …]
H A Dheat.c280 PetscReal x, y, hx, pause, sx, len, max, xx[2], yy[2]; in MyMonitor() local
327 …yy[0] = PetscRealPart(.25 * ctx->kappa * (u[i - 1] - u[i + 1]) * (u[i - 1] - u[i + 1]) * s… in MyMonitor()
328 if (ctx->allencahn) yy[1] = .25 * PetscRealPart((1. - u[i] * u[i]) * (1. - u[i] * u[i])); in MyMonitor()
329 PetscCall(PetscDrawLGAddPoint(lg, xx, yy)); in MyMonitor()
347 yy[0] = PetscRealPart(ctx->kappa * (u[i - 1] + u[i + 1] - 2.0 * u[i]) * sx); in MyMonitor()
348 max = PetscMax(max, PetscAbs(yy[0])); in MyMonitor()
350 yy[1] = PetscRealPart(u[i] - u[i] * u[i] * u[i]); in MyMonitor()
351 max = PetscMax(max, PetscAbs(yy[1])); in MyMonitor()
353 PetscCall(PetscDrawLGAddPoint(lg, xx, yy)); in MyMonitor()
371 yy[0] = PetscRealPart(u[i]); in MyMonitor()
[all …]
/petsc/src/mat/tests/
H A Dex60.c11 Vec yy; in main() local
57 PetscCall(MatCreateVecs(C, NULL, &yy)); in main()
58 PetscCall(VecSetFromOptions(yy)); in main()
60 PetscCall(MatGetColumnVector(C, yy, col)); in main()
62 PetscCall(VecView(yy, PETSC_VIEWER_STDOUT_WORLD)); in main()
64 PetscCall(VecDestroy(&yy)); in main()
H A Dex229.c17 PetscScalar xx, yy; in myF() local
19 yy = 0.0; in myF()
22 yy += xx; in myF()
24 ay[i] = yy; in myF()
H A Dex53.c11 Vec xx, yy, s1, s2; in main() local
44 PetscCall(VecDuplicate(xx, &yy)); in main()
74 PetscCall(VecSetRandom(yy, rand)); in main()
75 PetscCall(MatMultAdd(A, xx, yy, s1)); in main()
76 PetscCall(MatMultAdd(B, xx, yy, s2)); in main()
95 PetscCall(VecSetRandom(yy, rand)); in main()
96 PetscCall(MatMultTransposeAdd(A, xx, yy, s1)); in main()
97 PetscCall(MatMultTransposeAdd(B, xx, yy, s2)); in main()
179 PetscCall(VecDestroy(&yy)); in main()
H A Dex48.c8 Vec xx, s1, s2, yy; in main() local
32 PetscCall(VecDuplicate(xx, &yy)); in main()
166 PetscCall(VecSetRandom(yy, rdm)); in main()
167 PetscCall(MatForwardSolve(Fact, yy, xx)); in main()
171 PetscCall(MatMultAdd(A, s1, yy, yy)); in main()
172 PetscCall(VecNorm(yy, NORM_2, &rnorm)); in main()
182 PetscCall(VecSetRandom(yy, rdm)); in main()
183 PetscCall(MatSolveAdd(B, xx, yy, s2)); in main()
184 PetscCall(MatSolveAdd(A, xx, yy, s1)); in main()
235 PetscCall(VecDestroy(&yy)); in main()
/petsc/src/binding/petsc4py/demo/legacy/poisson3d/
H A Ddel2mat.h29 PetscScalar *yy; in Del2Mat_mult() local
35 PetscCall(VecGetArray(y,&yy)); in Del2Mat_mult()
37 Del2Apply(&ctx->N,ctx->F,xx,yy); in Del2Mat_mult()
40 PetscCall(VecRestoreArray(y,&yy)); in Del2Mat_mult()
/petsc/src/mat/impls/baij/seq/baijmkl/
H A Dbaijmkl.c209 static PetscErrorCode MatMult_SeqBAIJMKL_SpMV2(Mat A, Vec xx, Vec yy) in MatMult_SeqBAIJMKL_SpMV2() argument
219 PetscCall(VecSet(yy, 0.0)); in MatMult_SeqBAIJMKL_SpMV2()
224 PetscCall(VecGetArray(yy, &y)); in MatMult_SeqBAIJMKL_SpMV2()
236 PetscCall(VecRestoreArray(yy, &y)); in MatMult_SeqBAIJMKL_SpMV2()
240 static PetscErrorCode MatMultTranspose_SeqBAIJMKL_SpMV2(Mat A, Vec xx, Vec yy) in MatMultTranspose_SeqBAIJMKL_SpMV2() argument
250 PetscCall(VecSet(yy, 0.0)); in MatMultTranspose_SeqBAIJMKL_SpMV2()
255 PetscCall(VecGetArray(yy, &y)); in MatMultTranspose_SeqBAIJMKL_SpMV2()
267 PetscCall(VecRestoreArray(yy, &y)); in MatMultTranspose_SeqBAIJMKL_SpMV2()
271 static PetscErrorCode MatMultAdd_SeqBAIJMKL_SpMV2(Mat A, Vec xx, Vec yy, Vec zz) in MatMultAdd_SeqBAIJMKL_SpMV2() argument
283 PetscCall(VecCopy(yy, zz)); in MatMultAdd_SeqBAIJMKL_SpMV2()
[all …]
/petsc/src/mat/impls/aij/seq/aijmkl/
H A Daijmkl.c337 static PetscErrorCode MatMult_SeqAIJMKL(Mat A, Vec xx, Vec yy) in MatMult_SeqAIJMKL() argument
357 PetscCall(VecGetArray(yy, &y)); in MatMult_SeqAIJMKL()
367 PetscCall(VecRestoreArray(yy, &y)); in MatMult_SeqAIJMKL()
373 PetscErrorCode MatMult_SeqAIJMKL_SpMV2(Mat A, Vec xx, Vec yy) in MatMult_SeqAIJMKL_SpMV2() argument
384 PetscCall(VecGetArray(yy, &y)); in MatMult_SeqAIJMKL_SpMV2()
386 PetscCall(VecRestoreArray(yy, &y)); in MatMult_SeqAIJMKL_SpMV2()
391 PetscCall(VecGetArray(yy, &y)); in MatMult_SeqAIJMKL_SpMV2()
404 PetscCall(VecRestoreArray(yy, &y)); in MatMult_SeqAIJMKL_SpMV2()
410 static PetscErrorCode MatMultTranspose_SeqAIJMKL(Mat A, Vec xx, Vec yy) in MatMultTranspose_SeqAIJMKL() argument
430 PetscCall(VecGetArray(yy, &y)); in MatMultTranspose_SeqAIJMKL()
[all …]
/petsc/src/mat/utils/
H A Dgetcolv.c25 PetscErrorCode MatGetColumnVector(Mat A, Vec yy, PetscInt col) in MatGetColumnVector() argument
34 PetscValidHeaderSpecific(yy, VEC_CLASSID, 2); in MatGetColumnVector()
40 PetscCall(VecGetOwnershipRange(yy, &rs, &re)); in MatGetColumnVector()
43 if (A->ops->getcolumnvector) PetscUseTypeMethod(A, getcolumnvector, yy, col); in MatGetColumnVector()
45 PetscCall(VecSet(yy, 0.0)); in MatGetColumnVector()
46 PetscCall(VecGetArray(yy, &y)); in MatGetColumnVector()
63 PetscCall(VecRestoreArray(yy, &y)); in MatGetColumnVector()
/petsc/src/ksp/pc/impls/tfs/
H A Dtfs.c34 PetscScalar *yy; in PCApply_TFS_XXT() local
39 PetscCall(VecGetArray(y, &yy)); in PCApply_TFS_XXT()
40 PetscCall(XXT_solve(tfs->xxt, yy, (PetscScalar *)xx)); in PCApply_TFS_XXT()
42 PetscCall(VecRestoreArray(y, &yy)); in PCApply_TFS_XXT()
49 PetscScalar *yy; in PCApply_TFS_XYT() local
54 PetscCall(VecGetArray(y, &yy)); in PCApply_TFS_XYT()
55 PetscCall(XYT_solve(tfs->xyt, yy, (PetscScalar *)xx)); in PCApply_TFS_XYT()
57 PetscCall(VecRestoreArray(y, &yy)); in PCApply_TFS_XYT()
/petsc/src/snes/tutorials/
H A Dex15.c330 … const PetscReal xx = 2 * (PetscReal)i / (Mx - 1) - 1, yy = 2 * (PetscReal)j / (My - 1) - 1; in FormInitialGuess() local
331 x[j][i] = (1 - xx * xx) * (1 - yy * yy) * xx * yy; in FormInitialGuess()
336 … const PetscReal xx = 2 * (PetscReal)i / (Mx - 1) - 1, yy = 2 * (PetscReal)j / (My - 1) - 1; in FormInitialGuess() local
337 x[j][i] = (1 - xx * xx) * (1 - yy * yy) * xx * yy; in FormInitialGuess()
426 PetscReal xx = i * hx, yy = j * hy; in FormFunctionLocal() local
430 …e_E = eta(user, xx, yy, ux_E, uy_E), e_W = eta(user, xx, yy, ux_W, uy_W), e_N = eta(user, xx, yy, … in FormFunctionLocal()
507 PetscReal xx = i * hx, yy = j * hy; in FormJacobianLocal() local
516 …yy, ux_E, uy_E), e_W = eta(user, xx, yy, ux_W, uy_W), e_N = eta(user, xx, yy, ux_N, uy_N), e_S = e… in FormJacobianLocal()
772 PetscReal xx = i * hx, yy = j * hy; in NonlinearGS() local
784 …yy, ux_E, uy_E), e_W = eta(user, xx, yy, ux_W, uy_W), e_N = eta(user, xx, yy, ux_N, uy_N), e_S = e… in NonlinearGS()
/petsc/src/mat/impls/maij/
H A Dmaij.c194 …tic inline PetscErrorCode MatMult_MatMultAdd_SeqMAIJ_Template(Mat A, Vec xx, Vec yy, Vec zz, int N) in MatMult_MatMultAdd_SeqMAIJ_Template() argument
196 const PetscBool mult_add = yy == NULL ? PETSC_FALSE : PETSC_TRUE; in MatMult_MatMultAdd_SeqMAIJ_Template()
211 if (mult_add && yy != zz) PetscCall(VecCopy(yy, zz)); in MatMult_MatMultAdd_SeqMAIJ_Template()
255 …rorCode MatMultTranspose_MatMultTransposeAdd_SeqMAIJ_Template(Mat A, Vec xx, Vec yy, Vec zz, int N) in MatMultTranspose_MatMultTransposeAdd_SeqMAIJ_Template() argument
257 const PetscBool mult_add = yy == NULL ? PETSC_FALSE : PETSC_TRUE; in MatMultTranspose_MatMultTransposeAdd_SeqMAIJ_Template()
272 if (yy != zz) PetscCall(VecCopy(yy, zz)); in MatMultTranspose_MatMultTransposeAdd_SeqMAIJ_Template()
304 static PetscErrorCode PetscConcat(MatMult_SeqMAIJ_, N)(Mat A, Vec xx, Vec yy) \
307 PetscCall(MatMult_MatMultAdd_SeqMAIJ_Template(A, xx, NULL, yy, N)); \
310 static PetscErrorCode PetscConcat(MatMultTranspose_SeqMAIJ_, N)(Mat A, Vec xx, Vec yy) \
313 PetscCall(MatMultTranspose_MatMultTransposeAdd_SeqMAIJ_Template(A, xx, NULL, yy, N)); \
[all …]
/petsc/src/dm/tests/
H A Dex27.c26 Vec coords, x, y, z, xx, yy, zz; in main() local
81 PetscCall(DMGetGlobalVector(da, &yy)); in main()
82 PetscCall(PetscObjectSetName((PetscObject)yy, "FFTW frequency space vector")); in main()
131 PetscCall(MatMult(AA, xx, yy)); in main()
133 PetscCall(MatMultTranspose(AA, yy, zz)); in main()
140 PetscCall(VecView(yy, PETSC_VIEWER_STDOUT_WORLD)); in main()
166 PetscCall(VecAXPY(y, -1.0, yy)); in main()
182 PetscCall(DMRestoreGlobalVector(da, &yy)); in main()
H A Dex36.c61 PetscScalar xx, yy, zz; in DAApplyConformalMapping() local
105 xx = yy = zz = 0.0; in DAApplyConformalMapping()
108 yy += Ni[p] * yn[p]; in DAApplyConformalMapping()
112 XX[dim * i + 1] = yy; in DAApplyConformalMapping()
120 yy = XX[dim * i + 1] + 1.5; in DAApplyConformalMapping()
123 zeta.imag = PetscRealPart(yy); in DAApplyConformalMapping()
134 yy = XX[dim * i + 1]; in DAApplyConformalMapping()
136 zeta.imag = PetscRealPart(yy); in DAApplyConformalMapping()
145 yy = XX[dim * i + 1] + 1.5; in DAApplyConformalMapping()
148 zeta.imag = PetscRealPart(yy); in DAApplyConformalMapping()
[all …]
/petsc/src/mat/impls/aij/seq/aijsell/
H A Daijsell.c147 static PetscErrorCode MatMult_SeqAIJSELL(Mat A, Vec xx, Vec yy) in MatMult_SeqAIJSELL() argument
153 PetscCall(MatMult_SeqSELL(aijsell->S, xx, yy)); in MatMult_SeqAIJSELL()
157 static PetscErrorCode MatMultTranspose_SeqAIJSELL(Mat A, Vec xx, Vec yy) in MatMultTranspose_SeqAIJSELL() argument
163 PetscCall(MatMultTranspose_SeqSELL(aijsell->S, xx, yy)); in MatMultTranspose_SeqAIJSELL()
167 static PetscErrorCode MatMultAdd_SeqAIJSELL(Mat A, Vec xx, Vec yy, Vec zz) in MatMultAdd_SeqAIJSELL() argument
173 PetscCall(MatMultAdd_SeqSELL(aijsell->S, xx, yy, zz)); in MatMultAdd_SeqAIJSELL()
177 static PetscErrorCode MatMultTransposeAdd_SeqAIJSELL(Mat A, Vec xx, Vec yy, Vec zz) in MatMultTransposeAdd_SeqAIJSELL() argument
183 PetscCall(MatMultTransposeAdd_SeqSELL(aijsell->S, xx, yy, zz)); in MatMultTransposeAdd_SeqAIJSELL()
/petsc/src/vec/vec/tests/
H A Dex33.c9 PetscScalar value, *yy; in main() local
39 PetscCall(VecGetArray(y, &yy)); in main()
41 PetscCall(VecCreateMPIWithArray(PETSC_COMM_WORLD, 1, len, PETSC_DECIDE, yy, &y_t)); in main()
44 PetscCall(VecRestoreArray(y, &yy)); in main()
/petsc/src/ksp/pc/impls/pbjacobi/cuda/
H A Dpbjacobi_cuda.cu36 PetscScalar *yy; in PCApplyOrTranspose_PBJacobi_CUDA() local
44 PetscCall(VecCUDAGetArrayWrite(y, &yy)); in PCApplyOrTranspose_PBJacobi_CUDA()
51 …XgemvStridedBatched(handle, op, bs, bs, &alpha, A, bs, bs * bs, xx, 1, bs, &beta, yy, 1, bs, mbs)); in PCApplyOrTranspose_PBJacobi_CUDA()
54 …MatMultBatched<<<gridSize, 256>>>(bs, mbs, A, xx, yy, op == CUBLAS_OP_T ? PETSC_TRUE : PETSC_FALSE… in PCApplyOrTranspose_PBJacobi_CUDA()
58 PetscCall(VecCUDARestoreArrayWrite(y, &yy)); in PCApplyOrTranspose_PBJacobi_CUDA()
/petsc/src/mat/impls/centering/
H A Dcentering.c3 static PetscErrorCode MatMult_Centering(Mat A, Vec xx, Vec yy) in MatMult_Centering() argument
15 PetscCall(VecGetArray(yy, &y)); in MatMult_Centering()
18 PetscCall(VecRestoreArray(yy, &y)); in MatMult_Centering()
/petsc/src/ksp/ksp/tutorials/
H A Dexample100.py54 yy = y.getArray(readonly=0) # to numpy array
55 yy[0] = 2.0*xx[0] - xx[1]
56 yy[1:-1] = - xx[:-2] + 2.0*xx[1:-1] - xx[2:]
57 yy[-1] = - xx[-2] + 2.0*xx[-1]
59 yy *= 1.0/h**2

1234