vpbjacobi/kokkos/vpbjacobi_kok.kokkos.cxx

74   PC_VPBJacobi_Kokkos       *pckok = static_cast<PC_VPBJacobi_Kokkos *>(jac->spptr);
77   PetscScalarKokkosView      diag   = pckok->diag;
78   PetscIntKokkosView         bs     = pckok->bs_dual.view_device();
79   PetscIntKokkosView         bs2    = pckok->bs2_dual.view_device();
80   PetscIntKokkosView         blkMap = pckok->blkMap_dual.view_device();
108     label, Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, pckok->n), KOKKOS_LAMBDA(PetscInt row) {
129   PetscCall(PetscLogGpuFlops(pckok->nsize * 2)); /* FMA on entries in all blocks */
147   PC_VPBJacobi_Kokkos *pckok = static_cast<PC_VPBJacobi_Kokkos *>(jac->spptr);
171   if (pckok && (pckok->n != nlocal || pckok->nblocks != nblocks || pckok->nsize != nsize)) {
172     PetscCallCXX(delete pckok);
173     pckok = nullptr;
177   if (!pckok) {
178     PetscCallCXX(pckok = new PC_VPBJacobi_Kokkos(nlocal, nblocks, nsize, bsizes));
179     jac->spptr = pckok;
183   const auto &bs     = pckok->bs_dual.view_device();
184   const auto &bs2    = pckok->bs2_dual.view_device();
185   const auto &blkMap = pckok->blkMap_dual.view_device();
193   PetscCall(MatInvertVariableBlockDiagonal_SeqAIJKokkos(A, bs, bs2, blkMap, pckok->work, pckok->diag));
198   PetscCall(PetscLogGpuFlops(pckok->setupFlops));