Home
last modified time | relevance | path

Searched refs:vecmpi (Results 1 – 2 of 2) sorted by relevance

/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckokkosimpl.hpp108 PetscErrorCode SetUpCOO(const Vec_MPI *vecmpi, PetscInt m) in SetUpCOO()
111 …create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(vecmpi->jmap1, m + 1))); in SetUpCOO()
112 …mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(vecmpi->perm1, vecmpi->tot1))); in SetUpCOO()
113 …mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(vecmpi->imap2, vecmpi->nnz2))); in SetUpCOO()
114 …rror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(vecmpi->jmap2, vecmpi->nnz2 + 1)… in SetUpCOO()
115 …rror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(vecmpi->perm2, vecmpi->recvlen))… in SetUpCOO()
116 …rror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(vecmpi->Cperm, vecmpi->sendlen))… in SetUpCOO()
117 …or_view_and_copy(DefaultMemorySpace(), PetscScalarKokkosViewHost(vecmpi->sendbuf, vecmpi->sendlen)… in SetUpCOO()
118 …or_view_and_copy(DefaultMemorySpace(), PetscScalarKokkosViewHost(vecmpi->recvbuf, vecmpi->recvlen)… in SetUpCOO()
/petsc/src/vec/vec/impls/mpi/kokkos/
H A Dmpikok.kokkos.cxx139 const auto vecmpi = static_cast<Vec_MPI *>(x->data); in VecSetPreallocationCOO_MPIKokkos() local
146 PetscCall(veckok->SetUpCOO(vecmpi, m)); in VecSetPreallocationCOO_MPIKokkos()
152 const auto vecmpi = static_cast<Vec_MPI *>(x->data); in VecSetValuesCOO_MPIKokkos() local
171 …copy(DefaultMemorySpace(), PetscScalarKokkosViewHost(const_cast<PetscScalar *>(v), vecmpi->coo_n)); in VecSetValuesCOO_MPIKokkos()
173 vv = ConstPetscScalarKokkosView(v, vecmpi->coo_n); /* Directly use v[]'s memory */ in VecSetValuesCOO_MPIKokkos()
177 …Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, vecmpi->sendlen), KO… in VecSetValuesCOO_MPIKokkos()
178 …PetscCall(PetscSFReduceWithMemTypeBegin(vecmpi->coo_sf, MPIU_SCALAR, PETSC_MEMTYPE_KOKKOS, sendbuf… in VecSetValuesCOO_MPIKokkos()
190 …PetscCall(PetscSFReduceEnd(vecmpi->coo_sf, MPIU_SCALAR, sendbuf.data(), recvbuf.data(), MPI_REPLAC… in VecSetValuesCOO_MPIKokkos()
194 …Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, vecmpi->nnz2), KOKKOS_LAMBDA(PetscCount i… in VecSetValuesCOO_MPIKokkos()
269 Vec_MPI *vecmpi; in VecConvert_MPI_MPIKokkos_inplace() local
[all …]