1ae2b091fSJames Wright // SPDX-FileCopyrightText: Copyright (c) 2017-2024, HONEE contributors. 2ae2b091fSJames Wright // SPDX-License-Identifier: Apache-2.0 OR BSD-2-Clause 3a515125bSLeila Ghaffari 4a515125bSLeila Ghaffari /// @file 5a515125bSLeila Ghaffari /// Miscellaneous utility functions 6a515125bSLeila Ghaffari 7e419654dSJeremy L Thompson #include <ceed.h> 8e419654dSJeremy L Thompson #include <petscdm.h> 9926a6279SJames Wright #include <petscsf.h> 10e419654dSJeremy L Thompson #include <petscts.h> 11e419654dSJeremy L Thompson 12149fb536SJames Wright #include <navierstokes.h> 139f59f36eSJames Wright #include "../qfunctions/mass.h" 14a515125bSLeila Ghaffari 152b916ea7SJeremy L Thompson PetscErrorCode ICs_FixMultiplicity(DM dm, CeedData ceed_data, User user, Vec Q_loc, Vec Q, CeedScalar time) { 16b4c37c5cSJames Wright Ceed ceed = user->ceed; 17b2948607SJames Wright CeedVector mult_vec; 18b2948607SJames Wright PetscMemType m_mem_type; 19b2948607SJames Wright Vec Multiplicity, Multiplicity_loc; 20b2948607SJames Wright 21a515125bSLeila Ghaffari PetscFunctionBeginUser; 22b4c37c5cSJames Wright if (user->phys->ics_time_label) PetscCallCeed(ceed, CeedOperatorSetContextDouble(ceed_data->op_ics_ctx->op, user->phys->ics_time_label, &time)); 238f18bb8bSJames Wright PetscCall(ApplyCeedOperatorLocalToGlobal(NULL, Q, ceed_data->op_ics_ctx)); 24a515125bSLeila Ghaffari 25b4c37c5cSJames Wright PetscCallCeed(ceed, CeedElemRestrictionCreateVector(ceed_data->elem_restr_q, &mult_vec, NULL)); 26a515125bSLeila Ghaffari 27a515125bSLeila Ghaffari // -- Get multiplicity 28b2948607SJames Wright PetscCall(DMGetLocalVector(dm, &Multiplicity_loc)); 29a7dac1d5SJames Wright PetscCall(VecPetscToCeed(Multiplicity_loc, &m_mem_type, mult_vec)); 30b4c37c5cSJames Wright PetscCallCeed(ceed, CeedElemRestrictionGetMultiplicity(ceed_data->elem_restr_q, mult_vec)); 31a7dac1d5SJames Wright PetscCall(VecCeedToPetsc(mult_vec, m_mem_type, Multiplicity_loc)); 32a515125bSLeila Ghaffari 33b2948607SJames Wright PetscCall(DMGetGlobalVector(dm, &Multiplicity)); 34b2948607SJames Wright PetscCall(VecZeroEntries(Multiplicity)); 35b2948607SJames Wright PetscCall(DMLocalToGlobal(dm, Multiplicity_loc, ADD_VALUES, Multiplicity)); 36a515125bSLeila Ghaffari 37a515125bSLeila Ghaffari // -- Fix multiplicity 38b2948607SJames Wright PetscCall(VecPointwiseDivide(Q, Q, Multiplicity)); 39b2948607SJames Wright PetscCall(VecPointwiseDivide(Q_loc, Q_loc, Multiplicity_loc)); 40a515125bSLeila Ghaffari 41b2948607SJames Wright PetscCall(DMRestoreLocalVector(dm, &Multiplicity_loc)); 42b2948607SJames Wright PetscCall(DMRestoreGlobalVector(dm, &Multiplicity)); 43b4c37c5cSJames Wright PetscCallCeed(ceed, CeedVectorDestroy(&mult_vec)); 44d949ddfcSJames Wright PetscFunctionReturn(PETSC_SUCCESS); 45a515125bSLeila Ghaffari } 46a515125bSLeila Ghaffari 47c56e8d5bSJames Wright // Record boundary values from initial condition 48c56e8d5bSJames Wright PetscErrorCode SetBCsFromICs(DM dm, Vec Q, Vec Q_loc) { 49c56e8d5bSJames Wright PetscFunctionBeginUser; 50313f2f1eSJames Wright { // Capture initial condition values in Qbc 51313f2f1eSJames Wright Vec Qbc; 52313f2f1eSJames Wright 53c56e8d5bSJames Wright PetscCall(DMGetNamedLocalVector(dm, "Qbc", &Qbc)); 54c56e8d5bSJames Wright PetscCall(VecCopy(Q_loc, Qbc)); 55c56e8d5bSJames Wright PetscCall(VecZeroEntries(Q_loc)); 56c56e8d5bSJames Wright PetscCall(DMGlobalToLocal(dm, Q, INSERT_VALUES, Q_loc)); 57c56e8d5bSJames Wright PetscCall(VecAXPY(Qbc, -1., Q_loc)); 58c56e8d5bSJames Wright PetscCall(DMRestoreNamedLocalVector(dm, "Qbc", &Qbc)); 59313f2f1eSJames Wright } 60c56e8d5bSJames Wright PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexInsertBoundaryValues_C", DMPlexInsertBoundaryValues_FromICs)); 61c56e8d5bSJames Wright 62313f2f1eSJames Wright { // Set boundary mask to zero out essential BCs 63313f2f1eSJames Wright Vec boundary_mask, ones; 64313f2f1eSJames Wright 65c56e8d5bSJames Wright PetscCall(DMGetNamedLocalVector(dm, "boundary mask", &boundary_mask)); 66313f2f1eSJames Wright PetscCall(DMGetGlobalVector(dm, &ones)); 67c56e8d5bSJames Wright PetscCall(VecZeroEntries(boundary_mask)); 68313f2f1eSJames Wright PetscCall(VecSet(ones, 1.0)); 69313f2f1eSJames Wright PetscCall(DMGlobalToLocal(dm, ones, INSERT_VALUES, boundary_mask)); 70c56e8d5bSJames Wright PetscCall(DMRestoreNamedLocalVector(dm, "boundary mask", &boundary_mask)); 71313f2f1eSJames Wright PetscCall(DMRestoreGlobalVector(dm, &ones)); 72313f2f1eSJames Wright } 73c56e8d5bSJames Wright PetscFunctionReturn(PETSC_SUCCESS); 74c56e8d5bSJames Wright } 75c56e8d5bSJames Wright 76c56e8d5bSJames Wright PetscErrorCode DMPlexInsertBoundaryValues_FromICs(DM dm, PetscBool insert_essential, Vec Q_loc, PetscReal time, Vec face_geom_FVM, Vec cell_geom_FVM, 772b916ea7SJeremy L Thompson Vec grad_FVM) { 789d437337SJames Wright Vec Qbc, boundary_mask; 79a515125bSLeila Ghaffari 8006f41313SJames Wright PetscFunctionBeginUser; 812eb7bf1fSJames Wright // Mask (zero) Strong BC entries 829d437337SJames Wright PetscCall(DMGetNamedLocalVector(dm, "boundary mask", &boundary_mask)); 839d437337SJames Wright PetscCall(VecPointwiseMult(Q_loc, Q_loc, boundary_mask)); 849d437337SJames Wright PetscCall(DMRestoreNamedLocalVector(dm, "boundary mask", &boundary_mask)); 859d437337SJames Wright 862b916ea7SJeremy L Thompson PetscCall(DMGetNamedLocalVector(dm, "Qbc", &Qbc)); 872b916ea7SJeremy L Thompson PetscCall(VecAXPY(Q_loc, 1., Qbc)); 882b916ea7SJeremy L Thompson PetscCall(DMRestoreNamedLocalVector(dm, "Qbc", &Qbc)); 89d949ddfcSJames Wright PetscFunctionReturn(PETSC_SUCCESS); 90a515125bSLeila Ghaffari } 91a515125bSLeila Ghaffari 92a515125bSLeila Ghaffari // Compare reference solution values with current test run for CI 93c56e8d5bSJames Wright PetscErrorCode RegressionTest(AppCtx app_ctx, Vec Q) { 94a515125bSLeila Ghaffari Vec Qref; 95a515125bSLeila Ghaffari PetscViewer viewer; 96a515125bSLeila Ghaffari PetscReal error, Qrefnorm; 97e7754af5SKenneth E. Jansen MPI_Comm comm = PetscObjectComm((PetscObject)Q); 98a515125bSLeila Ghaffari 9906f41313SJames Wright PetscFunctionBeginUser; 100a515125bSLeila Ghaffari // Read reference file 1012b916ea7SJeremy L Thompson PetscCall(VecDuplicate(Q, &Qref)); 1024c07ec22SJames Wright PetscCheck(strcmp(app_ctx->test_file_path, "") != 0, comm, PETSC_ERR_FILE_READ, "File for regression test not given"); 103e7754af5SKenneth E. Jansen PetscCall(PetscViewerBinaryOpen(comm, app_ctx->test_file_path, FILE_MODE_READ, &viewer)); 104360b37c9SJames Wright PetscCall(HoneeLoadBinaryVec(viewer, Qref, NULL, NULL)); 105a515125bSLeila Ghaffari 106a515125bSLeila Ghaffari // Compute error with respect to reference solution 1072b916ea7SJeremy L Thompson PetscCall(VecAXPY(Q, -1.0, Qref)); 1082b916ea7SJeremy L Thompson PetscCall(VecNorm(Qref, NORM_MAX, &Qrefnorm)); 1092b916ea7SJeremy L Thompson PetscCall(VecScale(Q, 1. / Qrefnorm)); 1102b916ea7SJeremy L Thompson PetscCall(VecNorm(Q, NORM_MAX, &error)); 111a515125bSLeila Ghaffari 112a515125bSLeila Ghaffari // Check error 113a515125bSLeila Ghaffari if (error > app_ctx->test_tol) { 1142b916ea7SJeremy L Thompson PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Test failed with error norm %g\n", (double)error)); 115a515125bSLeila Ghaffari } 116a515125bSLeila Ghaffari 117a515125bSLeila Ghaffari // Cleanup 1182b916ea7SJeremy L Thompson PetscCall(PetscViewerDestroy(&viewer)); 1192b916ea7SJeremy L Thompson PetscCall(VecDestroy(&Qref)); 120d949ddfcSJames Wright PetscFunctionReturn(PETSC_SUCCESS); 121a515125bSLeila Ghaffari } 122a515125bSLeila Ghaffari 123a515125bSLeila Ghaffari // Get error for problems with exact solutions 124c56e8d5bSJames Wright PetscErrorCode PrintError(CeedData ceed_data, DM dm, User user, Vec Q, PetscScalar final_time) { 125a515125bSLeila Ghaffari PetscInt loc_nodes; 126a515125bSLeila Ghaffari Vec Q_exact, Q_exact_loc; 127a515125bSLeila Ghaffari PetscReal rel_error, norm_error, norm_exact; 128a515125bSLeila Ghaffari 12906f41313SJames Wright PetscFunctionBeginUser; 130a515125bSLeila Ghaffari // Get exact solution at final time 131b2948607SJames Wright PetscCall(DMGetGlobalVector(dm, &Q_exact)); 1322b916ea7SJeremy L Thompson PetscCall(DMGetLocalVector(dm, &Q_exact_loc)); 1332b916ea7SJeremy L Thompson PetscCall(VecGetSize(Q_exact_loc, &loc_nodes)); 1342b916ea7SJeremy L Thompson PetscCall(ICs_FixMultiplicity(dm, ceed_data, user, Q_exact_loc, Q_exact, final_time)); 135a515125bSLeila Ghaffari 136a515125bSLeila Ghaffari // Get |exact solution - obtained solution| 1372b916ea7SJeremy L Thompson PetscCall(VecNorm(Q_exact, NORM_1, &norm_exact)); 1382b916ea7SJeremy L Thompson PetscCall(VecAXPY(Q, -1.0, Q_exact)); 1392b916ea7SJeremy L Thompson PetscCall(VecNorm(Q, NORM_1, &norm_error)); 140a515125bSLeila Ghaffari 141a515125bSLeila Ghaffari rel_error = norm_error / norm_exact; 1422b916ea7SJeremy L Thompson PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Relative Error: %g\n", (double)rel_error)); 1432b916ea7SJeremy L Thompson PetscCall(DMRestoreLocalVector(dm, &Q_exact_loc)); 144b2948607SJames Wright PetscCall(DMRestoreGlobalVector(dm, &Q_exact)); 145d949ddfcSJames Wright PetscFunctionReturn(PETSC_SUCCESS); 146a515125bSLeila Ghaffari } 147a515125bSLeila Ghaffari 148a515125bSLeila Ghaffari // Post-processing 149991aef52SJames Wright PetscErrorCode PostProcess(TS ts, CeedData ceed_data, DM dm, ProblemData problem, User user, Vec Q, PetscScalar final_time) { 150a515125bSLeila Ghaffari PetscInt steps; 151f0784ed3SJed Brown TSConvergedReason reason; 152a515125bSLeila Ghaffari 15306f41313SJames Wright PetscFunctionBeginUser; 154a515125bSLeila Ghaffari // Print relative error 15558ce1233SJames Wright if (problem->compute_exact_solution_error && user->app_ctx->test_type == TESTTYPE_NONE) { 156c56e8d5bSJames Wright PetscCall(PrintError(ceed_data, dm, user, Q, final_time)); 157a515125bSLeila Ghaffari } 158a515125bSLeila Ghaffari 159a515125bSLeila Ghaffari // Print final time and number of steps 1602b916ea7SJeremy L Thompson PetscCall(TSGetStepNumber(ts, &steps)); 161f0784ed3SJed Brown PetscCall(TSGetConvergedReason(ts, &reason)); 1620e1e9333SJames Wright if (user->app_ctx->test_type == TESTTYPE_NONE) { 163f0784ed3SJed Brown PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Time integrator %s on time step %" PetscInt_FMT " with final time %g\n", TSConvergedReasons[reason], 164f0784ed3SJed Brown steps, (double)final_time)); 165a515125bSLeila Ghaffari } 166a515125bSLeila Ghaffari 167a515125bSLeila Ghaffari // Output numerical values from command line 1682b916ea7SJeremy L Thompson PetscCall(VecViewFromOptions(Q, NULL, "-vec_view")); 169a515125bSLeila Ghaffari 170a515125bSLeila Ghaffari // Compare reference solution values with current test run for CI 1710e1e9333SJames Wright if (user->app_ctx->test_type == TESTTYPE_SOLVER) { 172c56e8d5bSJames Wright PetscCall(RegressionTest(user->app_ctx, Q)); 173a515125bSLeila Ghaffari } 174d949ddfcSJames Wright PetscFunctionReturn(PETSC_SUCCESS); 175a515125bSLeila Ghaffari } 176a515125bSLeila Ghaffari 17715a3537eSJed Brown // Free a plain data context that was allocated using PETSc; returning libCEED error codes 17815a3537eSJed Brown int FreeContextPetsc(void *data) { 1792b916ea7SJeremy L Thompson if (PetscFree(data)) return CeedError(NULL, CEED_ERROR_ACCESS, "PetscFree failed"); 18015a3537eSJed Brown return CEED_ERROR_SUCCESS; 18115a3537eSJed Brown } 1829f59f36eSJames Wright 1834cb09fddSJames Wright // @brief Return mass qfunction specification for number of components N 1849f59f36eSJames Wright PetscErrorCode CreateMassQFunction(Ceed ceed, CeedInt N, CeedInt q_data_size, CeedQFunction *qf) { 1859f59f36eSJames Wright PetscFunctionBeginUser; 1869f59f36eSJames Wright switch (N) { 1879f59f36eSJames Wright case 1: 188b4c37c5cSJames Wright PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 1, Mass_1, Mass_1_loc, qf)); 1899f59f36eSJames Wright break; 19040b78511SJames Wright case 2: 19140b78511SJames Wright PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 2, Mass_2, Mass_2_loc, qf)); 19240b78511SJames Wright break; 19340b78511SJames Wright case 3: 19440b78511SJames Wright PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 3, Mass_3, Mass_3_loc, qf)); 19540b78511SJames Wright break; 1968c85b835SJames Wright case 4: 1978c85b835SJames Wright PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 4, Mass_4, Mass_4_loc, qf)); 1988c85b835SJames Wright break; 1999f59f36eSJames Wright case 5: 200b4c37c5cSJames Wright PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 1, Mass_5, Mass_5_loc, qf)); 2019f59f36eSJames Wright break; 202c38c977aSJames Wright case 7: 203b4c37c5cSJames Wright PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 1, Mass_7, Mass_7_loc, qf)); 204c38c977aSJames Wright break; 2059f59f36eSJames Wright case 9: 206b4c37c5cSJames Wright PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 1, Mass_9, Mass_9_loc, qf)); 2079f59f36eSJames Wright break; 2088c85b835SJames Wright case 12: 2098c85b835SJames Wright PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 1, Mass_12, Mass_12_loc, qf)); 2108c85b835SJames Wright break; 2119f59f36eSJames Wright case 22: 212b4c37c5cSJames Wright PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 1, Mass_22, Mass_22_loc, qf)); 2139f59f36eSJames Wright break; 2149f59f36eSJames Wright default: 21540b78511SJames Wright SETERRQ(PETSC_COMM_WORLD, PETSC_ERR_SUP, 21640b78511SJames Wright "Mass qfunction of size %" CeedInt_FMT " does not exist.\nA new mass qfunction can be easily added; see source code for pattern.", N); 2179f59f36eSJames Wright } 2189f59f36eSJames Wright 219b4c37c5cSJames Wright PetscCallCeed(ceed, CeedQFunctionAddInput(*qf, "u", N, CEED_EVAL_INTERP)); 220b4c37c5cSJames Wright PetscCallCeed(ceed, CeedQFunctionAddInput(*qf, "qdata", q_data_size, CEED_EVAL_NONE)); 221b4c37c5cSJames Wright PetscCallCeed(ceed, CeedQFunctionAddOutput(*qf, "v", N, CEED_EVAL_INTERP)); 2223170c09fSJames Wright PetscCallCeed(ceed, CeedQFunctionSetUserFlopsEstimate(*qf, N)); 223d949ddfcSJames Wright PetscFunctionReturn(PETSC_SUCCESS); 2249f59f36eSJames Wright } 225e5e81594SJames Wright 226457a5831SJames Wright PetscErrorCode NodalProjectionDataDestroy(NodalProjectionData context) { 227457a5831SJames Wright PetscFunctionBeginUser; 228d949ddfcSJames Wright if (context == NULL) PetscFunctionReturn(PETSC_SUCCESS); 229457a5831SJames Wright 230457a5831SJames Wright PetscCall(DMDestroy(&context->dm)); 231457a5831SJames Wright PetscCall(KSPDestroy(&context->ksp)); 232457a5831SJames Wright 233457a5831SJames Wright PetscCall(OperatorApplyContextDestroy(context->l2_rhs_ctx)); 234457a5831SJames Wright 235457a5831SJames Wright PetscCall(PetscFree(context)); 236d949ddfcSJames Wright PetscFunctionReturn(PETSC_SUCCESS); 237457a5831SJames Wright } 238457a5831SJames Wright 23959572072SJames Wright /** 24059572072SJames Wright @brief Sets the value of an option if it is not already set 24159572072SJames Wright 24259572072SJames Wright @param[in,out] options `PetscOptions` database, or `NULL` for default global database 24359572072SJames Wright @param[in] name Name of the option (with `-` prepended to it) 24459572072SJames Wright @param[in] value Value to set the option to 24559572072SJames Wright **/ 24659572072SJames Wright PetscErrorCode HoneeOptionsSetValueDefault(PetscOptions options, const char name[], const char value[]) { 24759572072SJames Wright PetscBool has_option; 24859572072SJames Wright 24959572072SJames Wright PetscFunctionBeginUser; 25059572072SJames Wright PetscCall(PetscOptionsHasName(options, NULL, name, &has_option)); 25159572072SJames Wright if (!has_option) PetscCall(PetscOptionsSetValue(options, name, value)); 25259572072SJames Wright PetscFunctionReturn(PETSC_SUCCESS); 25359572072SJames Wright } 25459572072SJames Wright 255926a6279SJames Wright // Print information about the given simulation run 2563678fae3SJames Wright PetscErrorCode PrintRunInfo(User user, Physics phys_ctx, ProblemData problem, TS ts) { 257b4c37c5cSJames Wright Ceed ceed = user->ceed; 2583678fae3SJames Wright MPI_Comm comm = PetscObjectComm((PetscObject)ts); 2593678fae3SJames Wright 260926a6279SJames Wright PetscFunctionBeginUser; 261926a6279SJames Wright // Header and rank 262926a6279SJames Wright char host_name[PETSC_MAX_PATH_LEN]; 263926a6279SJames Wright PetscMPIInt rank, comm_size; 264926a6279SJames Wright PetscCall(PetscGetHostName(host_name, sizeof host_name)); 265926a6279SJames Wright PetscCallMPI(MPI_Comm_rank(comm, &rank)); 266926a6279SJames Wright PetscCallMPI(MPI_Comm_size(comm, &comm_size)); 267926a6279SJames Wright PetscCall(PetscPrintf(comm, 268926a6279SJames Wright "\n-- Navier-Stokes solver - libCEED + PETSc --\n" 269926a6279SJames Wright " MPI:\n" 270926a6279SJames Wright " Host Name : %s\n" 271926a6279SJames Wright " Total ranks : %d\n", 272926a6279SJames Wright host_name, comm_size)); 273926a6279SJames Wright 274926a6279SJames Wright // Problem specific info 2752d49c0afSJames Wright PetscCall(problem->print_info(user, problem, user->app_ctx)); 276926a6279SJames Wright 277926a6279SJames Wright // libCEED 278926a6279SJames Wright const char *used_resource; 279926a6279SJames Wright CeedMemType mem_type_backend; 280b4c37c5cSJames Wright PetscCallCeed(ceed, CeedGetResource(user->ceed, &used_resource)); 281b4c37c5cSJames Wright PetscCallCeed(ceed, CeedGetPreferredMemType(user->ceed, &mem_type_backend)); 282926a6279SJames Wright PetscCall(PetscPrintf(comm, 283926a6279SJames Wright " libCEED:\n" 284926a6279SJames Wright " libCEED Backend : %s\n" 285926a6279SJames Wright " libCEED Backend MemType : %s\n", 286926a6279SJames Wright used_resource, CeedMemTypes[mem_type_backend])); 287926a6279SJames Wright // PETSc 28866d54740SJames Wright { 2893678fae3SJames Wright VecType vec_type; 290926a6279SJames Wright char box_faces_str[PETSC_MAX_PATH_LEN] = "3,3,3"; 29166d54740SJames Wright PetscInt dim; 29266d54740SJames Wright 29366d54740SJames Wright PetscCall(DMGetDimension(user->dm, &dim)); 29466d54740SJames Wright if (dim == 2) box_faces_str[3] = '\0'; 295926a6279SJames Wright PetscCall(PetscOptionsGetString(NULL, NULL, "-dm_plex_box_faces", box_faces_str, sizeof(box_faces_str), NULL)); 296926a6279SJames Wright PetscCall(DMGetVecType(user->dm, &vec_type)); 297926a6279SJames Wright PetscCall(PetscPrintf(comm, 298926a6279SJames Wright " PETSc:\n" 299926a6279SJames Wright " Box Faces : %s\n" 300926a6279SJames Wright " DM VecType : %s\n" 301926a6279SJames Wright " Time Stepping Scheme : %s\n", 3023678fae3SJames Wright box_faces_str, vec_type, phys_ctx->implicit ? "implicit" : "explicit")); 30366d54740SJames Wright } 3043678fae3SJames Wright { 3053678fae3SJames Wright char pmat_type_str[PETSC_MAX_PATH_LEN]; 3063678fae3SJames Wright MatType amat_type, pmat_type; 3073678fae3SJames Wright Mat Amat, Pmat; 3083678fae3SJames Wright TSIJacobianFn *ijacob_function; 3093678fae3SJames Wright 3103678fae3SJames Wright PetscCall(TSGetIJacobian(ts, &Amat, &Pmat, &ijacob_function, NULL)); 3113678fae3SJames Wright PetscCall(MatGetType(Amat, &amat_type)); 3123678fae3SJames Wright PetscCall(MatGetType(Pmat, &pmat_type)); 3133678fae3SJames Wright 3143678fae3SJames Wright PetscCall(PetscStrncpy(pmat_type_str, pmat_type, sizeof(pmat_type_str))); 3153678fae3SJames Wright if (!strcmp(pmat_type, MATCEED)) { 3163678fae3SJames Wright MatType pmat_coo_type; 3173678fae3SJames Wright char pmat_coo_type_str[PETSC_MAX_PATH_LEN]; 3183678fae3SJames Wright 3193678fae3SJames Wright PetscCall(MatCeedGetCOOMatType(Pmat, &pmat_coo_type)); 3203678fae3SJames Wright PetscCall(PetscSNPrintf(pmat_coo_type_str, sizeof(pmat_coo_type_str), " (COO MatType: %s)", pmat_coo_type)); 3213678fae3SJames Wright PetscCall(PetscStrlcat(pmat_type_str, pmat_coo_type_str, sizeof(pmat_type_str))); 3223678fae3SJames Wright } 3233678fae3SJames Wright if (ijacob_function) { 3243678fae3SJames Wright PetscCall(PetscPrintf(comm, 3253678fae3SJames Wright " IJacobian A MatType : %s\n" 3263678fae3SJames Wright " IJacobian P MatType : %s\n", 3273678fae3SJames Wright amat_type, pmat_type_str)); 3283678fae3SJames Wright } 3293678fae3SJames Wright } 330926a6279SJames Wright if (user->app_ctx->cont_steps) { 331926a6279SJames Wright PetscCall(PetscPrintf(comm, 332926a6279SJames Wright " Continue:\n" 333926a6279SJames Wright " Filename: : %s\n" 334926a6279SJames Wright " Step: : %" PetscInt_FMT "\n" 335926a6279SJames Wright " Time: : %g\n", 336926a6279SJames Wright user->app_ctx->cont_file, user->app_ctx->cont_steps, user->app_ctx->cont_time)); 337926a6279SJames Wright } 338926a6279SJames Wright // Mesh 339926a6279SJames Wright const PetscInt num_comp_q = 5; 340926a6279SJames Wright PetscInt glob_dofs, owned_dofs, local_dofs; 341926a6279SJames Wright const CeedInt num_P = user->app_ctx->degree + 1, num_Q = num_P + user->app_ctx->q_extra; 342926a6279SJames Wright PetscCall(DMGetGlobalVectorInfo(user->dm, &owned_dofs, &glob_dofs, NULL)); 343926a6279SJames Wright PetscCall(DMGetLocalVectorInfo(user->dm, &local_dofs, NULL, NULL)); 344926a6279SJames Wright PetscCall(PetscPrintf(comm, 345926a6279SJames Wright " Mesh:\n" 346926a6279SJames Wright " Number of 1D Basis Nodes (P) : %" CeedInt_FMT "\n" 347926a6279SJames Wright " Number of 1D Quadrature Points (Q) : %" CeedInt_FMT "\n" 348926a6279SJames Wright " Global DoFs : %" PetscInt_FMT "\n" 349926a6279SJames Wright " DoFs per node : %" PetscInt_FMT "\n" 350dfeb939dSJames Wright " Global %" PetscInt_FMT "-DoF nodes : %" PetscInt_FMT "\n", 351dfeb939dSJames Wright num_P, num_Q, glob_dofs, num_comp_q, num_comp_q, glob_dofs / num_comp_q)); 352926a6279SJames Wright // -- Get Partition Statistics 353926a6279SJames Wright PetscCall(PetscPrintf(comm, " Partition: (min,max,median,max/median)\n")); 354926a6279SJames Wright { 355926a6279SJames Wright PetscInt *gather_buffer = NULL; 356dfeb939dSJames Wright PetscInt part_owned_dofs[3], part_local_dofs[3], part_boundary_dofs[3], part_neighbors[3]; 357926a6279SJames Wright PetscInt median_index = comm_size % 2 ? comm_size / 2 : comm_size / 2 - 1; 358926a6279SJames Wright if (!rank) PetscCall(PetscMalloc1(comm_size, &gather_buffer)); 359926a6279SJames Wright 360dfeb939dSJames Wright PetscCallMPI(MPI_Gather(&owned_dofs, 1, MPIU_INT, gather_buffer, 1, MPIU_INT, 0, comm)); 361926a6279SJames Wright if (!rank) { 362926a6279SJames Wright PetscCall(PetscSortInt(comm_size, gather_buffer)); 363926a6279SJames Wright part_owned_dofs[0] = gather_buffer[0]; // min 364926a6279SJames Wright part_owned_dofs[1] = gather_buffer[comm_size - 1]; // max 365926a6279SJames Wright part_owned_dofs[2] = gather_buffer[median_index]; // median 366926a6279SJames Wright PetscReal part_owned_dof_ratio = (PetscReal)part_owned_dofs[1] / (PetscReal)part_owned_dofs[2]; 367dfeb939dSJames Wright PetscCall(PetscPrintf( 368dfeb939dSJames Wright comm, " Global Vector %" PetscInt_FMT "-DoF nodes : %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT ", %f\n", num_comp_q, 369926a6279SJames Wright part_owned_dofs[0] / num_comp_q, part_owned_dofs[1] / num_comp_q, part_owned_dofs[2] / num_comp_q, part_owned_dof_ratio)); 370926a6279SJames Wright } 371926a6279SJames Wright 372dfeb939dSJames Wright PetscCallMPI(MPI_Gather(&local_dofs, 1, MPIU_INT, gather_buffer, 1, MPIU_INT, 0, comm)); 373926a6279SJames Wright if (!rank) { 374926a6279SJames Wright PetscCall(PetscSortInt(comm_size, gather_buffer)); 375926a6279SJames Wright part_local_dofs[0] = gather_buffer[0]; // min 376926a6279SJames Wright part_local_dofs[1] = gather_buffer[comm_size - 1]; // max 377926a6279SJames Wright part_local_dofs[2] = gather_buffer[median_index]; // median 378926a6279SJames Wright PetscReal part_local_dof_ratio = (PetscReal)part_local_dofs[1] / (PetscReal)part_local_dofs[2]; 379dfeb939dSJames Wright PetscCall(PetscPrintf( 380dfeb939dSJames Wright comm, " Local Vector %" PetscInt_FMT "-DoF nodes : %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT ", %f\n", num_comp_q, 381926a6279SJames Wright part_local_dofs[0] / num_comp_q, part_local_dofs[1] / num_comp_q, part_local_dofs[2] / num_comp_q, part_local_dof_ratio)); 382926a6279SJames Wright } 383926a6279SJames Wright 38445abf96eSJames Wright if (comm_size != 1) { 385dfeb939dSJames Wright PetscInt num_remote_roots_total = 0, num_remote_leaves_total = 0, num_ghost_interface_ranks = 0, num_owned_interface_ranks = 0; 386926a6279SJames Wright { 387926a6279SJames Wright PetscSF sf; 388*3bbbcc04SJames Wright PetscMPIInt nrranks, niranks; 389dfeb939dSJames Wright const PetscInt *roffset, *rmine, *rremote, *ioffset, *irootloc; 390dfeb939dSJames Wright const PetscMPIInt *rranks, *iranks; 391926a6279SJames Wright PetscCall(DMGetSectionSF(user->dm, &sf)); 392926a6279SJames Wright PetscCall(PetscSFGetRootRanks(sf, &nrranks, &rranks, &roffset, &rmine, &rremote)); 393dfeb939dSJames Wright PetscCall(PetscSFGetLeafRanks(sf, &niranks, &iranks, &ioffset, &irootloc)); 394926a6279SJames Wright for (PetscInt i = 0; i < nrranks; i++) { 395926a6279SJames Wright if (rranks[i] == rank) continue; // Ignore same-part global->local transfers 396926a6279SJames Wright num_remote_roots_total += roffset[i + 1] - roffset[i]; 397dfeb939dSJames Wright num_ghost_interface_ranks++; 398dfeb939dSJames Wright } 399dfeb939dSJames Wright for (PetscInt i = 0; i < niranks; i++) { 400dfeb939dSJames Wright if (iranks[i] == rank) continue; 401dfeb939dSJames Wright num_remote_leaves_total += ioffset[i + 1] - ioffset[i]; 402dfeb939dSJames Wright num_owned_interface_ranks++; 403926a6279SJames Wright } 404926a6279SJames Wright } 405dfeb939dSJames Wright PetscCallMPI(MPI_Gather(&num_remote_roots_total, 1, MPIU_INT, gather_buffer, 1, MPIU_INT, 0, comm)); 406926a6279SJames Wright if (!rank) { 407926a6279SJames Wright PetscCall(PetscSortInt(comm_size, gather_buffer)); 408dfeb939dSJames Wright part_boundary_dofs[0] = gather_buffer[0]; // min 409dfeb939dSJames Wright part_boundary_dofs[1] = gather_buffer[comm_size - 1]; // max 410dfeb939dSJames Wright part_boundary_dofs[2] = gather_buffer[median_index]; // median 411dfeb939dSJames Wright PetscReal part_shared_dof_ratio = (PetscReal)part_boundary_dofs[1] / (PetscReal)part_boundary_dofs[2]; 412dfeb939dSJames Wright PetscCall(PetscPrintf( 41345abf96eSJames Wright comm, " Ghost Interface %" PetscInt_FMT "-DoF nodes : %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT ", %f\n", 41445abf96eSJames Wright num_comp_q, part_boundary_dofs[0] / num_comp_q, part_boundary_dofs[1] / num_comp_q, part_boundary_dofs[2] / num_comp_q, 41545abf96eSJames Wright part_shared_dof_ratio)); 416dfeb939dSJames Wright } 417dfeb939dSJames Wright 418dfeb939dSJames Wright PetscCallMPI(MPI_Gather(&num_ghost_interface_ranks, 1, MPIU_INT, gather_buffer, 1, MPIU_INT, 0, comm)); 419dfeb939dSJames Wright if (!rank) { 420dfeb939dSJames Wright PetscCall(PetscSortInt(comm_size, gather_buffer)); 421dfeb939dSJames Wright part_neighbors[0] = gather_buffer[0]; // min 422dfeb939dSJames Wright part_neighbors[1] = gather_buffer[comm_size - 1]; // max 423dfeb939dSJames Wright part_neighbors[2] = gather_buffer[median_index]; // median 424dfeb939dSJames Wright PetscReal part_neighbors_ratio = (PetscReal)part_neighbors[1] / (PetscReal)part_neighbors[2]; 425dfeb939dSJames Wright PetscCall(PetscPrintf(comm, " Ghost Interface Ranks : %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT ", %f\n", 426dfeb939dSJames Wright part_neighbors[0], part_neighbors[1], part_neighbors[2], part_neighbors_ratio)); 427dfeb939dSJames Wright } 428dfeb939dSJames Wright 429dfeb939dSJames Wright PetscCallMPI(MPI_Gather(&num_remote_leaves_total, 1, MPIU_INT, gather_buffer, 1, MPIU_INT, 0, comm)); 430dfeb939dSJames Wright if (!rank) { 431dfeb939dSJames Wright PetscCall(PetscSortInt(comm_size, gather_buffer)); 432dfeb939dSJames Wright part_boundary_dofs[0] = gather_buffer[0]; // min 433dfeb939dSJames Wright part_boundary_dofs[1] = gather_buffer[comm_size - 1]; // max 434dfeb939dSJames Wright part_boundary_dofs[2] = gather_buffer[median_index]; // median 435dfeb939dSJames Wright PetscReal part_shared_dof_ratio = (PetscReal)part_boundary_dofs[1] / (PetscReal)part_boundary_dofs[2]; 436dfeb939dSJames Wright PetscCall(PetscPrintf( 43745abf96eSJames Wright comm, " Owned Interface %" PetscInt_FMT "-DoF nodes : %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT ", %f\n", 43845abf96eSJames Wright num_comp_q, part_boundary_dofs[0] / num_comp_q, part_boundary_dofs[1] / num_comp_q, part_boundary_dofs[2] / num_comp_q, 43945abf96eSJames Wright part_shared_dof_ratio)); 440dfeb939dSJames Wright } 441dfeb939dSJames Wright 442dfeb939dSJames Wright PetscCallMPI(MPI_Gather(&num_owned_interface_ranks, 1, MPIU_INT, gather_buffer, 1, MPIU_INT, 0, comm)); 443dfeb939dSJames Wright if (!rank) { 444dfeb939dSJames Wright PetscCall(PetscSortInt(comm_size, gather_buffer)); 445dfeb939dSJames Wright part_neighbors[0] = gather_buffer[0]; // min 446dfeb939dSJames Wright part_neighbors[1] = gather_buffer[comm_size - 1]; // max 447dfeb939dSJames Wright part_neighbors[2] = gather_buffer[median_index]; // median 448dfeb939dSJames Wright PetscReal part_neighbors_ratio = (PetscReal)part_neighbors[1] / (PetscReal)part_neighbors[2]; 449dfeb939dSJames Wright PetscCall(PetscPrintf(comm, " Owned Interface Ranks : %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT ", %f\n", 450dfeb939dSJames Wright part_neighbors[0], part_neighbors[1], part_neighbors[2], part_neighbors_ratio)); 451926a6279SJames Wright } 45245abf96eSJames Wright } 453926a6279SJames Wright 454926a6279SJames Wright if (!rank) PetscCall(PetscFree(gather_buffer)); 455926a6279SJames Wright } 456926a6279SJames Wright PetscFunctionReturn(PETSC_SUCCESS); 457926a6279SJames Wright } 458