Home
last modified time | relevance | path

Searched refs:ax (Results 1 – 24 of 24) sorted by relevance

/petsc/src/mat/impls/baij/seq/ftn-kernels/
H A Dfsolvebaij.F9020 PetscInt :: idx, ax, jdx
38 ax = 16*jstart
44 …s(0) = s(0) - (a(ax + 0)*x(jdx + 0) + a(ax + 4)*x(jdx + 1) + a(ax + 8)*x(jdx + 2) + a(ax + 12)*x(j…
45 …s(1) = s(1) - (a(ax + 1)*x(jdx + 0) + a(ax + 5)*x(jdx + 1) + a(ax + 9)*x(jdx + 2) + a(ax + 13)*x(j…
46 …s(2) = s(2) - (a(ax + 2)*x(jdx + 0) + a(ax + 6)*x(jdx + 1) + a(ax + 10)*x(jdx + 2) + a(ax + 14)*x(…
47 …s(3) = s(3) - (a(ax + 3)*x(jdx + 0) + a(ax + 7)*x(jdx + 1) + a(ax + 11)*x(jdx + 2) + a(ax + 15)*x(…
48 ax = ax + 16
59 ax = 16*jstart
63 …s(0) = s(0) - (a(ax + 0)*x(jdx + 0) + a(ax + 4)*x(jdx + 1) + a(ax + 8)*x(jdx + 2) + a(ax + 12)*x(j…
64 …s(1) = s(1) - (a(ax + 1)*x(jdx + 0) + a(ax + 5)*x(jdx + 1) + a(ax + 9)*x(jdx + 2) + a(ax + 13)*x(j…
[all …]
/petsc/src/mat/impls/baij/seq/
H A Ddgefa2.c18 MatScalar *aa, *ax, *ay, work[4], stmp; in PetscKernel_A_gets_inverse_A_2() local
72 ax = &a[k4 + 1]; in PetscKernel_A_gets_inverse_A_2()
83 for (ll = 0; ll < i__3; ll++) ay[ll] += stmp * ax[ll]; in PetscKernel_A_gets_inverse_A_2()
105 ax = aa; in PetscKernel_A_gets_inverse_A_2()
111 for (ll = 0; ll < k; ll++) ay[ll] += stmp * ax[ll]; in PetscKernel_A_gets_inverse_A_2()
126 ax = &a[2 * j + 1]; in PetscKernel_A_gets_inverse_A_2()
128 ay[0] += stmp * ax[0]; in PetscKernel_A_gets_inverse_A_2()
129 ay[1] += stmp * ax[1]; in PetscKernel_A_gets_inverse_A_2()
133 ax = &a[k3 + 1]; in PetscKernel_A_gets_inverse_A_2()
135 stmp = ax[0]; in PetscKernel_A_gets_inverse_A_2()
[all …]
H A Ddgefa7.c18 MatScalar *aa, *ax, *ay, work[49], stmp; in PetscKernel_A_gets_inverse_A_7() local
73 ax = &a[k4 + 1]; in PetscKernel_A_gets_inverse_A_7()
84 for (ll = 0; ll < i__3; ll++) ay[ll] += stmp * ax[ll]; in PetscKernel_A_gets_inverse_A_7()
106 ax = aa; in PetscKernel_A_gets_inverse_A_7()
112 for (ll = 0; ll < k; ll++) ay[ll] += stmp * ax[ll]; in PetscKernel_A_gets_inverse_A_7()
128 ax = &a[7 * j + 1]; in PetscKernel_A_gets_inverse_A_7()
130 ay[0] += stmp * ax[0]; in PetscKernel_A_gets_inverse_A_7()
131 ay[1] += stmp * ax[1]; in PetscKernel_A_gets_inverse_A_7()
132 ay[2] += stmp * ax[2]; in PetscKernel_A_gets_inverse_A_7()
133 ay[3] += stmp * ax[3]; in PetscKernel_A_gets_inverse_A_7()
[all …]
H A Ddgefa6.c18 MatScalar *aa, *ax, *ay, work[36], stmp; in PetscKernel_A_gets_inverse_A_6() local
73 ax = &a[k4 + 1]; in PetscKernel_A_gets_inverse_A_6()
84 for (ll = 0; ll < i__3; ll++) ay[ll] += stmp * ax[ll]; in PetscKernel_A_gets_inverse_A_6()
106 ax = aa; in PetscKernel_A_gets_inverse_A_6()
112 for (ll = 0; ll < k; ll++) ay[ll] += stmp * ax[ll]; in PetscKernel_A_gets_inverse_A_6()
128 ax = &a[6 * j + 1]; in PetscKernel_A_gets_inverse_A_6()
130 ay[0] += stmp * ax[0]; in PetscKernel_A_gets_inverse_A_6()
131 ay[1] += stmp * ax[1]; in PetscKernel_A_gets_inverse_A_6()
132 ay[2] += stmp * ax[2]; in PetscKernel_A_gets_inverse_A_6()
133 ay[3] += stmp * ax[3]; in PetscKernel_A_gets_inverse_A_6()
[all …]
H A Ddgefa5.c18 MatScalar *aa, *ax, *ay, stmp; in PetscKernel_A_gets_inverse_A_5() local
73 ax = &a[k4 + 1]; in PetscKernel_A_gets_inverse_A_5()
84 for (ll = 0; ll < i__3; ll++) ay[ll] += stmp * ax[ll]; in PetscKernel_A_gets_inverse_A_5()
106 ax = aa; in PetscKernel_A_gets_inverse_A_5()
112 for (ll = 0; ll < k; ll++) ay[ll] += stmp * ax[ll]; in PetscKernel_A_gets_inverse_A_5()
128 ax = &a[5 * j + 1]; in PetscKernel_A_gets_inverse_A_5()
130 ay[0] += stmp * ax[0]; in PetscKernel_A_gets_inverse_A_5()
131 ay[1] += stmp * ax[1]; in PetscKernel_A_gets_inverse_A_5()
132 ay[2] += stmp * ax[2]; in PetscKernel_A_gets_inverse_A_5()
133 ay[3] += stmp * ax[3]; in PetscKernel_A_gets_inverse_A_5()
[all …]
H A Ddgefa4.c18 MatScalar *aa, *ax, *ay, work[16], stmp; in PetscKernel_A_gets_inverse_A_4() local
73 ax = &a[k4 + 1]; in PetscKernel_A_gets_inverse_A_4()
84 for (ll = 0; ll < i__3; ll++) ay[ll] += stmp * ax[ll]; in PetscKernel_A_gets_inverse_A_4()
106 ax = aa; in PetscKernel_A_gets_inverse_A_4()
112 for (ll = 0; ll < k; ll++) ay[ll] += stmp * ax[ll]; in PetscKernel_A_gets_inverse_A_4()
128 ax = &a[4 * j + 1]; in PetscKernel_A_gets_inverse_A_4()
130 ay[0] += stmp * ax[0]; in PetscKernel_A_gets_inverse_A_4()
131 ay[1] += stmp * ax[1]; in PetscKernel_A_gets_inverse_A_4()
132 ay[2] += stmp * ax[2]; in PetscKernel_A_gets_inverse_A_4()
133 ay[3] += stmp * ax[3]; in PetscKernel_A_gets_inverse_A_4()
[all …]
H A Ddgefa3.c18 MatScalar *aa, *ax, *ay, work[9], stmp; in PetscKernel_A_gets_inverse_A_3() local
73 ax = &a[k4 + 1]; in PetscKernel_A_gets_inverse_A_3()
84 for (ll = 0; ll < i__3; ll++) ay[ll] += stmp * ax[ll]; in PetscKernel_A_gets_inverse_A_3()
106 ax = aa; in PetscKernel_A_gets_inverse_A_3()
112 for (ll = 0; ll < k; ll++) ay[ll] += stmp * ax[ll]; in PetscKernel_A_gets_inverse_A_3()
128 ax = &a[3 * j + 1]; in PetscKernel_A_gets_inverse_A_3()
130 ay[0] += stmp * ax[0]; in PetscKernel_A_gets_inverse_A_3()
131 ay[1] += stmp * ax[1]; in PetscKernel_A_gets_inverse_A_3()
132 ay[2] += stmp * ax[2]; in PetscKernel_A_gets_inverse_A_3()
136 ax = &a[k3 + 1]; in PetscKernel_A_gets_inverse_A_3()
[all …]
H A Ddgedi.c20 MatScalar *aa, *ax, *ay, tmp; in PetscLINPACKgedi() local
40 ax = aa; in PetscLINPACKgedi()
46 for (ll = 0; ll < k; ll++) ay[ll] += t * ax[ll]; in PetscLINPACKgedi()
65 ax = &a[j * n + 1]; in PetscLINPACKgedi()
67 for (ll = 0; ll < n; ll++) ay[ll] += t * ax[ll]; in PetscLINPACKgedi()
71 ax = &a[kn + 1]; in PetscLINPACKgedi()
74 tmp = ax[ll]; in PetscLINPACKgedi()
75 ax[ll] = ay[ll]; in PetscLINPACKgedi()
H A Ddgefa.c18 MatScalar t, *ax, *ay, *aa; in PetscLINPACKgefa() local
71 ax = aa; in PetscLINPACKgefa()
82 for (ll = 0; ll < i__3; ll++) ay[ll] += t * ax[ll]; in PetscLINPACKgefa()
/petsc/src/mat/tests/
H A Dex229.c7 const PetscScalar *ax; in myF() local
12 PetscCall(VecGetArrayRead(x, &ax)); in myF()
21 xx = PetscPowScalarInt(ax[j], i + 1); in myF()
27 PetscCall(VecRestoreArrayRead(x, &ax)); in myF()
/petsc/src/dm/impls/stag/
H A Dstagmulti.c86 static PetscErrorCode SetInterpolationWeight1d_Private(PetscScalar ax, PetscScalar weight[]) in SetInterpolationWeight1d_Private() argument
89 weight[0] = 1.0 - ax; in SetInterpolationWeight1d_Private()
90 weight[1] = ax; in SetInterpolationWeight1d_Private()
94 static PetscErrorCode SetInterpolationWeight2d_Private(PetscScalar ax, PetscScalar ay, PetscScalar … in SetInterpolationWeight2d_Private() argument
97 weight[0] = (1.0 - ax) * (1.0 - ay); in SetInterpolationWeight2d_Private()
98 weight[1] = ax * (1.0 - ay); in SetInterpolationWeight2d_Private()
99 weight[2] = (1.0 - ax) * ay; in SetInterpolationWeight2d_Private()
100 weight[3] = ax * ay; in SetInterpolationWeight2d_Private()
140 static PetscErrorCode SetInterpolationWeight3d_Private(PetscScalar ax, PetscScalar ay, PetscScalar … in SetInterpolationWeight3d_Private() argument
143 weight[0] = (1.0 - ax) * (1.0 - ay) * (1.0 - az); in SetInterpolationWeight3d_Private()
[all …]
/petsc/include/
H A Dpetscdmstag.h147 …tatic inline PetscErrorCode DMStagGet1dCoordinateArraysDOFRead(DM dm, void *ax, void *ay, void *az) in DMStagGet1dCoordinateArraysDOFRead() argument
149 return DMStagGetProductCoordinateArraysRead(dm, ax, ay, az); in DMStagGet1dCoordinateArraysDOFRead()
159 …c inline PetscErrorCode DMStagRestore1dCoordinateArraysDOFRead(DM dm, void *ax, void *ay, void *az) in DMStagRestore1dCoordinateArraysDOFRead() argument
161 return DMStagRestoreProductCoordinateArraysRead(dm, ax, ay, az); in DMStagRestore1dCoordinateArraysDOFRead()
/petsc/src/mat/impls/h2opus/cuda/
H A Dmath2opusutils.cu103 PetscScalar *ax; in VecSetDelta() local
104 PetscCall(VecCUDAGetArrayWrite(x, &ax)); in VecSetDelta()
105 StandardBasis_Functor delta(ax, i - st); in VecSetDelta()
107 PetscCall(VecCUDARestoreArrayWrite(x, &ax)); in VecSetDelta()
/petsc/src/mat/impls/dense/mpi/
H A Dmpidense.c476 const PetscScalar *ax; in MatMultColumnRange_MPIDense() local
482 PetscCall(VecGetArrayReadAndMemType(xx, &ax, &axmtype)); in MatMultColumnRange_MPIDense()
484 …PetscCall(PetscSFBcastWithMemTypeBegin(mdn->Mvctx, MPIU_SCALAR, axmtype, ax, aymtype, ay, MPI_REPL… in MatMultColumnRange_MPIDense()
485 PetscCall(PetscSFBcastEnd(mdn->Mvctx, MPIU_SCALAR, ax, ay, MPI_REPLACE)); in MatMultColumnRange_MPIDense()
487 PetscCall(VecRestoreArrayReadAndMemType(xx, &ax)); in MatMultColumnRange_MPIDense()
495 const PetscScalar *ax; in MatMult_MPIDense() local
501 PetscCall(VecGetArrayReadAndMemType(xx, &ax, &axmtype)); in MatMult_MPIDense()
503 …PetscCall(PetscSFBcastWithMemTypeBegin(mdn->Mvctx, MPIU_SCALAR, axmtype, ax, aymtype, ay, MPI_REPL… in MatMult_MPIDense()
504 PetscCall(PetscSFBcastEnd(mdn->Mvctx, MPIU_SCALAR, ax, ay, MPI_REPLACE)); in MatMult_MPIDense()
506 PetscCall(VecRestoreArrayReadAndMemType(xx, &ax)); in MatMult_MPIDense()
[all …]
/petsc/src/snes/utils/dm/
H A Ddmadapt.c767 static PetscErrorCode DMAdaptorTransferSolution(DMAdaptor adaptor, DM dm, Vec x, DM adm, Vec ax) in DMAdaptorTransferSolution() argument
775 …if (adaptor->ops->transfersolution) PetscUseTypeMethod(adaptor, transfersolution, dm, x, adm, ax, … in DMAdaptorTransferSolution()
779 PetscCall(DMForestTransferVec(dm, x, adm, ax, PETSC_TRUE, time)); in DMAdaptorTransferSolution()
784 PetscCall(MatInterpolate(interp, x, ax)); in DMAdaptorTransferSolution()
1311 …de DMAdaptorAdapt_Sequence_Private(DMAdaptor adaptor, Vec inx, PetscBool doSolve, DM *adm, Vec *ax) in DMAdaptorAdapt_Sequence_Private() argument
1337 Vec x = adaptIter ? *ax : inx, locX, ox; in DMAdaptorAdapt_Sequence_Private()
1346 PetscCall(SNESSolve(snes, NULL, adaptIter ? *ax : x)); in DMAdaptorAdapt_Sequence_Private()
1348 PetscCall(DMGlobalToLocalBegin(dm, adaptIter ? *ax : x, INSERT_VALUES, locX)); in DMAdaptorAdapt_Sequence_Private()
1349 PetscCall(DMGlobalToLocalEnd(dm, adaptIter ? *ax : x, INSERT_VALUES, locX)); in DMAdaptorAdapt_Sequence_Private()
1350 …PetscCall(VecViewFromOptions(adaptIter ? *ax : x, (PetscObject)adaptor, "-adapt_primal_sol_vec_vie… in DMAdaptorAdapt_Sequence_Private()
[all …]
/petsc/src/ts/impls/mimex/
H A Dmimex.c82 const PetscScalar *ax; in SNESTSFormFunction_Mimex() local
93 PetscCall(VecGetArrayRead(x, &ax)); in SNESTSFormFunction_Mimex()
106 PetscCall(DMPlexPointGlobalFieldRead(dm, p, f, ax, &a)); in SNESTSFormFunction_Mimex()
111 PetscCall(VecRestoreArrayRead(x, &ax)); in SNESTSFormFunction_Mimex()
/petsc/share/petsc/bin/
H A Ddmnetwork_view.py368 cbar = plt.colorbar(matplotlib.cm.ScalarMappable(norm=norm, cmap=colors.colormap), ax=axis)
369 cbar.ax.set_ylim([ylow - epsilon, yhigh + epsilon])
371 cbar = plt.colorbar(matplotlib.cm.ScalarMappable(norm=norm, cmap=colors.colormap), ax=axis)
372 cbar.ax.set_ylim([ylow, yhigh])
/petsc/src/binding/petsc4py/demo/regressor/
H A Dtest_regressor_synthetic.py93 fig,ax = plt.subplots(nrows=2, ncols=3,figsize=(16,7)) variable
/petsc/src/ksp/ksp/utils/dm/
H A Ddmproject.c995 const PetscReal ax = PetscAbsReal(x); in W_3_Interpolation_Private() local
1000 if (ax <= 1.) *w = 1. - 2.5 * PetscSqr(ax) + 1.5 * PetscSqr(ax) * ax; in W_3_Interpolation_Private()
1002 else if (ax <= 2.) *w = 0.5 * PetscSqr(2. - ax) * (1. - ax); in W_3_Interpolation_Private()
/petsc/src/ts/impls/arkimex/
H A Darkimex.c1791 TS_ARKIMEX *ax = (TS_ARKIMEX *)ts->data; in TSARKIMEXGetVecs() local
1796 else *Z = ax->Z; in TSARKIMEXGetVecs()
1800 else *Ydot = ax->Ydot; in TSARKIMEXGetVecs()
1863 TS_ARKIMEX *ax = (TS_ARKIMEX *)ts->data; in TSARKIMEXGetAlgebraicIS() local
1867 else *alg_is = ax->alg_is; in TSARKIMEXGetAlgebraicIS()
/petsc/src/mat/impls/aij/seq/seqhipsparse/
H A Daijhipsparse.hip.cxx3265 const PetscScalar *ax; in MatAXPY_SeqAIJHIPSPARSE() local
3298 PetscCall(MatSeqAIJHIPSPARSEGetArrayRead(X, &ax)); in MatAXPY_SeqAIJHIPSPARSE()
3302 …_bufferSize(cy->handle, Y->rmap->n, Y->cmap->n, &a, cx->mat->descr, x->nz, ax, csrx->row_offsets->… in MatAXPY_SeqAIJHIPSPARSE()
3306 …_csr_spgeam(cy->handle, Y->rmap->n, Y->cmap->n, &a, cx->mat->descr, x->nz, ax, csrx->row_offsets->… in MatAXPY_SeqAIJHIPSPARSE()
3313 …_csr_spgeam(cy->handle, Y->rmap->n, Y->cmap->n, &a, cx->mat->descr, x->nz, ax, csrx->row_offsets->… in MatAXPY_SeqAIJHIPSPARSE()
3319 PetscCall(MatSeqAIJHIPSPARSERestoreArrayRead(X, &ax)); in MatAXPY_SeqAIJHIPSPARSE()
3325 PetscCall(MatSeqAIJHIPSPARSEGetArrayRead(X, &ax)); in MatAXPY_SeqAIJHIPSPARSE()
3330 PetscCallHIPBLAS(hipblasXaxpy(hipblasv2handle, bnz, &a, ax, one, ay, one)); in MatAXPY_SeqAIJHIPSPARSE()
3333 PetscCall(MatSeqAIJHIPSPARSERestoreArrayRead(X, &ax)); in MatAXPY_SeqAIJHIPSPARSE()
/petsc/src/mat/impls/aij/seq/seqcusparse/
H A Daijcusparse.cu3872 const PetscScalar *ax; in MatAXPY_SeqAIJCUSPARSE() local
3906 PetscCall(MatSeqAIJCUSPARSEGetArrayRead(X, &ax)); in MatAXPY_SeqAIJCUSPARSE()
3910 …_bufferSize(cy->handle, Y->rmap->n, Y->cmap->n, &a, cx->mat->descr, x->nz, ax, csrx->row_offsets->… in MatAXPY_SeqAIJCUSPARSE()
3914 …_csr_spgeam(cy->handle, Y->rmap->n, Y->cmap->n, &a, cx->mat->descr, x->nz, ax, csrx->row_offsets->… in MatAXPY_SeqAIJCUSPARSE()
3921 …_csr_spgeam(cy->handle, Y->rmap->n, Y->cmap->n, &a, cx->mat->descr, x->nz, ax, csrx->row_offsets->… in MatAXPY_SeqAIJCUSPARSE()
3927 PetscCall(MatSeqAIJCUSPARSERestoreArrayRead(X, &ax)); in MatAXPY_SeqAIJCUSPARSE()
3933 PetscCall(MatSeqAIJCUSPARSEGetArrayRead(X, &ax)); in MatAXPY_SeqAIJCUSPARSE()
3938 PetscCallCUBLAS(cublasXaxpy(cublasv2handle, bnz, &a, ax, one, ay, one)); in MatAXPY_SeqAIJCUSPARSE()
3941 PetscCall(MatSeqAIJCUSPARSERestoreArrayRead(X, &ax)); in MatAXPY_SeqAIJCUSPARSE()
/petsc/src/dm/impls/plex/
H A Dplexgeometry.c4014 const PetscInt ax = (PetscInt)PetscRealPart(constants[0]); in coordMap_shear() local
4017 for (c = 0; c < Nc; ++c) coords[c] = u[c] + constants[c + 1] * u[ax]; in coordMap_shear()
/petsc/src/ksp/pc/impls/bddc/
H A Dbddcprivate.c2182 const PetscScalar *ax; in PCBDDCBenignMatMult_Private_Private() local
2186 PetscCall(VecGetArrayRead(x, &ax)); in PCBDDCBenignMatMult_Private_Private()
2187 PetscCall(PetscArraycpy(ctx->work, ax, nl)); in PCBDDCBenignMatMult_Private_Private()
2188 PetscCall(VecRestoreArrayRead(x, &ax)); in PCBDDCBenignMatMult_Private_Private()