Searched refs:exec_space (Results 1 – 2 of 2) sorted by relevance
| /petsc/include/petsc/private/ |
| H A D | pcbjkokkosimpl.h | 32 typedef Kokkos::DefaultExecutionSpace exec_space; typedef
34 using IntView = Kokkos::View<PetscInt **, layout, exec_space>;
35 using AMatrixValueView = const Kokkos::View<PetscScalar **, layout, exec_space>;
36 using XYType = const Kokkos::View<PetscScalar **, layout, exec_space>;
|
| /petsc/src/ksp/pc/impls/bjacobi/bjkokkos/ |
| H A D | bjkokkoskernels.kokkos.cxx | 145 exec_space().fence(); in run()
148 exec_space().fence(); in run()
177 …Kokkos::View<PetscScalar **, layout, exec_space, Kokkos::MemoryTraits<Kokkos::Unmanaged>> inv_diag… in PCApply_BJKOKKOSKERNELS()
243 using Norm2DViewType = Kokkos::View<MagnitudeType **, layout, exec_space>; in PCApply_BJKOKKOSKERNELS()
244 using Scalar3DViewType = Kokkos::View<ScalarType ***, layout, exec_space>; in PCApply_BJKOKKOSKERNELS()
245 using IntViewType = Kokkos::View<int *, layout, exec_space>; in PCApply_BJKOKKOSKERNELS()
255 …Functor_TestBatchedTeamVectorGMRES<exec_space, AMatrixValueView, IntView, XYType, KrylovHandleType… in PCApply_BJKOKKOSKERNELS()
|