Lines Matching refs:DeviceExecutionSpace
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()
387 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in FetchAndOpLocal()
391 Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) { in FetchAndOpLocal()
568 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in PetscSFLinkSyncStream_Kokkos()
577 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in PetscSFLinkMemcpy_Kokkos()