Home
last modified time | relevance | path

Searched refs:HostMirrorMemorySpace (Results 1 – 10 of 10) sorted by relevance

/petsc/include/petsc/private/
H A Dkokkosimpl.hpp16 using HostMirrorMemorySpace = Kokkos::DualView<PetscScalar *>::host_mirror_space::memory_space; typedef
78 using PetscScalarKokkosViewHost = PetscScalarKokkosViewType<HostMirrorMemorySpace>;
79 using PetscScalarKokkosView1DHost = PetscScalarKokkosView1DType<HostMirrorMemorySpace>;
80 using PetscScalarKokkosView2DHost = PetscScalarKokkosView2DType<HostMirrorMemorySpace>;
81 using PetscScalarKokkosView3DHost = PetscScalarKokkosView3DType<HostMirrorMemorySpace>;
82 using PetscScalarKokkosView4DHost = PetscScalarKokkosView4DType<HostMirrorMemorySpace>;
90 using ConstPetscScalarKokkosViewHost = ConstPetscScalarKokkosViewType<HostMirrorMemorySpace>;
91 using ConstPetscScalarKokkosView1DHost = ConstPetscScalarKokkosView1DType<HostMirrorMemorySpace>;
92 using ConstPetscScalarKokkosView2DHost = ConstPetscScalarKokkosView2DType<HostMirrorMemorySpace>;
93 using ConstPetscScalarKokkosView3DHost = ConstPetscScalarKokkosView3DType<HostMirrorMemorySpace>;
[all …]
/petsc/src/mat/tests/
H A Dex302k.kokkos.cxx10 using HostMirrorMemorySpace = Kokkos::DualView<PetscScalar *>::host_mirror_space::memory_space; typedef
96 Kokkos::View<PetscScalar *, HostMirrorMemorySpace> a_local_h; in main()
97 Kokkos::View<PetscInt *, HostMirrorMemorySpace> i_local_h; in main()
98 Kokkos::View<PetscInt *, HostMirrorMemorySpace> j_local_h; in main()
99 Kokkos::View<PetscScalar *, HostMirrorMemorySpace> a_nonlocal_h; in main()
100 Kokkos::View<PetscInt *, HostMirrorMemorySpace> i_nonlocal_h; in main()
101 Kokkos::View<PetscInt *, HostMirrorMemorySpace> j_nonlocal_h; in main()
103 PetscCallCXX(a_local_h = Kokkos::View<PetscScalar *, HostMirrorMemorySpace>(da, di[5])); in main()
104 PetscCallCXX(i_local_h = Kokkos::View<PetscInt *, HostMirrorMemorySpace>(di, m + 1)); in main()
105 PetscCallCXX(j_local_h = Kokkos::View<PetscInt *, HostMirrorMemorySpace>(dj, di[5])); in main()
[all …]
/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.hpp28 using KokkosCsrGraphHost = KokkosCsrGraphType<HostMirrorMemorySpace>;
31 using KokkosCsrMatrixHost = KokkosCsrMatrixType<HostMirrorMemorySpace>;
155 … a_h = Kokkos::create_mirror_view(Kokkos::WithoutInitializing, HostMirrorMemorySpace(), a_d); in Mat_SeqAIJKokkos()
156 auto i_h = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), i_d); in Mat_SeqAIJKokkos()
157 auto j_h = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), j_d); in Mat_SeqAIJKokkos()
H A Daijkok.kokkos.cxx2360 …const auto &iv = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), csrmat.graph.… in PrintCsrMatrix()
2361 …const auto &jv = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), csrmat.graph.… in PrintCsrMatrix()
2362 …const auto &av = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), csrmat.values… in PrintCsrMatrix()
/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckokkosimpl.hpp66 …template <typename MemorySpace, std::enable_if_t<std::is_same<MemorySpace, HostMirrorMemorySpace>:…
78 …template <typename MemorySpace, std::enable_if_t<std::is_same<MemorySpace, HostMirrorMemorySpace>:…
89 …template <typename MemorySpace, std::enable_if_t<!std::is_same<MemorySpace, HostMirrorMemorySpace>…
H A Dveckok.kokkos.cxx1251 PetscCall(veckok->UpdateArray<HostMirrorMemorySpace>(vecseq->array)); in VecPlaceArray_SeqKokkos()
1264 PetscCall(veckok->UpdateArray<HostMirrorMemorySpace>(vecseq->array)); in VecResetArray_SeqKokkos()
1349 PetscCall(veckok->UpdateArray<HostMirrorMemorySpace>(vecseq->array)); in VecReplaceArray_SeqKokkos()
1746 if (std::is_same<DefaultMemorySpace, HostMirrorMemorySpace>::value) { in VecCreateSeqKokkosWithArray()
/petsc/src/dm/tests/
H A Dex2k.kokkos.cxx27 using HostMirrorMemorySpace = Kokkos::DualView<PetscScalar *>::host_mirror_space::memo… typedef
28 … = Kokkos::Experimental::OffsetView<PetscScalar ***, Kokkos::LayoutRight, HostMirrorMemorySpace>;
29 …okkos::Experimental::OffsetView<const PetscScalar ***, Kokkos::LayoutRight, HostMirrorMemorySpace>;
/petsc/src/vec/vec/impls/mpi/kokkos/
H A Dmpikok.kokkos.cxx445 if (std::is_same<DefaultMemorySpace, HostMirrorMemorySpace>::value) { in VecCreateMPIKokkosWithArray()
452 …if (!std::is_same<DefaultMemorySpace, HostMirrorMemorySpace>::value) vecmpi->array_allocated = har… in VecCreateMPIKokkosWithArray()
499 …if (std::is_same<DefaultMemorySpace, HostMirrorMemorySpace>::value) PetscCheck(harray == darray, P… in VecCreateMPIKokkosWithArrays_Private()
/petsc/src/vec/is/sf/impls/basic/kokkos/
H A Dsfkok.kokkos.cxx9 typedef Kokkos::View<char *, HostMirrorMemorySpace> HostBuffer_t;
12 typedef Kokkos::View<const char *, HostMirrorMemorySpace> HostConstBuffer_t;
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx352 …auto Ai_h = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), A.graph.r… in MatMPIAIJKokkosReduceBegin()
353 …auto Aj_h = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), A.graph.e… in MatMPIAIJKokkosReduceBegin()
354 …auto Bi_h = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), B.graph.r… in MatMPIAIJKokkosReduceBegin()
355 …auto Bj_h = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), B.graph.e… in MatMPIAIJKokkosReduceBegin()