6 using DeviceExecutionSpace = Kokkos::DefaultExecutionSpace;
8 typedef Kokkos::View<char *, DefaultMemorySpace>    deviceBuffer_t;
9 typedef Kokkos::View<char *, HostMirrorMemorySpace> HostBuffer_t;
11 typedef Kokkos::View<const char *, DefaultMemorySpace>    deviceConstBuffer_t;
12 typedef Kokkos::View<const char *, HostMirrorMemorySpace> HostConstBuffer_t;
142   KOKKOS_INLINE_FUNCTION void operator()(Type &x, Type y) const { Kokkos::atomic_store(&x, y); }  in operator ()()
146   KOKKOS_INLINE_FUNCTION void operator()(Type &x, Type y) const { Kokkos::atomic_add(&x, y); }  in operator ()()
150   KOKKOS_INLINE_FUNCTION void operator()(Type &x, Type y) const { Kokkos::atomic_and(&x, y); }  in operator ()()
154   KOKKOS_INLINE_FUNCTION void operator()(Type &x, Type y) const { Kokkos::atomic_or(&x, y); }  in operator ()()
158   KOKKOS_INLINE_FUNCTION void operator()(Type &x, Type y) const { Kokkos::atomic_fetch_xor(&x, y); }  in operator ()()
165     Kokkos::atomic_and(&x, y ? one : zero);  in operator ()()
173     Kokkos::atomic_or(&x, y ? one : zero);  in operator ()()
178   KOKKOS_INLINE_FUNCTION void operator()(Type &x, Type y) const { Kokkos::atomic_fetch_mul(&x, y); }  in operator ()()
182   KOKKOS_INLINE_FUNCTION void operator()(Type &x, Type y) const { Kokkos::atomic_fetch_min(&x, y); }  in operator ()()
186   KOKKOS_INLINE_FUNCTION void operator()(Type &x, Type y) const { Kokkos::atomic_fetch_max(&x, y); }  in operator ()()
191 …KOKKOS_INLINE_FUNCTION Type operator()(Type &x, Type y) const { return Kokkos::atomic_fetch_add(&x…  in operator ()()
241   Kokkos::parallel_for(  in Pack()
242     Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) {  in Pack()
264   Kokkos::parallel_for(  in UnpackAndOp()
265     Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) {  in UnpackAndOp()
283   Kokkos::parallel_for(  in FetchAndOp()
284     Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) {  in FetchAndOp()
327   Kokkos::parallel_for(  in ScatterAndOp()
328     Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) {  in ScatterAndOp()
371     Kokkos::deep_copy(exec, dbuf, sbuf);  in ScatterAndInsert()
390   Kokkos::parallel_for(  in FetchAndOpLocal()
391     Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) {  in FetchAndOpLocal()
562   Kokkos::fence();  in PetscSFLinkSyncDevice_Kokkos()
588       PetscCallCXX(Kokkos::deep_copy(exec, dbuf, sbuf));  in PetscSFLinkMemcpy_Kokkos()
593       PetscCallCXX(Kokkos::deep_copy(exec, dbuf, sbuf));  in PetscSFLinkMemcpy_Kokkos()
599       PetscCallCXX(Kokkos::deep_copy(exec, dbuf, sbuf));  in PetscSFLinkMemcpy_Kokkos()
611     PetscCallCXX(*ptr = Kokkos::kokkos_malloc<DefaultMemorySpace>(size));  in PetscSFMalloc_Kokkos()
621     PetscCallCXX(Kokkos::kokkos_free<DefaultMemorySpace>(ptr));  in PetscSFFree_Kokkos()
662     PackInit_PairType<Kokkos::pair<int, int>>(link);  in PetscSFLinkSetUp_Kokkos()
664     PackInit_PairType<Kokkos::pair<PetscInt, PetscInt>>(link);  in PetscSFLinkSetUp_Kokkos()
728     if (nPetscComplex == 8) PackInit_ComplexType<Kokkos::complex<PetscReal>, 8, 1>(link);  in PetscSFLinkSetUp_Kokkos()
729     else if (nPetscComplex % 8 == 0) PackInit_ComplexType<Kokkos::complex<PetscReal>, 8, 0>(link);  in PetscSFLinkSetUp_Kokkos()
730     else if (nPetscComplex == 4) PackInit_ComplexType<Kokkos::complex<PetscReal>, 4, 1>(link);  in PetscSFLinkSetUp_Kokkos()
731     else if (nPetscComplex % 4 == 0) PackInit_ComplexType<Kokkos::complex<PetscReal>, 4, 0>(link);  in PetscSFLinkSetUp_Kokkos()
732     else if (nPetscComplex == 2) PackInit_ComplexType<Kokkos::complex<PetscReal>, 2, 1>(link);  in PetscSFLinkSetUp_Kokkos()
733     else if (nPetscComplex % 2 == 0) PackInit_ComplexType<Kokkos::complex<PetscReal>, 2, 0>(link);  in PetscSFLinkSetUp_Kokkos()
734     else if (nPetscComplex == 1) PackInit_ComplexType<Kokkos::complex<PetscReal>, 1, 1>(link);  in PetscSFLinkSetUp_Kokkos()
737       PackInit_ComplexType<Kokkos::complex<PetscReal>, 1, 0>(link);  in PetscSFLinkSetUp_Kokkos()