seq/kokkos/aijkok.kokkos.cxx

259 +  perm_d - the permutation array on device, which connects Ta(i) = Aa(perm(i))
321   const auto &Aa = akok->a_dual.view_device();
332         PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, nz), KOKKOS_LAMBDA(const PetscInt i) { Ta(i) = Aa(perm(i)); }));
339       PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, nz), KOKKOS_LAMBDA(const PetscInt i) { T.values(i) = Aa(perm(i)); }));
359   const auto &Aa = akok->a_dual.view_device();
370         PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, nz), KOKKOS_LAMBDA(const PetscInt i) { Ta(i) = PetscConj(Aa(perm(i))); }));
377       PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, nz), KOKKOS_LAMBDA(const PetscInt i) { T.values(i) = PetscConj(Aa(perm(i))); }));
1046     const auto &Aa     = aijkok->a_dual.view_device();
1048     PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, n), KOKKOS_LAMBDA(const PetscInt i) { Aa(Adiag(i)) += a; }));
1075     const auto &Aa     = aijkok->a_dual.view_device();
1079         if (is == INSERT_VALUES) Aa(Adiag(i)) = dv(i);
1080         else Aa(Adiag(i)) += dv(i);
1102   const auto &Aa     = aijkok->a_dual.view_device();
1114         Kokkos::parallel_for(Kokkos::TeamThreadRange(t, len), [&](PetscInt j) { Aa(Ai(i) + j) *= lv(i); });
1124       Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, nz), KOKKOS_LAMBDA(const PetscInt k) { Aa(k) *= rv(Aj(k)); }));
1162   const auto &Aa    = aijkok->a_dual.view_device();
1169       if (Adiag(i) < Ai(i + 1)) xv(i) = Aa(Adiag(i));
1395   MatScalarKokkosView        Aa;
1417   if (imode == INSERT_VALUES) PetscCall(MatSeqAIJGetKokkosViewWrite(A, &Aa)); /* write matrix values */
1418   else PetscCall(MatSeqAIJGetKokkosView(A, &Aa));                             /* read & write matrix values */
1425       Aa(i) = (imode == INSERT_VALUES ? 0.0 : Aa(i)) + sum;
1429   if (imode == INSERT_VALUES) PetscCall(MatSeqAIJRestoreKokkosViewWrite(A, &Aa));
1430   else PetscCall(MatSeqAIJRestoreKokkosView(A, &Aa));
1501   auto Aa    = akok->a_dual.view_device();
1529               B(r, c) = Aa(first + c);
1539             if (rstart <= Aj(j) && Aj(j) < rstart + m) B(r, Aj(j) - rstart) = Aa(j);
2241       D[i]      = Ba[Bdiag[i]];     // actually Aa[Adiag[i]] is the inverse of the diagonal