Home
last modified time | relevance | path

Searched refs:MemorySpace (Results 1 – 8 of 8) sorted by relevance

/petsc/include/
H A Dpetscdmda_kokkos.hpp120 template <class MemorySpace>
121 …GetKokkosOffsetView(DM, Vec, Kokkos::Experimental::OffsetView<const PetscScalar *, MemorySpace> *);
122 template <class MemorySpace>
123 …MDAVecGetKokkosOffsetView(DM, Vec, Kokkos::Experimental::OffsetView<PetscScalar *, MemorySpace> *);
124 template <class MemorySpace>
125 …cGetKokkosOffsetViewWrite(DM, Vec, Kokkos::Experimental::OffsetView<PetscScalar *, MemorySpace> *);
127 template <class MemorySpace>
128 …, Vec, Kokkos::Experimental::OffsetView<const PetscScalar **, Kokkos::LayoutRight, MemorySpace> *);
129 template <class MemorySpace>
130 …iew(DM, Vec, Kokkos::Experimental::OffsetView<PetscScalar **, Kokkos::LayoutRight, MemorySpace> *);
[all …]
H A Dpetscvec_kokkos.hpp52 template <class MemorySpace>
53 PetscErrorCode VecGetKokkosView(Vec, Kokkos::View<const PetscScalar *, MemorySpace> *);
54 template <class MemorySpace>
55 PetscErrorCode VecGetKokkosView(Vec, Kokkos::View<PetscScalar *, MemorySpace> *);
80 template <class MemorySpace>
81 PetscErrorCode VecRestoreKokkosView(Vec, Kokkos::View<const PetscScalar *, MemorySpace> *) in VecRestoreKokkosView() argument
85 template <class MemorySpace>
86 PetscErrorCode VecRestoreKokkosView(Vec, Kokkos::View<PetscScalar *, MemorySpace> *);
117 template <class MemorySpace>
118 PetscErrorCode VecGetKokkosViewWrite(Vec, Kokkos::View<PetscScalar *, MemorySpace> *);
[all …]
/petsc/include/petsc/private/
H A Dkokkosimpl.hpp24 template <class MemorySpace>
25 using PetscScalarKokkosViewType = Kokkos::View<PetscScalar *, MemorySpace>;
26 template <class MemorySpace>
27 using PetscScalarKokkosView1DType = Kokkos::View<PetscScalar *, MemorySpace>;
28 template <class MemorySpace>
29 using PetscScalarKokkosView2DType = Kokkos::View<PetscScalar **, Kokkos::LayoutRight, MemorySpace>;
30 template <class MemorySpace>
31 using PetscScalarKokkosView3DType = Kokkos::View<PetscScalar ***, Kokkos::LayoutRight, MemorySpace>;
32 template <class MemorySpace>
33 using PetscScalarKokkosView4DType = Kokkos::View<PetscScalar ****, Kokkos::LayoutRight, MemorySpace…
[all …]
/petsc/src/dm/impls/da/kokkos/
H A Ddagetov.kokkos.cxx26 template <class MemorySpace>
27 …osOffsetView_Private(DM da, Vec vec, PetscScalarKokkosOffsetView1DType<MemorySpace> *ov, PetscBool… in DMDAVecGetKokkosOffsetView_Private()
30 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetView_Private()
41 *ov = PetscScalarKokkosOffsetView1DType<MemorySpace>(kv.data(), {gxs * dof}, {(gxs + gxm) * dof}); in DMDAVecGetKokkosOffsetView_Private()
45 template <class MemorySpace>
46 …osOffsetView_Private(DM da, Vec vec, PetscScalarKokkosOffsetView1DType<MemorySpace> *ov, PetscBool… in DMDAVecRestoreKokkosOffsetView_Private()
48 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetView_Private()
60 template <class MemorySpace>
61 …DMDAVecGetKokkosOffsetView(DM da, Vec vec, ConstPetscScalarKokkosOffsetView1DType<MemorySpace> *ov) in DMDAVecGetKokkosOffsetView()
64 ConstPetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetView()
[all …]
/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckokkosimpl.hpp66 …name MemorySpace, std::enable_if_t<std::is_same<MemorySpace, HostMirrorMemorySpace>::value, bool> …
78 …name MemorySpace, std::enable_if_t<std::is_same<MemorySpace, HostMirrorMemorySpace>::value, bool> …
89 …name MemorySpace, std::enable_if_t<!std::is_same<MemorySpace, HostMirrorMemorySpace>::value, bool>…
H A Dveckok.kokkos.cxx17 template <class MemorySpace>
18 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()
32 template <class MemorySpace>
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()
46 template <class MemorySpace>
47 PetscErrorCode VecGetKokkosView(Vec v, ConstPetscScalarKokkosViewType<MemorySpace> *kv) in VecGetKokkosView()
50 using ExecutionSpace = typename PetscScalarKokkosViewType<MemorySpace>::traits::device_type; in VecGetKokkosView()
/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.hpp22 template <class MemorySpace>
23 …xType = typename KokkosSparse::CrsMatrix<MatScalarType, MatColIdxType, MemorySpace, void /* Memory…
24 template <class MemorySpace>
25 using KokkosCsrGraphType = typename KokkosCsrMatrixType<MemorySpace>::staticcrsgraph_type;
/petsc/doc/manual/
H A Dvec.md443 VecGetKokkosView(Vec v, Kokkos::View<const PetscScalar*, MemorySpace> *kv)
570 DMDAVecGetKokkosOffsetView(DM dm, Vec vec, Kokkos::View<const PetscScalar*XX*, MemorySpace> *ov)