Home
last modified time | relevance | path

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

/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckokkosimpl.hpp24 PetscScalarKokkosView unplaced_d; /* Unplaced device array in VecKokkosPlaceArray() */
34 PetscScalarKokkosView sendbuf_d, recvbuf_d; /* Buffers for remote values in VecSetValuesCOO() */
49 PetscScalarKokkosView v_d; in Vec_Kokkos()
52 v_d = PetscScalarKokkosView(array_d, n); /* Use the given device array */ in Vec_Kokkos()
69 PetscScalarKokkosView v_d(array, v_dual.extent(0)); in UpdateArray()
92 PetscScalarKokkosView v_d(array, v_dual.extent(0)); in UpdateArray()
H A Dveckok.kokkos.cxx62 PETSC_VISIBILITY_PUBLIC PetscErrorCode VecGetKokkosView(Vec v, PetscScalarKokkosView *kv) in VecGetKokkosView()
67 PETSC_VISIBILITY_PUBLIC PetscErrorCode VecRestoreKokkosView(Vec v, PetscScalarKokkosView *kv) in VecRestoreKokkosView()
72 PETSC_VISIBILITY_PUBLIC PetscErrorCode VecGetKokkosViewWrite(Vec v, PetscScalarKokkosView *kv) in VecGetKokkosViewWrite()
77 PETSC_VISIBILITY_PUBLIC PetscErrorCode VecRestoreKokkosViewWrite(Vec v, PetscScalarKokkosView *kv) in VecRestoreKokkosViewWrite()
121 PetscScalarKokkosView xv; in VecAbs_SeqKokkos()
136 PetscScalarKokkosView xv; in VecReciprocal_SeqKokkos()
213 PetscScalarKokkosView xv; in VecShift_SeqKokkos()
237 PetscScalarKokkosView yv; in VecAXPY_SeqKokkos()
577 …auto zv = PetscScalarKokkosDualView(PetscScalarKokkosView(z_d + i, m), PetscScalarKokkosVie… in VecMultiDot_SeqKokkos_GEMV()
618 PetscScalarKokkosView xv; in VecSet_SeqKokkos()
[all …]
/petsc/src/ksp/pc/impls/vpbjacobi/kokkos/
H A Dvpbjacobi_kok.kokkos.cxx23 PetscScalarKokkosView diag; // buffer to store diagonal blocks
24 PetscScalarKokkosView work; // work buffer, with the same size as diag[]
76 PetscScalarKokkosView yv; in PCApplyOrTranspose_VPBJacobi_Kokkos()
77 PetscScalarKokkosView diag = pckok->diag; in PCApplyOrTranspose_VPBJacobi_Kokkos()
99 const auto &y1 = PetscScalarKokkosView(ybuf, n); in PCApplyOrTranspose_VPBJacobi_Kokkos()
/petsc/src/ksp/pc/impls/pbjacobi/kokkos/
H A Dpbjacobi_kok.kokkos.cxx36 PetscScalarKokkosView yv; in PCApplyOrTranspose_PBJacobi_Kokkos()
37 PetscScalarKokkosView Av = pckok->diag_dual.view_device(); in PCApplyOrTranspose_PBJacobi_Kokkos()
/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.hpp63 PetscScalarKokkosView workVector;
231 …st PetscIntKokkosView &, const PetscIntKokkosView &, PetscScalarKokkosView &, PetscScalarKokkosVie…
H A Daijkok.kokkos.cxx390 PetscScalarKokkosView yv; in MatMult_SeqAIJKokkos()
413 PetscScalarKokkosView yv; in MatMultTranspose_SeqAIJKokkos()
443 PetscScalarKokkosView yv; in MatMultHermitianTranspose_SeqAIJKokkos()
472 PetscScalarKokkosView zv; in MatMultAdd_SeqAIJKokkos()
495 PetscScalarKokkosView zv; in MatMultTransposeAdd_SeqAIJKokkos()
526 PetscScalarKokkosView zv; in MatMultHermitianTransposeAdd_SeqAIJKokkos()
1152 PetscScalarKokkosView xv; in MatGetDiagonal_SeqAIJKokkos()
1492 …KokkosView &bs2, const PetscIntKokkosView &blkMap, PetscScalarKokkosView &work, PetscScalarKokkosV… in MatInvertVariableBlockDiagonal_SeqAIJKokkos()
1517 const auto &W = PetscScalarKokkosView(&work(bs2(bid)), m * m); in MatInvertVariableBlockDiagonal_SeqAIJKokkos()
1785 PetscScalarKokkosView D = factors->D_d; in MatSolve_SeqAIJKokkos_Cholesky()
[all …]
/petsc/src/vec/vec/impls/mpi/kokkos/
H A Dmpikok.kokkos.cxx160 PetscScalarKokkosView &sendbuf = veckok->sendbuf_d; in VecSetValuesCOO_MPIKokkos()
161 PetscScalarKokkosView &recvbuf = veckok->recvbuf_d; in VecSetValuesCOO_MPIKokkos()
162 PetscScalarKokkosView xv; in VecSetValuesCOO_MPIKokkos()
/petsc/src/snes/tutorials/
H A Dex55k.kokkos.cxx19 using PetscScalarKokkosView = Kokkos::View<PetscScalar *, DefaultMemorySpace>; typedef
257 PetscScalarKokkosView coo_v("coo_v", user->ncoo); in FormJacobianLocalVec()
/petsc/include/petsc/private/
H A Dkokkosimpl.hpp72 using PetscScalarKokkosView = PetscScalarKokkosViewType<DefaultMemorySpace>; typedef
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx22 PetscScalarKokkosView v; in MatAssemblyEnd_MPIAIJKokkos()