Searched refs:zv (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/vec/vec/impls/seq/kokkos/ |
| H A D | veckok.kokkos.cxx | 347 PetscScalarKokkosViewHost zv(z, nv); in VecMultiDot_Private() local
356 …lel_reduce(Kokkos::RangePolicy<WorkTag>(exec, 0, N), mdot, Kokkos::subview(zv, Kokkos::pair<PetscI… in VecMultiDot_Private()
365 …auto results = Kokkos::subview(zv, Kokkos::pair<PetscInt, PetscInt>(cur, c… in VecMultiDot_Private()
577 …auto zv = PetscScalarKokkosDualView(PetscScalarKokkosView(z_d + i, m), PetscScalarKokkosVie… in VecMultiDot_SeqKokkos_GEMV() local
579 …PetscCallCXX(KokkosBlas::gemv(PetscGetKokkosExecutionSpace(), trans, 1.0, Y, xv, 0.0, zv.view_devi… in VecMultiDot_SeqKokkos_GEMV()
581 PetscCallCXX(zv.modify_device()); in VecMultiDot_SeqKokkos_GEMV()
582 PetscCall(KokkosDualViewSyncHost(zv, PetscGetKokkosExecutionSpace())); in VecMultiDot_SeqKokkos_GEMV()
938 PetscScalarKokkosView zv; in VecAXPBYPCZ_SeqKokkos() local
943 PetscCall(VecGetKokkosView(zin, &zv)); in VecAXPBYPCZ_SeqKokkos()
949 policy, KOKKOS_LAMBDA(const PetscInt &i) { zv(i) = alpha * (xv(i) - yv(i)); })); in VecAXPBYPCZ_SeqKokkos()
[all …]
|
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.kokkos.cxx | 472 PetscScalarKokkosView zv; in MatMultAdd_SeqAIJKokkos() local
479 PetscCall(VecGetKokkosView(zz, &zv)); // do after VecCopy(yy, zz) to get the latest data on device in MatMultAdd_SeqAIJKokkos()
481 …kosExecutionSpace(), "N", 1.0 /*alpha*/, aijkok->csrmat, xv, 1.0 /*beta*/, zv)); /* z = alpha A x … in MatMultAdd_SeqAIJKokkos()
483 PetscCall(VecRestoreKokkosView(zz, &zv)); in MatMultAdd_SeqAIJKokkos()
495 PetscScalarKokkosView zv; in MatMultTransposeAdd_SeqAIJKokkos() local
503 PetscCall(VecGetKokkosView(zz, &zv)); in MatMultTransposeAdd_SeqAIJKokkos()
512 …cGetKokkosExecutionSpace(), mode, 1.0 /*alpha*/, csrmat, xv, 1.0 /*beta*/, zv)); /* z = alpha A^T … in MatMultTransposeAdd_SeqAIJKokkos()
514 PetscCall(VecRestoreKokkosView(zz, &zv)); in MatMultTransposeAdd_SeqAIJKokkos()
526 PetscScalarKokkosView zv; in MatMultHermitianTransposeAdd_SeqAIJKokkos() local
534 PetscCall(VecGetKokkosView(zz, &zv)); in MatMultHermitianTransposeAdd_SeqAIJKokkos()
[all …]
|