seq/kokkos/aijkok.kokkos.cxx

61    In the latter case, it is important to set a_dual's sync state correctly.
117   if (aijkok->a_dual.need_sync_device()) {
118     PetscCall(KokkosDualViewSyncDevice(aijkok->a_dual, PetscGetKokkosExecutionSpace()));
132   aijkok->a_dual.clear_sync_state();
133   aijkok->a_dual.modify_device();
150   PetscCall(KokkosDualViewSyncHost(aijkok->a_dual, exec));
165     PetscCall(KokkosDualViewSyncHost(aijkok->a_dual, PetscGetKokkosExecutionSpace()));
166     *array = aijkok->a_dual.view_host().data();
178   if (aijkok && A->nonzerostate == aijkok->nonzerostate) aijkok->a_dual.modify_host();
188     PetscCall(KokkosDualViewSyncHost(aijkok->a_dual, PetscGetKokkosExecutionSpace()));
189     *array = aijkok->a_dual.view_host().data();
209     *array = aijkok->a_dual.view_host().data();
222     aijkok->a_dual.clear_sync_state();
223     aijkok->a_dual.modify_host();
238     PetscCall(KokkosDualViewSyncDevice(aijkok->a_dual, PetscGetKokkosExecutionSpace()));
319   PetscCall(KokkosDualViewSyncDevice(akok->a_dual, PetscGetKokkosExecutionSpace())); // Sync A's values since we are going to access them on device
321   const auto &Aa = akok->a_dual.view_device();
357   PetscCall(KokkosDualViewSyncDevice(akok->a_dual, PetscGetKokkosExecutionSpace())); // Sync A's values since we are going to access them on device
359   const auto &Aa = akok->a_dual.view_device();
612     bkok->a_dual.clear_sync_state(); /* Clear B's sync state as it will be decided below */
615       if (akok->a_dual.need_sync_device()) {
616         Kokkos::deep_copy(bkok->a_dual.view_host(), akok->a_dual.view_host());
617         bkok->a_dual.modify_host();
619         Kokkos::deep_copy(bkok->a_dual.view_device(), akok->a_dual.view_device());
620         bkok->a_dual.modify_device();
624       bkok->a_dual.modify_host();
654       PetscCallCXX(Kokkos::deep_copy(bkok->a_dual.view_device(), internT.values));
735   aa   = akok->a_dual.view_device();
737   ba   = bkok->a_dual.view_device();
777     ca   = ckok->a_dual.view_device();
1027   KokkosBlas::scal(PetscGetKokkosExecutionSpace(), aijkok->a_dual.view_device(), a, aijkok->a_dual.view_device());
1029   PetscCall(PetscLogGpuFlops(aijkok->a_dual.extent(0)));
1046     const auto &Aa     = aijkok->a_dual.view_device();
1075     const auto &Aa     = aijkok->a_dual.view_device();
1102   const auto &Aa     = aijkok->a_dual.view_device();
1140     KokkosBlas::fill(PetscGetKokkosExecutionSpace(), aijkok->a_dual.view_device(), 0.0);
1162   const auto &Aa    = aijkok->a_dual.view_device();
1187   *kv    = aijkok->a_dual.view_device();
1210   *kv    = aijkok->a_dual.view_device();
1233   *kv    = aijkok->a_dual.view_device();
1290   Xa   = xkok->a_dual.view_device();
1291   Ya   = ykok->a_dual.view_device();
1339   PetscCall(PetscLogGpuFlops(xkok->a_dual.extent(0) * 2)); // Because we scaled X and then added it to Y
1501   auto Aa    = akok->a_dual.view_device();
2063   auto a_d = aijkok->a_dual.view_device();