Home
last modified time | relevance | path

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

/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.hpp127 MatScalarKokkosViewHost a_h(aseq->a, aseq->nz); in Mat_SeqAIJKokkos() local
132 auto a_d = Kokkos::create_mirror_view(Kokkos::WithoutInitializing, exec, a_h); in Mat_SeqAIJKokkos()
136 a_dual = MatScalarKokkosDualView(a_d, a_h); in Mat_SeqAIJKokkos()
155 …auto a_h = Kokkos::create_mirror_view(Kokkos::WithoutInitializing, HostMirrorMemory… in Mat_SeqAIJKokkos() local
160 a_dual = MatScalarKokkosDualView(a_d, a_h); in Mat_SeqAIJKokkos()
H A Daijkok.kokkos.cxx658 MatScalarKokkosViewHost a_h(bseq->a, internT.nnz()); /* bseq->nz = 0 if unassembled */ in MatTranspose_SeqAIJKokkos() local
660 PetscCallCXX(Kokkos::deep_copy(a_h, internT.values)); in MatTranspose_SeqAIJKokkos()
/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckok.kokkos.cxx837 PetscErrorCode VecMAXPY_SeqKokkos_GEMV(Vec yin, PetscInt nv, const PetscScalar *a_h, Vec *xin) in VecMAXPY_SeqKokkos_GEMV() argument
852 a_d = const_cast<PetscScalar *>(a_h); in VecMAXPY_SeqKokkos_GEMV()
883 …tscScalarKokkosView(a_d + i, m), PetscScalarKokkosViewHost(const_cast<PetscScalar *>(a_h) + i, m)); in VecMAXPY_SeqKokkos_GEMV()
890 if (nfail++ == 0) PetscCall(VecAXPY_SeqKokkos(yin, a_h[i], xin[i])); in VecMAXPY_SeqKokkos_GEMV()
897 if (i < nv) PetscCall(VecMAXPY_SeqKokkos(yin, nv - i, a_h + i, xin + i)); in VecMAXPY_SeqKokkos_GEMV()