Searched refs:DeviceExecutionSpace (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/vec/is/sf/impls/basic/kokkos/ |
| H A D | sfkok.kokkos.cxx | 6 using DeviceExecutionSpace = Kokkos::DefaultExecutionSpace; typedef
238 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in Pack()
242 Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) { in Pack()
261 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in UnpackAndOp()
265 Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) { in UnpackAndOp()
280 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in FetchAndOp()
284 Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) { in FetchAndOp()
299 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in ScatterAndOp()
328 Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) { in ScatterAndOp()
362 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in ScatterAndInsert()
[all …]
|