Searched refs:VecRestoreKokkosView (Results 1 – 7 of 7) sorted by relevance
| /petsc/src/vec/vec/impls/seq/kokkos/ |
| H A D | veckok.kokkos.cxx | 67 PETSC_VISIBILITY_PUBLIC PetscErrorCode VecRestoreKokkosView(Vec v, PetscScalarKokkosView *kv) in VecRestoreKokkosView() function
90 PETSC_VISIBILITY_PUBLIC PetscErrorCode VecRestoreKokkosView(Vec v, PetscScalarKokkosViewHost *kv) in VecRestoreKokkosView() function
128 PetscCall(VecRestoreKokkosView(xin, &xv)); in VecAbs_SeqKokkos()
145 PetscCall(VecRestoreKokkosView(xin, &xv)); in VecReciprocal_SeqKokkos()
169 PetscCall(VecRestoreKokkosView(xin, &xv)); in VecMin_SeqKokkos()
193 PetscCall(VecRestoreKokkosView(xin, &xv)); in VecMax_SeqKokkos()
206 PetscCall(VecRestoreKokkosView(xin, &xv)); in VecSum_SeqKokkos()
218 … KOKKOS_LAMBDA(const PetscInt &i) { xv(i) += shift; }); PetscCall(VecRestoreKokkosView(xin, &xv))); in VecShift_SeqKokkos()
244 PetscCall(VecRestoreKokkosView(xin, &xv)); in VecAXPY_SeqKokkos()
245 PetscCall(VecRestoreKokkosView(yin, &yv)); in VecAXPY_SeqKokkos()
[all …]
|
| /petsc/include/ |
| H A D | petscvec_kokkos.hpp | 81 PetscErrorCode VecRestoreKokkosView(Vec, Kokkos::View<const PetscScalar *, MemorySpace> *) in VecRestoreKokkosView() function
86 PetscErrorCode VecRestoreKokkosView(Vec, Kokkos::View<PetscScalar *, MemorySpace> *);
|
| /petsc/src/dm/impls/da/kokkos/ |
| H A D | dagetov.kokkos.cxx | 56 else PetscCall(VecRestoreKokkosView(vec, &kv)); in DMDAVecRestoreKokkosOffsetView_Private()
87 PetscCall(VecRestoreKokkosView(vec, &kv)); in DMDAVecRestoreKokkosOffsetView()
122 else PetscCall(VecRestoreKokkosView(vec, &kv)); in DMDAVecRestoreKokkosOffsetView_Private()
153 PetscCall(VecRestoreKokkosView(vec, &kv)); in DMDAVecRestoreKokkosOffsetView()
187 else PetscCall(VecRestoreKokkosView(vec, &kv)); in DMDAVecRestoreKokkosOffsetView_Private()
218 PetscCall(VecRestoreKokkosView(vec, &kv)); in DMDAVecRestoreKokkosOffsetView()
393 else PetscCall(VecRestoreKokkosView(vec, &kv)); in DMDAVecRestoreKokkosOffsetViewDOF_Private()
424 PetscCall(VecRestoreKokkosView(vec, &kv)); in DMDAVecRestoreKokkosOffsetViewDOF()
458 else PetscCall(VecRestoreKokkosView(vec, &kv)); in DMDAVecRestoreKokkosOffsetViewDOF_Private()
489 PetscCall(VecRestoreKokkosView(vec, &kv)); in DMDAVecRestoreKokkosOffsetViewDOF()
[all …]
|
| /petsc/src/ksp/pc/impls/pbjacobi/kokkos/ |
| H A D | pbjacobi_kok.kokkos.cxx | 65 PetscCall(VecRestoreKokkosView(x, &xv)); in PCApplyOrTranspose_PBJacobi_Kokkos()
|
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.kokkos.cxx | 399 PetscCall(VecRestoreKokkosView(xx, &xv)); in MatMult_SeqAIJKokkos()
430 PetscCall(VecRestoreKokkosView(xx, &xv)); in MatMultTranspose_SeqAIJKokkos()
460 PetscCall(VecRestoreKokkosView(xx, &xv)); in MatMultHermitianTranspose_SeqAIJKokkos()
482 PetscCall(VecRestoreKokkosView(xx, &xv)); in MatMultAdd_SeqAIJKokkos()
483 PetscCall(VecRestoreKokkosView(zz, &zv)); in MatMultAdd_SeqAIJKokkos()
513 PetscCall(VecRestoreKokkosView(xx, &xv)); in MatMultTransposeAdd_SeqAIJKokkos()
514 PetscCall(VecRestoreKokkosView(zz, &zv)); in MatMultTransposeAdd_SeqAIJKokkos()
544 PetscCall(VecRestoreKokkosView(xx, &xv)); in MatMultHermitianTransposeAdd_SeqAIJKokkos()
545 PetscCall(VecRestoreKokkosView(zz, &zv)); in MatMultHermitianTransposeAdd_SeqAIJKokkos()
1082 PetscCall(VecRestoreKokkosView(D, &dv)); in MatDiagonalSet_SeqAIJKokkos()
[all …]
|
| /petsc/src/ksp/pc/impls/vpbjacobi/kokkos/ |
| H A D | vpbjacobi_kok.kokkos.cxx | 127 PetscCall(VecRestoreKokkosView(x, &xv)); in PCApplyOrTranspose_VPBJacobi_Kokkos()
|
| /petsc/src/vec/vec/impls/mpi/kokkos/ |
| H A D | mpikok.kokkos.cxx | 199 else PetscCall(VecRestoreKokkosView(x, &xv)); in VecSetValuesCOO_MPIKokkos()
|