Home
last modified time | relevance | path

Searched refs:DefaultMemorySpace (Results 1 – 12 of 12) sorted by relevance

/petsc/include/petsc/private/
H A Dkokkosimpl.hpp15 using DefaultMemorySpace = Kokkos::DefaultExecutionSpace::memory_space; typedef
72 using PetscScalarKokkosView = PetscScalarKokkosViewType<DefaultMemorySpace>;
73 using PetscScalarKokkosView1D = PetscScalarKokkosView1DType<DefaultMemorySpace>;
74 using PetscScalarKokkosView2D = PetscScalarKokkosView2DType<DefaultMemorySpace>;
75 using PetscScalarKokkosView3D = PetscScalarKokkosView3DType<DefaultMemorySpace>;
76 using PetscScalarKokkosView4D = PetscScalarKokkosView4DType<DefaultMemorySpace>;
84 using ConstPetscScalarKokkosView = ConstPetscScalarKokkosViewType<DefaultMemorySpace>;
85 using ConstPetscScalarKokkosView1D = ConstPetscScalarKokkosView1DType<DefaultMemorySpace>;
86 using ConstPetscScalarKokkosView2D = ConstPetscScalarKokkosView2DType<DefaultMemorySpace>;
87 using ConstPetscScalarKokkosView3D = ConstPetscScalarKokkosView3DType<DefaultMemorySpace>;
[all …]
/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckokkosimpl.hpp54 …v_d = Kokkos::create_mirror_view(Kokkos::WithoutInitializing, DefaultMemorySpace(), v_h); /* Creat… in Vec_Kokkos()
66 …>::value, bool> = true, std::enable_if_t<std::is_same<MemorySpace, DefaultMemorySpace>::value, boo…
78 …::value, bool> = true, std::enable_if_t<!std::is_same<MemorySpace, DefaultMemorySpace>::value, boo…
84 PetscCallCXX(v_dual = PetscScalarKokkosDualView(v_dual.view<DefaultMemorySpace>(), v_h)); in UpdateArray()
89 …>::value, bool> = true, std::enable_if_t<std::is_same<MemorySpace, DefaultMemorySpace>::value, boo…
103 …PetscCallCXX(jmap1_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosV… in SetUpCOO()
104 …PetscCallCXX(perm1_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosV… in SetUpCOO()
111 …PetscCallCXX(jmap1_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosV… in SetUpCOO()
112 …PetscCallCXX(perm1_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosV… in SetUpCOO()
113 …PetscCallCXX(imap2_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosV… in SetUpCOO()
[all …]
H A Dveckok.kokkos.cxx1303 PetscCall(veckok->UpdateArray<DefaultMemorySpace>(a)); in VecKokkosPlaceArray()
1335 PetscCall(veckok->UpdateArray<DefaultMemorySpace>(veckok->unplaced_d.data())); in VecKokkosResetArray()
1589 …PetscCallCXX(vv = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), ConstPetscScalarKokkos… in VecSetValuesCOO_SeqKokkos()
1746 if (std::is_same<DefaultMemorySpace, HostMirrorMemorySpace>::value) { in VecCreateSeqKokkosWithArray()
/petsc/src/mat/tutorials/
H A Dex18k.kokkos.cxx5 using DefaultMemorySpace = Kokkos::DefaultExecutionSpace::memory_space; typedef
9 Kokkos::View<PetscScalar *, DefaultMemorySpace> v("v", 3 * 3 * fe->Ne); in FillMatrixKokkosCOO()
/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.hpp27 using KokkosCsrGraph = KokkosCsrGraphType<DefaultMemorySpace>;
30 using KokkosCsrMatrix = KokkosCsrMatrixType<DefaultMemorySpace>;
53 …Type, MatColIdxType, MatScalarType, DefaultExecutionSpace, DefaultMemorySpace, DefaultMemorySpace>;
189 …auto diag_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), diag_h); in SetDiagonal()
H A Daijkok.kokkos.cxx103 …auto diag_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), diag_h); in MatAssemblyEnd_SeqAIJKokkos()
301 auto Ti_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), Ti_h); in MatSeqAIJKokkosGenerateTransposeStructure()
302 auto Tj_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), Tj_h); in MatSeqAIJKokkosGenerateTransposeStructure()
304 PetscCallCXX(perm_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), perm_h)); in MatSeqAIJKokkosGenerateTransposeStructure()
1355 …jmap = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(coo_h->j… in MatCOOStruct_SeqAIJKokkos()
1356 …perm = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(coo_h->p… in MatCOOStruct_SeqAIJKokkos()
1412 …kv = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), ConstMatScalarKokkosViewHost(v, n)); in MatSetValuesCOO_SeqAIJKokkos()
1974 factors->iL_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), factors->iL_h); in MatLUFactorNumeric_SeqAIJKokkos()
1975 factors->jL_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), factors->jL_h); in MatLUFactorNumeric_SeqAIJKokkos()
1976 factors->iU_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), factors->iU_h); in MatLUFactorNumeric_SeqAIJKokkos()
[all …]
/petsc/src/snes/tutorials/
H A Dex55k.kokkos.cxx13 using DefaultMemorySpace = Kokkos::DefaultExecutionSpace::memory_space; typedef
14 … = Kokkos::Experimental::OffsetView<const PetscScalar **, Kokkos::LayoutRight, DefaultMemorySpace>;
15 …D = Kokkos::Experimental::OffsetView<PetscScalar **, Kokkos::LayoutRight, DefaultMemorySpace>;
17 using PetscCountKokkosView = Kokkos::View<PetscCount *, DefaultMemorySpace>;
18 using PetscIntKokkosView = Kokkos::View<PetscInt *, DefaultMemorySpace>;
19 using PetscScalarKokkosView = Kokkos::View<PetscScalar *, DefaultMemorySpace>;
H A Dex3k.kokkos.cxx85 using DefaultMemorySpace = Kokkos::DefaultExecutionSpace::memory_space; typedef
86 …cScalarKokkosOffsetView = Kokkos::Experimental::OffsetView<PetscScalar *, DefaultMemorySpace>;
87 …ScalarKokkosOffsetView = Kokkos::Experimental::OffsetView<const PetscScalar *, DefaultMemorySpace>;
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx603 mm->E_NzLeft = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), E_NzLeft_h); in MatMPIAIJKokkosReduceBegin()
604 mm->Fdjmap = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), Fdjmap_h); in MatMPIAIJKokkosReduceBegin()
605 mm->Fdjperm = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), Fdjperm_h); in MatMPIAIJKokkosReduceBegin()
606 mm->Fojmap = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), Fojmap_h); in MatMPIAIJKokkosReduceBegin()
607 mm->Fojperm = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), Fojperm_h); in MatMPIAIJKokkosReduceBegin()
611 MatRowMapKokkosView Fdi_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), Fdi_h); in MatMPIAIJKokkosReduceBegin()
612 MatColIdxKokkosView Fdj_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), Fdj_h); in MatMPIAIJKokkosReduceBegin()
614 MatRowMapKokkosView Foi_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), Foi_h); in MatMPIAIJKokkosReduceBegin()
615 MatColIdxKokkosView Foj_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), Foj_h); in MatMPIAIJKokkosReduceBegin()
914 mm->E_NzLeft = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), E_NzLeft_h); in MatMPIAIJKokkosBcastBegin()
[all …]
/petsc/src/vec/vec/impls/mpi/kokkos/
H A Dmpikok.kokkos.cxx171 …vv = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscScalarKokkosViewHost(const_cas… in VecSetValuesCOO_MPIKokkos()
445 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.cxx8 typedef Kokkos::View<char *, DefaultMemorySpace> deviceBuffer_t;
11 typedef Kokkos::View<const char *, DefaultMemorySpace> deviceConstBuffer_t;
611 PetscCallCXX(*ptr = Kokkos::kokkos_malloc<DefaultMemorySpace>(size)); in PetscSFMalloc_Kokkos()
621 PetscCallCXX(Kokkos::kokkos_free<DefaultMemorySpace>(ptr)); in PetscSFFree_Kokkos()
/petsc/src/ksp/pc/impls/bjacobi/bjkokkos/
H A Dbjkokkos.kokkos.cxx809 …jac->d_isrow_k = new Kokkos::View<PetscInt *>(Kokkos::create_mirror(DefaultMemorySpace(), h_isrow… in PCSetUp_BJKOKKOS()
810 …jac->d_isicol_k = new Kokkos::View<PetscInt *>(Kokkos::create_mirror(DefaultMemorySpace(), h_isico… in PCSetUp_BJKOKKOS()