Home
last modified time | relevance | path

Searched refs:dm_Nf (Results 1 – 3 of 3) sorted by relevance

/petsc/src/ksp/pc/impls/bjacobi/bjkokkos/
H A Dbjkokkos.kokkos.cxx676 for (PetscInt f = 0, idx = head; f < jac->dm_Nf[dmIdx]; f++, idx++, s++) { in PCApply_BJKOKKOS()
679 …jac->max_nits += h_metadata[idx + bid * jac->dm_Nf[dmIdx]].its; // report total number of iteratio… in PCApply_BJKOKKOS()
680 if (h_metadata[idx + bid * jac->dm_Nf[dmIdx]].its > max_nnit) { in PCApply_BJKOKKOS()
681 max_nnit = h_metadata[idx + bid * jac->dm_Nf[dmIdx]].its; in PCApply_BJKOKKOS()
685 if (h_metadata[idx + bid * jac->dm_Nf[dmIdx]].its > max_nnit) { in PCApply_BJKOKKOS()
686 jac->max_nits = max_nnit = h_metadata[idx + bid * jac->dm_Nf[dmIdx]].its; in PCApply_BJKOKKOS()
693 …ObjectComm((PetscObject)A), "%3" PetscInt_FMT " ", h_metadata[idx + bid * jac->dm_Nf[dmIdx]].its)); in PCApply_BJKOKKOS()
699 head += batch_sz * jac->dm_Nf[dmIdx]; in PCApply_BJKOKKOS()
919 PetscCall(PetscMalloc(sizeof(*jac->dm_Nf) * nDMs, &jac->dm_Nf)); in PCSetUp_BJKOKKOS()
937 jac->dm_Nf[ii] = Nf; in PCSetUp_BJKOKKOS()
[all …]
H A Dbjkokkoskernels.kokkos.cxx275 for (PetscInt f = 0, idx = head; f < jac->dm_Nf[dmIdx]; f++, s++, idx++) { in PCApply_BJKOKKOSKERNELS()
278 …etscObjectComm((PetscObject)A), "%3D ", handle.get_iteration_host(idx + bid * jac->dm_Nf[dmIdx]))); in PCApply_BJKOKKOSKERNELS()
283 if ((ii = handle.get_iteration_host(idx + bid * jac->dm_Nf[dmIdx])) > count) count = ii; in PCApply_BJKOKKOSKERNELS()
288 head += batch_sz * jac->dm_Nf[dmIdx]; in PCApply_BJKOKKOSKERNELS()
/petsc/include/petsc/private/
H A Dpcbjkokkosimpl.h50 PetscInt *dm_Nf; // Number of fields in each DM member