Home
last modified time | relevance | path

Searched refs:PetscScalarKokkosViewType (Results 1 – 3 of 3) sorted by relevance

/petsc/src/dm/impls/da/kokkos/
H A Ddagetov.kokkos.cxx30 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetView_Private()
48 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetView_Private()
96 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetView_Private()
113 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetView_Private()
120 kv = PetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1)); in DMDAVecRestoreKokkosOffsetView_Private()
162 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetView_Private()
179 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetView_Private()
185 …kv = PetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1) * ov->extent… in DMDAVecRestoreKokkosOffsetView_Private()
368 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetViewDOF_Private()
385 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetViewDOF_Private()
[all …]
/petsc/include/petsc/private/
H A Dkokkosimpl.hpp25 using PetscScalarKokkosViewType = Kokkos::View<PetscScalar *, MemorySpace>; typedef
72 using PetscScalarKokkosView = PetscScalarKokkosViewType<DefaultMemorySpace>;
78 using PetscScalarKokkosViewHost = PetscScalarKokkosViewType<HostMirrorMemorySpace>;
/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckok.kokkos.cxx18 static PetscErrorCode VecGetKokkosView_Private(Vec v, PetscScalarKokkosViewType<MemorySpace> *kv, P… in VecGetKokkosView_Private()
21 using ExecutionSpace = typename PetscScalarKokkosViewType<MemorySpace>::traits::device_type; in VecGetKokkosView_Private()
33 static PetscErrorCode VecRestoreKokkosView_Private(Vec v, PetscScalarKokkosViewType<MemorySpace> *k… in VecRestoreKokkosView_Private()
36 using ExecutionSpace = typename PetscScalarKokkosViewType<MemorySpace>::traits::device_type; in VecRestoreKokkosView_Private()
50 using ExecutionSpace = typename PetscScalarKokkosViewType<MemorySpace>::traits::device_type; in VecGetKokkosView()