Home
last modified time | relevance | path

Searched refs:xv (Results 1 – 23 of 23) sorted by relevance

/petsc/src/mat/impls/baij/seq/
H A Dbaijsolvnat14.c14 PetscScalar *x, xv; in MatSolve_SeqBAIJ_14_NaturalOrdering() local
44 xv = x[k + idx]; in MatSolve_SeqBAIJ_14_NaturalOrdering()
45 x[idt] -= v[0] * xv; in MatSolve_SeqBAIJ_14_NaturalOrdering()
46 x[1 + idt] -= v[1] * xv; in MatSolve_SeqBAIJ_14_NaturalOrdering()
47 x[2 + idt] -= v[2] * xv; in MatSolve_SeqBAIJ_14_NaturalOrdering()
48 x[3 + idt] -= v[3] * xv; in MatSolve_SeqBAIJ_14_NaturalOrdering()
49 x[4 + idt] -= v[4] * xv; in MatSolve_SeqBAIJ_14_NaturalOrdering()
50 x[5 + idt] -= v[5] * xv; in MatSolve_SeqBAIJ_14_NaturalOrdering()
51 x[6 + idt] -= v[6] * xv; in MatSolve_SeqBAIJ_14_NaturalOrdering()
52 x[7 + idt] -= v[7] * xv; in MatSolve_SeqBAIJ_14_NaturalOrdering()
[all …]
H A Dbaijsolvnat11.c14 PetscScalar *x, xv; in MatSolve_SeqBAIJ_11_NaturalOrdering() local
41 xv = x[k + idx]; in MatSolve_SeqBAIJ_11_NaturalOrdering()
42 x[idt] -= v[0] * xv; in MatSolve_SeqBAIJ_11_NaturalOrdering()
43 x[1 + idt] -= v[1] * xv; in MatSolve_SeqBAIJ_11_NaturalOrdering()
44 x[2 + idt] -= v[2] * xv; in MatSolve_SeqBAIJ_11_NaturalOrdering()
45 x[3 + idt] -= v[3] * xv; in MatSolve_SeqBAIJ_11_NaturalOrdering()
46 x[4 + idt] -= v[4] * xv; in MatSolve_SeqBAIJ_11_NaturalOrdering()
47 x[5 + idt] -= v[5] * xv; in MatSolve_SeqBAIJ_11_NaturalOrdering()
48 x[6 + idt] -= v[6] * xv; in MatSolve_SeqBAIJ_11_NaturalOrdering()
49 x[7 + idt] -= v[7] * xv; in MatSolve_SeqBAIJ_11_NaturalOrdering()
[all …]
H A Dbaijsolvnat15.c203 PetscScalar *x, xv; in MatSolve_SeqBAIJ_15_NaturalOrdering_ver1() local
234 xv = x[k + idx]; in MatSolve_SeqBAIJ_15_NaturalOrdering_ver1()
235 x[idt] -= v[0] * xv; in MatSolve_SeqBAIJ_15_NaturalOrdering_ver1()
236 x[1 + idt] -= v[1] * xv; in MatSolve_SeqBAIJ_15_NaturalOrdering_ver1()
237 x[2 + idt] -= v[2] * xv; in MatSolve_SeqBAIJ_15_NaturalOrdering_ver1()
238 x[3 + idt] -= v[3] * xv; in MatSolve_SeqBAIJ_15_NaturalOrdering_ver1()
239 x[4 + idt] -= v[4] * xv; in MatSolve_SeqBAIJ_15_NaturalOrdering_ver1()
240 x[5 + idt] -= v[5] * xv; in MatSolve_SeqBAIJ_15_NaturalOrdering_ver1()
241 x[6 + idt] -= v[6] * xv; in MatSolve_SeqBAIJ_15_NaturalOrdering_ver1()
242 x[7 + idt] -= v[7] * xv; in MatSolve_SeqBAIJ_15_NaturalOrdering_ver1()
[all …]
H A Dbaij2.c824 PetscScalar *zarray, xv; in MatMult_SeqBAIJ_11() local
865 xv = xb[k]; in MatMult_SeqBAIJ_11()
866 sum1 += v[0] * xv; in MatMult_SeqBAIJ_11()
867 sum2 += v[1] * xv; in MatMult_SeqBAIJ_11()
868 sum3 += v[2] * xv; in MatMult_SeqBAIJ_11()
869 sum4 += v[3] * xv; in MatMult_SeqBAIJ_11()
870 sum5 += v[4] * xv; in MatMult_SeqBAIJ_11()
871 sum6 += v[5] * xv; in MatMult_SeqBAIJ_11()
872 sum7 += v[6] * xv; in MatMult_SeqBAIJ_11()
873 sum8 += v[7] * xv; in MatMult_SeqBAIJ_11()
[all …]
/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckok.kokkos.cxx121 PetscScalarKokkosView xv; in VecAbs_SeqKokkos() local
126 PetscCall(VecGetKokkosView(xin, &xv)); in VecAbs_SeqKokkos()
127 PetscCallCXX(KokkosBlas::abs(exec, xv, xv)); in VecAbs_SeqKokkos()
128 PetscCall(VecRestoreKokkosView(xin, &xv)); in VecAbs_SeqKokkos()
136 PetscScalarKokkosView xv; in VecReciprocal_SeqKokkos() local
140 PetscCall(VecGetKokkosView(xin, &xv)); in VecReciprocal_SeqKokkos()
143 if (xv(i) != (PetscScalar)0.0) xv(i) = (PetscScalar)1.0 / xv(i); in VecReciprocal_SeqKokkos()
145 PetscCall(VecRestoreKokkosView(xin, &xv)); in VecReciprocal_SeqKokkos()
152 ConstPetscScalarKokkosView xv; in VecMin_SeqKokkos() local
157 PetscCall(VecGetKokkosView(xin, &xv)); in VecMin_SeqKokkos()
[all …]
/petsc/src/snes/tutorials/
H A Dex55k.kokkos.cxx44 ConstPetscScalarKokkosOffsetView2D xv; in FormFunctionLocalVec() local
57 PetscCall(DMDAVecGetKokkosOffsetView(info->da, x, &xv)); in FormFunctionLocalVec()
69 fv(j, i) = 2.0 * (hydhx + hxdhy) * (xv(j, i) - mms_solution); in FormFunctionLocalVec()
71 u = xv(j, i); in FormFunctionLocalVec()
72 uw = xv(j, i - 1); in FormFunctionLocalVec()
73 ue = xv(j, i + 1); in FormFunctionLocalVec()
74 un = xv(j - 1, i); in FormFunctionLocalVec()
75 us = xv(j + 1, i); in FormFunctionLocalVec()
109 PetscCall(DMDAVecRestoreKokkosOffsetView(info->da, x, &xv)); in FormFunctionLocalVec()
122 ConstPetscScalarKokkosOffsetView2D xv; in FormObjectiveLocalVec() local
[all …]
/petsc/src/mat/impls/sbaij/seq/
H A Drelax.h9 #define PetscSparseDensePlusDot_no_function(sum, r, xv, xi, nnz) \ argument
15 sum += *xv++ * r[*xi++]; \
17 sum += *xv++ * r[*xi++]; \
19 sum += *xv++ * r[*xi++]; \
23 sum += xv[0] * r[xi[0]] + xv[1] * r[xi[1]] + xv[2] * r[xi[2]] + xv[3] * r[xi[3]]; \
24 xv += 4; \
28 xv -= nnz; \
34 #define PetscSparseDensePlusDot_no_function(sum, r, xv, xi, nnz) \ argument
40 sum += (xv[__i] * r[__i1] + xv[__i + 1] * r[__i2]); \
42 if (nnz & 0x1) sum += xv[__i] * r[xi[__i]]; \
[all …]
/petsc/src/mat/impls/aij/seq/
H A Daij.h505 #define PetscSparseDenseMinusDot(sum, r, xv, xi, nnz) \ argument
511 sum -= *xv++ * r[*xi++]; \
513 sum -= *xv++ * r[*xi++]; \
515 sum -= *xv++ * r[*xi++]; \
519 sum -= xv[0] * r[xi[0]] + xv[1] * r[xi[1]] + xv[2] * r[xi[2]] + xv[3] * r[xi[3]]; \
520 xv += 4; \
524 xv -= nnz; \
530 #define PetscSparseDenseMinusDot(sum, r, xv, xi, nnz) \ argument
536 sum -= (xv[__i] * r[__i1] + xv[__i + 1] * r[__i2]); \
538 if (nnz & 0x1) sum -= xv[__i] * r[xi[__i]]; \
[all …]
/petsc/src/dm/impls/network/
H A Dnetworkcreate.c23 static PetscErrorCode VecArrayPrint_private(PetscViewer viewer, PetscInt n, const PetscScalar *xv) in VecArrayPrint_private() argument
30 if (PetscImaginaryPart(xv[i]) > 0.0) { in VecArrayPrint_private()
31 …IIPrintf(viewer, " %g + %g i\n", (double)PetscRealPart(xv[i]), (double)PetscImaginaryPart(xv[i]… in VecArrayPrint_private()
32 } else if (PetscImaginaryPart(xv[i]) < 0.0) { in VecArrayPrint_private()
33 …IPrintf(viewer, " %g - %g i\n", (double)PetscRealPart(xv[i]), -(double)PetscImaginaryPart(xv[i]… in VecArrayPrint_private()
34 } else PetscCall(PetscViewerASCIIPrintf(viewer, " %g\n", (double)PetscRealPart(xv[i]))); in VecArrayPrint_private()
36 PetscCall(PetscViewerASCIIPrintf(viewer, " %g\n", (double)xv[i])); in VecArrayPrint_private()
45 const PetscScalar *xv; in VecView_Network_Seq() local
48 PetscCall(VecGetArrayRead(X, &xv)); in VecView_Network_Seq()
60 PetscCall(VecArrayPrint_private(viewer, nvar, xv + offset)); in VecView_Network_Seq()
[all …]
/petsc/src/snes/interface/
H A Dsnesob.c145 PetscScalar fv, xv; in SNESObjectiveComputeFunctionDefaultFD() local
167 xv = dx; in SNESObjectiveComputeFunctionDefaultFD()
168 PetscCall(VecSetValues(Xh, 1, &i, &xv, ADD_VALUES)); in SNESObjectiveComputeFunctionDefaultFD()
177 xv = 2. * dx; in SNESObjectiveComputeFunctionDefaultFD()
178 PetscCall(VecSetValues(Xh, 1, &i, &xv, ADD_VALUES)); in SNESObjectiveComputeFunctionDefaultFD()
187 xv = -dx; in SNESObjectiveComputeFunctionDefaultFD()
188 PetscCall(VecSetValues(Xh, 1, &i, &xv, ADD_VALUES)); in SNESObjectiveComputeFunctionDefaultFD()
/petsc/src/vec/vec/impls/seq/
H A Dbvec2.c242 const PetscScalar *xv; in VecView_Seq_ASCII() local
245 PetscCall(VecGetArrayRead(xin, &xv)); in VecView_Seq_ASCII()
252 if (PetscImaginaryPart(xv[i]) > 0.0) { in VecView_Seq_ASCII()
253 …ntf(viewer, "%18.16e + %18.16ei\n", (double)PetscRealPart(xv[i]), (double)PetscImaginaryPart(xv[i]… in VecView_Seq_ASCII()
254 } else if (PetscImaginaryPart(xv[i]) < 0.0) { in VecView_Seq_ASCII()
255 …tf(viewer, "%18.16e - %18.16ei\n", (double)PetscRealPart(xv[i]), -(double)PetscImaginaryPart(xv[i]… in VecView_Seq_ASCII()
257 PetscCall(PetscViewerASCIIPrintf(viewer, "%18.16e\n", (double)PetscRealPart(xv[i]))); in VecView_Seq_ASCII()
260 PetscCall(PetscViewerASCIIPrintf(viewer, "%18.16e\n", (double)xv[i])); in VecView_Seq_ASCII()
267 …Printf(viewer, "%18.16e %18.16e\n", (double)PetscRealPart(xv[i]), (double)PetscImaginaryPart(xv[i]… in VecView_Seq_ASCII()
269 PetscCall(PetscViewerASCIIPrintf(viewer, "%18.16e\n", (double)xv[i])); in VecView_Seq_ASCII()
[all …]
/petsc/src/ts/utils/
H A Ddmnetworkts.c116 const PetscScalar *xv; in TSMonitorLGCtxNetworkSolution()
146 xv = yv; in TSMonitorLGCtxNetworkSolution()
148 PetscCall(VecGetArrayRead(u, &xv)); in TSMonitorLGCtxNetworkSolution()
159 PetscCall(PetscDrawLGAddCommonPoint(ctx->lg[i], ptime, xv + offset)); in TSMonitorLGCtxNetworkSolution()
170 PetscCall(PetscDrawLGAddCommonPoint(ctx->lg[i], ptime, xv + offset)); in TSMonitorLGCtxNetworkSolution()
177 PetscCall(VecRestoreArrayRead(u, &xv)); in TSMonitorLGCtxNetworkSolution()
/petsc/src/vec/is/sf/tests/
H A Dex17.c18 const PetscScalar *xv; in main() local
78 PetscCall(VecGetArrayRead(x, &xv)); in main()
79 PetscCheck(xv[0] == 6.0, PETSC_COMM_SELF, PETSC_ERR_PLIB, "VecScatter: wrong results"); in main()
80 PetscCall(VecRestoreArrayRead(x, &xv)); in main()
/petsc/src/ksp/pc/impls/pbjacobi/kokkos/
H A Dpbjacobi_kok.kokkos.cxx35 ConstPetscScalarKokkosView xv; in PCApplyOrTranspose_PBJacobi_Kokkos() local
45 PetscCall(VecGetKokkosView(x, &xv)); in PCApplyOrTranspose_PBJacobi_Kokkos()
56 xp = &xv(bs * k); in PCApplyOrTranspose_PBJacobi_Kokkos()
65 PetscCall(VecRestoreKokkosView(x, &xv)); in PCApplyOrTranspose_PBJacobi_Kokkos()
/petsc/src/ksp/pc/impls/vpbjacobi/kokkos/
H A Dvpbjacobi_kok.kokkos.cxx75 ConstPetscScalarKokkosView xv; in PCApplyOrTranspose_VPBJacobi_Kokkos() local
87 PetscCall(VecGetKokkosView(x, &xv)); in PCApplyOrTranspose_VPBJacobi_Kokkos()
95 const PetscScalar *xbuf = &xv(bs(bid)); in PCApplyOrTranspose_VPBJacobi_Kokkos()
117 xp = &xv(bs(k)); in PCApplyOrTranspose_VPBJacobi_Kokkos()
127 PetscCall(VecRestoreKokkosView(x, &xv)); in PCApplyOrTranspose_VPBJacobi_Kokkos()
/petsc/src/vec/vec/impls/mpi/kokkos/
H A Dmpikok.kokkos.cxx162 PetscScalarKokkosView xv; in VecSetValuesCOO_MPIKokkos() local
180 if (imode == INSERT_VALUES) PetscCall(VecGetKokkosViewWrite(x, &xv)); /* write vector */ in VecSetValuesCOO_MPIKokkos()
181 else PetscCall(VecGetKokkosView(x, &xv)); /* read & write vector */ in VecSetValuesCOO_MPIKokkos()
187 xv(i) = (imode == INSERT_VALUES ? 0.0 : xv(i)) + sum; in VecSetValuesCOO_MPIKokkos()
195 for (PetscCount k = jmap2(i); k < jmap2(i + 1); k++) xv(imap2(i)) += recvbuf(perm2(k)); in VecSetValuesCOO_MPIKokkos()
198 if (imode == INSERT_VALUES) PetscCall(VecRestoreKokkosViewWrite(x, &xv)); in VecSetValuesCOO_MPIKokkos()
199 else PetscCall(VecRestoreKokkosView(x, &xv)); in VecSetValuesCOO_MPIKokkos()
/petsc/src/vec/vec/impls/mpi/cupm/
H A Dvecmpicupm_impl.hpp282 …ount *PETSC_RESTRICT jmap2, const PetscCount *PETSC_RESTRICT perm2, PetscScalar *PETSC_RESTRICT xv) in add_remote_coo_values() argument
284 …add_coo_values_impl(vv, nnz2, jmap2, perm2, ADD_VALUES, xv, [=](PetscCount i) { return imap2[i]; }… in add_remote_coo_values()
333 …const auto xv = imode == INSERT_VALUES ? DeviceArrayWrite(dctx, x).data() : DeviceArrayRead… in SetValuesCOO() local
354 …unchKernel1D(n, 0, stream, kernels::add_coo_values, vv, n, vcu->jmap1_d, vcu->perm1_d, imode, xv)); in SetValuesCOO()
359 …m, kernels::add_remote_coo_values, recvbuf_d, nnz2, vcu->imap2_d, vcu->jmap2_d, vcu->perm2_d, xv)); in SetValuesCOO()
/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.kokkos.cxx389 ConstPetscScalarKokkosView xv; in MatMult_SeqAIJKokkos() local
395 PetscCall(VecGetKokkosView(xx, &xv)); in MatMult_SeqAIJKokkos()
398 …::spmv(PetscGetKokkosExecutionSpace(), "N", 1.0 /*alpha*/, aijkok->csrmat, xv, 0.0 /*beta*/, yv));… in MatMult_SeqAIJKokkos()
399 PetscCall(VecRestoreKokkosView(xx, &xv)); in MatMult_SeqAIJKokkos()
412 ConstPetscScalarKokkosView xv; in MatMultTranspose_SeqAIJKokkos() local
419 PetscCall(VecGetKokkosView(xx, &xv)); in MatMultTranspose_SeqAIJKokkos()
429 …PetscCallCXX(KokkosSparse::spmv(PetscGetKokkosExecutionSpace(), mode, 1.0 /*alpha*/, csrmat, xv, 0… in MatMultTranspose_SeqAIJKokkos()
430 PetscCall(VecRestoreKokkosView(xx, &xv)); in MatMultTranspose_SeqAIJKokkos()
442 ConstPetscScalarKokkosView xv; in MatMultHermitianTranspose_SeqAIJKokkos() local
449 PetscCall(VecGetKokkosView(xx, &xv)); in MatMultHermitianTranspose_SeqAIJKokkos()
[all …]
/petsc/src/vec/vec/impls/seq/cupm/
H A Dvecseqcupm.hpp143 …nst PetscCount *PETSC_RESTRICT perm, InsertMode imode, PetscScalar *PETSC_RESTRICT xv, F &&xvindex) in add_coo_values_impl() argument
153 xv[idx] = sum; in add_coo_values_impl()
155 xv[idx] += sum; in add_coo_values_impl()
164 …CT jmap1, const PetscCount *PETSC_RESTRICT perm1, InsertMode imode, PetscScalar *PETSC_RESTRICT xv) in add_coo_values() argument
166 add_coo_values_impl(v, n, jmap1, perm1, imode, xv, [](PetscCount i) { return i; }); in add_coo_values()
H A Dvecseqcupm_impl.hpp1662 PetscScalar xv = 0.; in Norm() local
1667 …PetscCall(PetscCUPMMemcpyAsync(&xv, xptr.data() + max_loc - 1, 1, cupmMemcpyDeviceToHost, stream)); in Norm()
1668 *z = PetscAbsScalar(xv); in Norm()
/petsc/include/
H A Dpetscvec.h602 static inline PetscErrorCode VecGetArrayPair(Vec x, Vec y, PetscScalar *xv[], PetscScalar *yv[]) in VecGetArrayPair() argument
606 if (x == y) *xv = *yv; in VecGetArrayPair()
607 else PetscCall(VecGetArrayRead(x, (const PetscScalar **)xv)); in VecGetArrayPair()
626 static inline PetscErrorCode VecRestoreArrayPair(Vec x, Vec y, PetscScalar *xv[], PetscScalar *yv[]) in VecRestoreArrayPair() argument
630 if (x != y) PetscCall(VecRestoreArrayRead(x, (const PetscScalar **)xv)); in VecRestoreArrayPair()
/petsc/src/mat/impls/dense/seq/
H A Ddense.c240 const PetscScalar *xv; in MatAXPY_SeqDense() local
245 PetscCall(MatDenseGetArrayRead(X, &xv)); in MatAXPY_SeqDense()
252 …PetscCallBLAS("BLASaxpy", BLASaxpy_(&m, &alpha, PetscSafePointerPlusOffset(xv, j * ldax), &one, Pe… in MatAXPY_SeqDense()
254 PetscCallBLAS("BLASaxpy", BLASaxpy_(&N, &alpha, xv, &one, yv, &one)); in MatAXPY_SeqDense()
256 PetscCall(MatDenseRestoreArrayRead(X, &xv)); in MatAXPY_SeqDense()
704 PetscScalar *xv; in MatMatSolve_SeqDense_TearDown() local
705 PetscCall(MatDenseGetArray(X, &xv)); in MatMatSolve_SeqDense_TearDown()
706 for (PetscInt j = 0; j < nrhs; j++) PetscCall(PetscArraycpy(&xv[j * ldx], &y[j * ldy], k)); in MatMatSolve_SeqDense_TearDown()
707 PetscCall(MatDenseRestoreArray(X, &xv)); in MatMatSolve_SeqDense_TearDown()
/petsc/src/ksp/pc/impls/hypre/
H A Dhypre.c275 HYPRE_ParVector bv, xv; in PCSetUp_HYPRE() local
555 PetscCallHYPRE(HYPRE_IJVectorGetObject(hjac->x->ij, (void **)&xv)); in PCSetUp_HYPRE()
557 PetscCallHYPRE((*jac->setup)(jac->hsolver, hmat, bv, xv)); in PCSetUp_HYPRE()