| /petsc/src/vec/vec/impls/seq/kokkos/ |
| H A D | veckok.kokkos.cxx | 237 PetscScalarKokkosView yv; in VecAXPY_SeqKokkos() local
242 PetscCall(VecGetKokkosView(yin, &yv)); in VecAXPY_SeqKokkos()
243 PetscCallCXX(KokkosBlas::axpy(PetscGetKokkosExecutionSpace(), alpha, xv, yv)); in VecAXPY_SeqKokkos()
245 PetscCall(VecRestoreKokkosView(yin, &yv)); in VecAXPY_SeqKokkos()
267 ConstPetscScalarKokkosView xv, yv; in VecTDot_SeqKokkos() local
272 PetscCall(VecGetKokkosView(yin, &yv)); in VecTDot_SeqKokkos()
274 …>map->n), KOKKOS_LAMBDA(const PetscInt &i, PetscScalar &update) { update += yv(i) * xv(i); }, *z)); in VecTDot_SeqKokkos()
275 PetscCall(VecRestoreKokkosView(yin, &yv)); in VecTDot_SeqKokkos()
295 ConstPetscScalarKokkosView xv, yv[8]; member
300 yv[0] = yv0; in MDotFunctor()
[all …]
|
| /petsc/src/ts/utils/ |
| H A D | dmnetworkts.c | 117 PetscScalar *yv; in TSMonitorLGCtxNetworkSolution()
140 PetscCall(VecGetArray(uv, &yv)); in TSMonitorLGCtxNetworkSolution()
143 if (PetscRealPart(yv[j]) <= 0.0) yv[j] = -12; in TSMonitorLGCtxNetworkSolution()
144 else yv[j] = PetscLog10Real(PetscRealPart(yv[j])); in TSMonitorLGCtxNetworkSolution()
146 xv = yv; in TSMonitorLGCtxNetworkSolution()
174 PetscCall(VecRestoreArray(uv, &yv)); in TSMonitorLGCtxNetworkSolution()
|
| /petsc/src/vec/is/sf/tests/ |
| H A D | ex7.c | 11 const PetscScalar *yv; in main() local
53 PetscCall(VecGetArrayRead(y, &yv)); in main()
54 PetscCall(VecCreateMPIWithArray(PETSC_COMM_WORLD, 1, n, PETSC_DECIDE, yv, &z)); in main()
56 PetscCall(VecRestoreArrayRead(y, &yv)); in main()
|
| /petsc/src/ksp/pc/impls/pbjacobi/kokkos/ |
| H A D | pbjacobi_kok.kokkos.cxx | 36 PetscScalarKokkosView yv; in PCApplyOrTranspose_PBJacobi_Kokkos() local
46 PetscCall(VecGetKokkosViewWrite(y, &yv)); in PCApplyOrTranspose_PBJacobi_Kokkos()
57 yp = &yv(bs * k); in PCApplyOrTranspose_PBJacobi_Kokkos()
66 PetscCall(VecRestoreKokkosViewWrite(y, &yv)); in PCApplyOrTranspose_PBJacobi_Kokkos()
|
| /petsc/src/ksp/pc/impls/vpbjacobi/kokkos/ |
| H A D | vpbjacobi_kok.kokkos.cxx | 76 PetscScalarKokkosView yv; in PCApplyOrTranspose_VPBJacobi_Kokkos() local
88 PetscCall(VecGetKokkosViewWrite(y, &yv)); in PCApplyOrTranspose_VPBJacobi_Kokkos()
96 PetscScalar *ybuf = &yv(bs(bid)); in PCApplyOrTranspose_VPBJacobi_Kokkos()
118 yp = &yv(bs(k)); in PCApplyOrTranspose_VPBJacobi_Kokkos()
128 PetscCall(VecRestoreKokkosViewWrite(y, &yv)); in PCApplyOrTranspose_VPBJacobi_Kokkos()
|
| /petsc/src/snes/tutorials/ |
| H A D | ex19.c | 386 PetscScalar yu = 0, yv = 0, yo = 0, yt = 0; in NonlinearGS() local
577 yv = fv / dfvdv; in NonlinearGS()
578 yo = (fomega - (dfodu * yu + dfodv * yv)) / dfodo; in NonlinearGS()
579 yt = (ftemp - (dftdu * yu + dftdv * yv)) / dftdt; in NonlinearGS()
582 x[j][i].v = x[j][i].v - yv; in NonlinearGS()
621 pynorm = PetscRealPart(yu * yu + yv * yv + yo * yo + yt * yt); in NonlinearGS()
|
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.kokkos.cxx | 390 PetscScalarKokkosView yv; in MatMult_SeqAIJKokkos() local
396 PetscCall(VecGetKokkosViewWrite(yy, &yv)); in MatMult_SeqAIJKokkos()
398 …kosExecutionSpace(), "N", 1.0 /*alpha*/, aijkok->csrmat, xv, 0.0 /*beta*/, yv)); /* y = alpha A x … in MatMult_SeqAIJKokkos()
400 PetscCall(VecRestoreKokkosViewWrite(yy, &yv)); in MatMult_SeqAIJKokkos()
413 PetscScalarKokkosView yv; in MatMultTranspose_SeqAIJKokkos() local
420 PetscCall(VecGetKokkosViewWrite(yy, &yv)); in MatMultTranspose_SeqAIJKokkos()
429 …cGetKokkosExecutionSpace(), mode, 1.0 /*alpha*/, csrmat, xv, 0.0 /*beta*/, yv)); /* y = alpha A^T … in MatMultTranspose_SeqAIJKokkos()
431 PetscCall(VecRestoreKokkosViewWrite(yy, &yv)); in MatMultTranspose_SeqAIJKokkos()
443 PetscScalarKokkosView yv; in MatMultHermitianTranspose_SeqAIJKokkos() local
450 PetscCall(VecGetKokkosViewWrite(yy, &yv)); in MatMultHermitianTranspose_SeqAIJKokkos()
[all …]
|
| /petsc/include/ |
| H A D | petscvec.h | 602 static inline PetscErrorCode VecGetArrayPair(Vec x, Vec y, PetscScalar *xv[], PetscScalar *yv[]) in VecGetArrayPair() argument
605 PetscCall(VecGetArray(y, yv)); in VecGetArrayPair()
606 if (x == y) *xv = *yv; in VecGetArrayPair()
626 static inline PetscErrorCode VecRestoreArrayPair(Vec x, Vec y, PetscScalar *xv[], PetscScalar *yv[]) in VecRestoreArrayPair() argument
629 PetscCall(VecRestoreArray(y, yv)); in VecRestoreArrayPair()
|
| /petsc/src/vec/vec/impls/seq/ |
| H A D | bvec2.c | 520 PetscScalar yv = y ? y[i] : 0.0; in VecSetValues_Seq() local
526 xx[ix[i]] = yv; in VecSetValues_Seq()
528 xx[ix[i]] += yv; in VecSetValues_Seq()
|
| /petsc/src/vec/vec/impls/mpi/ |
| H A D | pdvec.c | 746 PetscScalar yv = y ? y[i] : 0; in VecSetValues_MPI() local
752 xx[row - start] = yv; in VecSetValues_MPI()
754 xx[row - start] += yv; in VecSetValues_MPI()
758 PetscCall(VecStashValue_Private(&xin->stash, row, yv)); in VecSetValues_MPI()
|
| /petsc/src/mat/impls/dense/seq/cupm/ |
| H A D | matseqdensecupm.hpp | 696 PetscScalar *yv; in MatSolve_Factored_Dispatch_() local
702 PetscCall(VecCUPMGetArrayWriteAsync<T>(y, &yv, dctx)); in MatSolve_Factored_Dispatch_()
704 PetscCall(VecGetArray(y, &yv)); in MatSolve_Factored_Dispatch_()
706 PetscCall(PetscCUPMMemcpyAsync(yv, y_array, k, copy_mode, stream)); in MatSolve_Factored_Dispatch_()
708 PetscCall(VecCUPMRestoreArrayWriteAsync<T>(y, &yv, dctx)); in MatSolve_Factored_Dispatch_()
710 PetscCall(VecRestoreArray(y, &yv)); in MatSolve_Factored_Dispatch_()
|
| /petsc/src/mat/impls/dense/seq/ |
| H A D | dense.c | 241 PetscScalar *yv; in MatAXPY_SeqDense() local
246 PetscCall(MatDenseGetArray(Y, &yv)); in MatAXPY_SeqDense()
252 …, PetscSafePointerPlusOffset(xv, j * ldax), &one, PetscSafePointerPlusOffset(yv, j * lday), &one)); in MatAXPY_SeqDense()
254 PetscCallBLAS("BLASaxpy", BLASaxpy_(&N, &alpha, xv, &one, yv, &one)); in MatAXPY_SeqDense()
257 PetscCall(MatDenseRestoreArray(Y, &yv)); in MatAXPY_SeqDense()
555 PetscScalar *yv; in MatSolve_SeqDense_TearDown() local
556 PetscCall(VecGetArray(yy, &yv)); in MatSolve_SeqDense_TearDown()
557 PetscCall(PetscArraycpy(yv, y, k)); in MatSolve_SeqDense_TearDown()
558 PetscCall(VecRestoreArray(yy, &yv)); in MatSolve_SeqDense_TearDown()
|