Searched refs:pckok (Results 1 – 2 of 2) sorted by relevance
74 PC_VPBJacobi_Kokkos *pckok = static_cast<PC_VPBJacobi_Kokkos *>(jac->spptr); in PCApplyOrTranspose_VPBJacobi_Kokkos() local77 PetscScalarKokkosView diag = pckok->diag; in PCApplyOrTranspose_VPBJacobi_Kokkos()78 PetscIntKokkosView bs = pckok->bs_dual.view_device(); in PCApplyOrTranspose_VPBJacobi_Kokkos()79 PetscIntKokkosView bs2 = pckok->bs2_dual.view_device(); in PCApplyOrTranspose_VPBJacobi_Kokkos()80 PetscIntKokkosView blkMap = pckok->blkMap_dual.view_device(); in PCApplyOrTranspose_VPBJacobi_Kokkos()108 …label, Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, pckok->n), KOKKOS_LAMBDA(PetscInt … in PCApplyOrTranspose_VPBJacobi_Kokkos()129 PetscCall(PetscLogGpuFlops(pckok->nsize * 2)); /* FMA on entries in all blocks */ in PCApplyOrTranspose_VPBJacobi_Kokkos()147 PC_VPBJacobi_Kokkos *pckok = static_cast<PC_VPBJacobi_Kokkos *>(jac->spptr); in PCSetUp_VPBJacobi_Kokkos() local171 if (pckok && (pckok->n != nlocal || pckok->nblocks != nblocks || pckok->nsize != nsize)) { in PCSetUp_VPBJacobi_Kokkos()172 PetscCallCXX(delete pckok); in PCSetUp_VPBJacobi_Kokkos()[all …]
34 PC_PBJacobi_Kokkos *pckok = static_cast<PC_PBJacobi_Kokkos *>(jac->spptr); in PCApplyOrTranspose_PBJacobi_Kokkos() local37 PetscScalarKokkosView Av = pckok->diag_dual.view_device(); in PCApplyOrTranspose_PBJacobi_Kokkos()93 PC_PBJacobi_Kokkos *pckok = static_cast<PC_PBJacobi_Kokkos *>(jac->spptr); in PCSetUp_PBJacobi_Kokkos() local94 PetscCall(pckok->Update(jac->diag)); in PCSetUp_PBJacobi_Kokkos()