Home
last modified time | relevance | path

Searched refs:dmIdx (Results 1 – 2 of 2) sorted by relevance

/petsc/src/ksp/pc/impls/bjacobi/bjkokkos/
H A Dbjkokkoskernels.kokkos.cxx274 for (PetscInt dmIdx = 0, s = 0, head = 0; dmIdx < jac->num_dms; dmIdx += batch_sz) { in PCApply_BJKOKKOSKERNELS() local
275 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()
H A Dbjkokkos.kokkos.cxx675 for (PetscInt dmIdx = 0, head = 0, s = 0; dmIdx < jac->num_dms; dmIdx += batch_sz) { in PCApply_BJKOKKOS() local
676 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()