Home
last modified time | relevance | path

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

/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.hpp132 auto a_d = Kokkos::create_mirror_view(Kokkos::WithoutInitializing, exec, a_h); in Mat_SeqAIJKokkos() local
136 a_dual = MatScalarKokkosDualView(a_d, a_h); in Mat_SeqAIJKokkos()
144 csrmat = KokkosCsrMatrix("csrmat", ncols, a_d, KokkosCsrGraph(j_d, i_d)); in Mat_SeqAIJKokkos()
151 auto a_d = csr.values; in Mat_SeqAIJKokkos() local
155 … a_h = Kokkos::create_mirror_view(Kokkos::WithoutInitializing, HostMirrorMemorySpace(), a_d); in Mat_SeqAIJKokkos()
160 a_dual = MatScalarKokkosDualView(a_d, a_h); in Mat_SeqAIJKokkos()
169 …osView &j_d, const MatScalarKokkosView &a_d) : Mat_SeqAIJKokkos(KokkosCsrMatrix("csrmat", nrows, n… in Mat_SeqAIJKokkos()
H A Daijkok.kokkos.cxx1247 …os::View<PetscInt *> &i_d, Kokkos::View<PetscInt *> &j_d, Kokkos::View<PetscScalar *> &a_d, Mat *A) in MatCreateSeqAIJKokkosWithKokkosViews() argument
1252 PetscCallCXX(akok = new Mat_SeqAIJKokkos(m, n, j_d.extent(0), i_d, j_d, a_d)); in MatCreateSeqAIJKokkosWithKokkosViews()
2063 auto a_d = aijkok->a_dual.view_device(); in MatILUFactorNumeric_SeqAIJKokkos_ILU0() local
2067 …PetscCallCXX(spiluk_numeric(&factors->kh, fill_lev, i_d, j_d, a_d, factors->iL_d, factors->jL_d, f… in MatILUFactorNumeric_SeqAIJKokkos_ILU0()
/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckok.kokkos.cxx846 PetscScalar *a_d = nullptr; in VecMAXPY_SeqKokkos_GEMV() local
852 a_d = const_cast<PetscScalar *>(a_h); in VecMAXPY_SeqKokkos_GEMV()
874 if (!a_d) { in VecMAXPY_SeqKokkos_GEMV()
879 a_d = PetscScalarPool; in VecMAXPY_SeqKokkos_GEMV()
883 …auto av = PetscScalarKokkosDualView(PetscScalarKokkosView(a_d + i, m), PetscScalarKokkosVie… in VecMAXPY_SeqKokkos_GEMV()