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()