Home
last modified time | relevance | path

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

/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckok.kokkos.cxx21 using ExecutionSpace = typename PetscScalarKokkosViewType<MemorySpace>::traits::device_type; in VecGetKokkosView_Private() typedef
26 PetscCallCXX(veckok->v_dual.sync<ExecutionSpace>()); in VecGetKokkosView_Private()
28 PetscCallCXX(*kv = veckok->v_dual.view<ExecutionSpace>()); in VecGetKokkosView_Private()
36 using ExecutionSpace = typename PetscScalarKokkosViewType<MemorySpace>::traits::device_type; in VecRestoreKokkosView_Private() typedef
41 PetscCallCXX(veckok->v_dual.modify<ExecutionSpace>()); in VecRestoreKokkosView_Private()
50 using ExecutionSpace = typename PetscScalarKokkosViewType<MemorySpace>::traits::device_type; in VecGetKokkosView() typedef
54 PetscCallCXX(veckok->v_dual.sync<ExecutionSpace>()); in VecGetKokkosView()
55 PetscCallCXX(*kv = veckok->v_dual.view<ExecutionSpace>()); in VecGetKokkosView()
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx183 template <class ExecutionSpace>
187 constexpr bool is_gpu_exec_space = KokkosKernels::Impl::kk_is_gpu_exec_space<ExecutionSpace>(); in MatMergeGetLaunchParameters()
189 constexpr bool is_gpu_exec_space = KokkosKernels::Impl::is_gpu_exec_space_v<ExecutionSpace>; in MatMergeGetLaunchParameters()
191 Kokkos::TeamPolicy<ExecutionSpace> teamPolicy(128, Kokkos::AUTO); in MatMergeGetLaunchParameters()
227 PetscInt conc = ExecutionSpace().concurrency(); in MatMergeGetLaunchParameters()