Lines Matching refs:exec
238 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in Pack() local
242 Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) { in Pack()
261 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in UnpackAndOp() local
265 Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) { in UnpackAndOp()
280 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in FetchAndOp() local
284 Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) { in FetchAndOp()
299 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in ScatterAndOp() local
328 Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) { in ScatterAndOp()
362 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in ScatterAndInsert() local
371 Kokkos::deep_copy(exec, dbuf, sbuf); in ScatterAndInsert()
387 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in FetchAndOpLocal() local
391 Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) { in FetchAndOpLocal()
568 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in PetscSFLinkSyncStream_Kokkos() local
571 exec.fence(); in PetscSFLinkSyncStream_Kokkos()
577 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in PetscSFLinkMemcpy_Kokkos() local
582 …PetscCallCXX(exec.fence()); // make sure async kernels on src ar… in PetscSFLinkMemcpy_Kokkos()
588 PetscCallCXX(Kokkos::deep_copy(exec, dbuf, sbuf)); in PetscSFLinkMemcpy_Kokkos()
593 PetscCallCXX(Kokkos::deep_copy(exec, dbuf, sbuf)); in PetscSFLinkMemcpy_Kokkos()
594 PetscCallCXX(exec.fence()); // make sure dbuf is ready for use immediately on host in PetscSFLinkMemcpy_Kokkos()
599 PetscCallCXX(Kokkos::deep_copy(exec, dbuf, sbuf)); in PetscSFLinkMemcpy_Kokkos()