Lines Matching refs:h_metadata

658       auto h_metadata = Kokkos::create_mirror(Kokkos::HostSpace::memory_space(), d_metadata);
659       Kokkos::deep_copy(h_metadata, d_metadata);
679               jac->max_nits += h_metadata[idx + bid * jac->dm_Nf[dmIdx]].its; // report total number of iterations with high verbose
680               if (h_metadata[idx + bid * jac->dm_Nf[dmIdx]].its > max_nnit) {
681                 max_nnit = h_metadata[idx + bid * jac->dm_Nf[dmIdx]].its;
685               if (h_metadata[idx + bid * jac->dm_Nf[dmIdx]].its > max_nnit) {
686                 jac->max_nits = max_nnit = h_metadata[idx + bid * jac->dm_Nf[dmIdx]].its;
693             for (int bid = 0; bid < batch_sz; bid++) PetscCall(PetscPrintf(PetscObjectComm((PetscObject)A), "%3" PetscInt_FMT " ", h_metadata[idx + bid * jac->dm_Nf[dmIdx]].its));
707             if (h_metadata[blkID].its > max_nnit) {
708               jac->max_nits = max_nnit = h_metadata[blkID].its;
721             PetscCall(PetscPrintf(PETSC_COMM_SELF, "[%d] Linear solve converged due to %s iterations %d (max), on block %" PetscInt_FMT ", species %" PetscInt_FMT " (max)\n", out[1], KSPConvergedReasons[h_metadata[mbid].reason], out[0], mbid % batch_sz, mbid / batch_sz));
722           else PetscCall(PetscPrintf(PETSC_COMM_SELF, "[%d] Linear solve converged due to %s iterations %d (max), on block %" PetscInt_FMT "\n", out[1], KSPConvergedReasons[h_metadata[mbid].reason], out[0], mbid));
727         PetscCall(PetscLogGpuFlops((PetscLogDouble)h_metadata[blkID].flops));
728         PetscCheck(h_metadata[blkID].reason >= 0 || !jac->ksp->errorifnotconverged, PetscObjectComm((PetscObject)pc), PETSC_ERR_CONV_FAILED, "ERROR reason=%s, its=%" PetscInt_FMT ". species %" PetscInt_FMT ", batch %" PetscInt_FMT,
729                    KSPConvergedReasons[h_metadata[blkID].reason], h_metadata[blkID].its, blkID / batch_sz, blkID % batch_sz);
742             if (h_metadata[blkID].its > jac->max_nits) jac->max_nits = h_metadata[blkID].its;