1c4762a1bSJed Brown static char help[] = "Performance tests for DMPlex query operations\n\n"; 2c4762a1bSJed Brown 3c4762a1bSJed Brown #include <petscdmplex.h> 4c4762a1bSJed Brown 5c4762a1bSJed Brown typedef struct { 6c4762a1bSJed Brown PetscInt dim; /* The topological mesh dimension */ 7c4762a1bSJed Brown PetscBool cellSimplex; /* Flag for simplices */ 8c4762a1bSJed Brown PetscBool spectral; /* Flag for spectral element layout */ 9c4762a1bSJed Brown PetscBool interpolate; /* Flag for mesh interpolation */ 10c4762a1bSJed Brown PetscReal refinementLimit; /* Maximum volume of a refined cell */ 11c4762a1bSJed Brown PetscInt numFields; /* The number of section fields */ 12c4762a1bSJed Brown PetscInt *numComponents; /* The number of field components */ 13c4762a1bSJed Brown PetscInt *numDof; /* The dof signature for the section */ 14c4762a1bSJed Brown PetscBool reuseArray; /* Pass in user allocated array to VecGetClosure() */ 15c4762a1bSJed Brown /* Test data */ 16c4762a1bSJed Brown PetscBool errors; /* Treat failures as errors */ 17c4762a1bSJed Brown PetscInt iterations; /* The number of iterations for a query */ 18c4762a1bSJed Brown PetscReal maxConeTime; /* Max time per run for DMPlexGetCone() */ 19c4762a1bSJed Brown PetscReal maxClosureTime; /* Max time per run for DMPlexGetTransitiveClosure() */ 20c4762a1bSJed Brown PetscReal maxVecClosureTime; /* Max time per run for DMPlexVecGetClosure() */ 21c4762a1bSJed Brown PetscBool printTimes; /* Print total times, do not check limits */ 22c4762a1bSJed Brown } AppCtx; 23c4762a1bSJed Brown 24d71ae5a4SJacob Faibussowitsch static PetscErrorCode ProcessOptions(AppCtx *options) 25d71ae5a4SJacob Faibussowitsch { 26c4762a1bSJed Brown PetscInt len; 27c4762a1bSJed Brown PetscBool flg; 28c4762a1bSJed Brown 29c4762a1bSJed Brown PetscFunctionBegin; 30c4762a1bSJed Brown options->dim = 2; 31c4762a1bSJed Brown options->cellSimplex = PETSC_TRUE; 32c4762a1bSJed Brown options->spectral = PETSC_FALSE; 33c4762a1bSJed Brown options->interpolate = PETSC_FALSE; 34c4762a1bSJed Brown options->refinementLimit = 0.0; 35c4762a1bSJed Brown options->numFields = 0; 36c4762a1bSJed Brown options->numComponents = NULL; 37c4762a1bSJed Brown options->numDof = NULL; 38c4762a1bSJed Brown options->reuseArray = PETSC_FALSE; 39c4762a1bSJed Brown options->errors = PETSC_FALSE; 40c4762a1bSJed Brown options->iterations = 1; 41c4762a1bSJed Brown options->maxConeTime = 0.0; 42c4762a1bSJed Brown options->maxClosureTime = 0.0; 43c4762a1bSJed Brown options->maxVecClosureTime = 0.0; 44c4762a1bSJed Brown options->printTimes = PETSC_FALSE; 45c4762a1bSJed Brown 46d0609cedSBarry Smith PetscOptionsBegin(PETSC_COMM_SELF, "", "Meshing Problem Options", "DMPLEX"); 479566063dSJacob Faibussowitsch PetscCall(PetscOptionsRangeInt("-dim", "The topological mesh dimension", "ex9.c", options->dim, &options->dim, NULL, 1, 3)); 489566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-cellSimplex", "Flag for simplices", "ex9.c", options->cellSimplex, &options->cellSimplex, NULL)); 499566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-spectral", "Flag for spectral element layout", "ex9.c", options->spectral, &options->spectral, NULL)); 509566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-interpolate", "Flag for mesh interpolation", "ex9.c", options->interpolate, &options->interpolate, NULL)); 519566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-refinement_limit", "The maximum volume of a refined cell", "ex9.c", options->refinementLimit, &options->refinementLimit, NULL)); 529566063dSJacob Faibussowitsch PetscCall(PetscOptionsBoundedInt("-num_fields", "The number of section fields", "ex9.c", options->numFields, &options->numFields, NULL, 0)); 53c4762a1bSJed Brown if (options->numFields) { 54c4762a1bSJed Brown len = options->numFields; 559566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(len, &options->numComponents)); 569566063dSJacob Faibussowitsch PetscCall(PetscOptionsIntArray("-num_components", "The number of components per field", "ex9.c", options->numComponents, &len, &flg)); 5763a3b9bcSJacob Faibussowitsch PetscCheck(!flg || !(len != options->numFields), PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Length of components array is %" PetscInt_FMT " should be %" PetscInt_FMT, len, options->numFields); 58c4762a1bSJed Brown } 59c4762a1bSJed Brown len = (options->dim + 1) * PetscMax(1, options->numFields); 609566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(len, &options->numDof)); 619566063dSJacob Faibussowitsch PetscCall(PetscOptionsIntArray("-num_dof", "The dof signature for the section", "ex9.c", options->numDof, &len, &flg)); 621dca8a05SBarry Smith PetscCheck(!flg || len == (options->dim + 1) * PetscMax(1, options->numFields), PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Length of dof array is %" PetscInt_FMT " should be %" PetscInt_FMT, len, (options->dim + 1) * PetscMax(1, options->numFields)); 63c4762a1bSJed Brown 64c4762a1bSJed Brown /* We are specifying the scalar dof, so augment it for multiple components */ 65c4762a1bSJed Brown { 66c4762a1bSJed Brown PetscInt f, d; 67c4762a1bSJed Brown 68c4762a1bSJed Brown for (f = 0; f < options->numFields; ++f) { 69c4762a1bSJed Brown for (d = 0; d <= options->dim; ++d) options->numDof[f * (options->dim + 1) + d] *= options->numComponents[f]; 70c4762a1bSJed Brown } 71c4762a1bSJed Brown } 72c4762a1bSJed Brown 739566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-reuse_array", "Pass in user allocated array to VecGetClosure()", "ex9.c", options->reuseArray, &options->reuseArray, NULL)); 749566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-errors", "Treat failures as errors", "ex9.c", options->errors, &options->errors, NULL)); 759566063dSJacob Faibussowitsch PetscCall(PetscOptionsBoundedInt("-iterations", "The number of iterations for a query", "ex9.c", options->iterations, &options->iterations, NULL, 0)); 769566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-max_cone_time", "The maximum time per run for DMPlexGetCone()", "ex9.c", options->maxConeTime, &options->maxConeTime, NULL)); 779566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-max_closure_time", "The maximum time per run for DMPlexGetTransitiveClosure()", "ex9.c", options->maxClosureTime, &options->maxClosureTime, NULL)); 789566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-max_vec_closure_time", "The maximum time per run for DMPlexVecGetClosure()", "ex9.c", options->maxVecClosureTime, &options->maxVecClosureTime, NULL)); 799566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-print_times", "Print total times, do not check limits", "ex9.c", options->printTimes, &options->printTimes, NULL)); 80d0609cedSBarry Smith PetscOptionsEnd(); 813ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 82c4762a1bSJed Brown } 83c4762a1bSJed Brown 84d71ae5a4SJacob Faibussowitsch static PetscErrorCode CreateSimplex_2D(MPI_Comm comm, DM *newdm) 85d71ae5a4SJacob Faibussowitsch { 86c4762a1bSJed Brown DM dm; 87c4762a1bSJed Brown PetscInt numPoints[2] = {4, 2}; 88c4762a1bSJed Brown PetscInt coneSize[6] = {3, 3, 0, 0, 0, 0}; 89c4762a1bSJed Brown PetscInt cones[6] = {2, 3, 4, 5, 4, 3}; 90c4762a1bSJed Brown PetscInt coneOrientations[6] = {0, 0, 0, 0, 0, 0}; 91c4762a1bSJed Brown PetscScalar vertexCoords[8] = {-0.5, 0.5, 0.0, 0.0, 0.0, 1.0, 0.5, 0.5}; 92c4762a1bSJed Brown PetscInt markerPoints[8] = {2, 1, 3, 1, 4, 1, 5, 1}; 93c4762a1bSJed Brown PetscInt dim = 2, depth = 1, p; 94c4762a1bSJed Brown 95c4762a1bSJed Brown PetscFunctionBegin; 969566063dSJacob Faibussowitsch PetscCall(DMCreate(comm, &dm)); 979566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)dm, "triangular")); 989566063dSJacob Faibussowitsch PetscCall(DMSetType(dm, DMPLEX)); 999566063dSJacob Faibussowitsch PetscCall(DMSetDimension(dm, dim)); 1009566063dSJacob Faibussowitsch PetscCall(DMPlexCreateFromDAG(dm, depth, numPoints, coneSize, cones, coneOrientations, vertexCoords)); 10148a46eb9SPierre Jolivet for (p = 0; p < 4; ++p) PetscCall(DMSetLabelValue(dm, "marker", markerPoints[p * 2], markerPoints[p * 2 + 1])); 102c4762a1bSJed Brown *newdm = dm; 1033ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 104c4762a1bSJed Brown } 105c4762a1bSJed Brown 106d71ae5a4SJacob Faibussowitsch static PetscErrorCode CreateSimplex_3D(MPI_Comm comm, DM *newdm) 107d71ae5a4SJacob Faibussowitsch { 108c4762a1bSJed Brown DM dm; 109c4762a1bSJed Brown PetscInt numPoints[2] = {5, 2}; 110c4762a1bSJed Brown PetscInt coneSize[23] = {4, 4, 0, 0, 0, 0, 0}; 111c4762a1bSJed Brown PetscInt cones[8] = {2, 4, 3, 5, 3, 4, 6, 5}; 112c4762a1bSJed Brown PetscInt coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0}; 113c4762a1bSJed Brown PetscScalar vertexCoords[15] = {0.0, 0.0, -0.5, 0.0, -0.5, 0.0, 1.0, 0.0, 0.0, 0.0, 0.5, 0.0, 0.0, 0.0, 0.5}; 114c4762a1bSJed Brown PetscInt markerPoints[10] = {2, 1, 3, 1, 4, 1, 5, 1, 6, 1}; 115c4762a1bSJed Brown PetscInt dim = 3, depth = 1, p; 116c4762a1bSJed Brown 117c4762a1bSJed Brown PetscFunctionBegin; 1189566063dSJacob Faibussowitsch PetscCall(DMCreate(comm, &dm)); 1199566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)dm, "tetrahedral")); 1209566063dSJacob Faibussowitsch PetscCall(DMSetType(dm, DMPLEX)); 1219566063dSJacob Faibussowitsch PetscCall(DMSetDimension(dm, dim)); 1229566063dSJacob Faibussowitsch PetscCall(DMPlexCreateFromDAG(dm, depth, numPoints, coneSize, cones, coneOrientations, vertexCoords)); 12348a46eb9SPierre Jolivet for (p = 0; p < 5; ++p) PetscCall(DMSetLabelValue(dm, "marker", markerPoints[p * 2], markerPoints[p * 2 + 1])); 124c4762a1bSJed Brown *newdm = dm; 1253ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 126c4762a1bSJed Brown } 127c4762a1bSJed Brown 128d71ae5a4SJacob Faibussowitsch static PetscErrorCode CreateQuad_2D(MPI_Comm comm, DM *newdm) 129d71ae5a4SJacob Faibussowitsch { 130c4762a1bSJed Brown DM dm; 131c4762a1bSJed Brown PetscInt numPoints[2] = {6, 2}; 132c4762a1bSJed Brown PetscInt coneSize[8] = {4, 4, 0, 0, 0, 0, 0, 0}; 133c4762a1bSJed Brown PetscInt cones[8] = {2, 3, 4, 5, 3, 6, 7, 4}; 134c4762a1bSJed Brown PetscInt coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0}; 135c4762a1bSJed Brown PetscScalar vertexCoords[12] = {-0.5, 0.0, 0.0, 0.0, 0.0, 1.0, -0.5, 1.0, 0.5, 0.0, 0.5, 1.0}; 136c4762a1bSJed Brown PetscInt markerPoints[12] = {2, 1, 3, 1, 4, 1, 5, 1, 6, 1, 7, 1}; 137c4762a1bSJed Brown PetscInt dim = 2, depth = 1, p; 138c4762a1bSJed Brown 139c4762a1bSJed Brown PetscFunctionBegin; 1409566063dSJacob Faibussowitsch PetscCall(DMCreate(comm, &dm)); 1419566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)dm, "quadrilateral")); 1429566063dSJacob Faibussowitsch PetscCall(DMSetType(dm, DMPLEX)); 1439566063dSJacob Faibussowitsch PetscCall(DMSetDimension(dm, dim)); 1449566063dSJacob Faibussowitsch PetscCall(DMPlexCreateFromDAG(dm, depth, numPoints, coneSize, cones, coneOrientations, vertexCoords)); 14548a46eb9SPierre Jolivet for (p = 0; p < 6; ++p) PetscCall(DMSetLabelValue(dm, "marker", markerPoints[p * 2], markerPoints[p * 2 + 1])); 146c4762a1bSJed Brown *newdm = dm; 1473ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 148c4762a1bSJed Brown } 149c4762a1bSJed Brown 150d71ae5a4SJacob Faibussowitsch static PetscErrorCode CreateHex_3D(MPI_Comm comm, DM *newdm) 151d71ae5a4SJacob Faibussowitsch { 152c4762a1bSJed Brown DM dm; 153c4762a1bSJed Brown PetscInt numPoints[2] = {12, 2}; 154c4762a1bSJed Brown PetscInt coneSize[14] = {8, 8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}; 155c4762a1bSJed Brown PetscInt cones[16] = {2, 5, 4, 3, 6, 7, 8, 9, 3, 4, 11, 10, 7, 12, 13, 8}; 156c4762a1bSJed Brown PetscInt coneOrientations[16] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}; 1579371c9d4SSatish Balay PetscScalar vertexCoords[36] = {-0.5, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, -0.5, 1.0, 0.0, -0.5, 0.0, 1.0, 0.0, 0.0, 1.0, 0.0, 1.0, 1.0, -0.5, 1.0, 1.0, 0.5, 0.0, 0.0, 0.5, 1.0, 0.0, 0.5, 0.0, 1.0, 0.5, 1.0, 1.0}; 158c4762a1bSJed Brown PetscInt markerPoints[24] = {2, 1, 3, 1, 4, 1, 5, 1, 6, 1, 7, 1, 8, 1, 9, 1, 10, 1, 11, 1, 12, 1, 13, 1}; 159c4762a1bSJed Brown PetscInt dim = 3, depth = 1, p; 160c4762a1bSJed Brown 161c4762a1bSJed Brown PetscFunctionBegin; 1629566063dSJacob Faibussowitsch PetscCall(DMCreate(comm, &dm)); 1639566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)dm, "hexahedral")); 1649566063dSJacob Faibussowitsch PetscCall(DMSetType(dm, DMPLEX)); 1659566063dSJacob Faibussowitsch PetscCall(DMSetDimension(dm, dim)); 1669566063dSJacob Faibussowitsch PetscCall(DMPlexCreateFromDAG(dm, depth, numPoints, coneSize, cones, coneOrientations, vertexCoords)); 16748a46eb9SPierre Jolivet for (p = 0; p < 12; ++p) PetscCall(DMSetLabelValue(dm, "marker", markerPoints[p * 2], markerPoints[p * 2 + 1])); 168c4762a1bSJed Brown *newdm = dm; 1693ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 170c4762a1bSJed Brown } 171c4762a1bSJed Brown 172d71ae5a4SJacob Faibussowitsch static PetscErrorCode CreateMesh(MPI_Comm comm, AppCtx *user, DM *newdm) 173d71ae5a4SJacob Faibussowitsch { 174c4762a1bSJed Brown PetscInt dim = user->dim; 175c4762a1bSJed Brown PetscBool cellSimplex = user->cellSimplex; 176c4762a1bSJed Brown 177c4762a1bSJed Brown PetscFunctionBegin; 178c4762a1bSJed Brown switch (dim) { 179c4762a1bSJed Brown case 2: 180c4762a1bSJed Brown if (cellSimplex) { 1819566063dSJacob Faibussowitsch PetscCall(CreateSimplex_2D(comm, newdm)); 182c4762a1bSJed Brown } else { 1839566063dSJacob Faibussowitsch PetscCall(CreateQuad_2D(comm, newdm)); 184c4762a1bSJed Brown } 185c4762a1bSJed Brown break; 186c4762a1bSJed Brown case 3: 187c4762a1bSJed Brown if (cellSimplex) { 1889566063dSJacob Faibussowitsch PetscCall(CreateSimplex_3D(comm, newdm)); 189c4762a1bSJed Brown } else { 1909566063dSJacob Faibussowitsch PetscCall(CreateHex_3D(comm, newdm)); 191c4762a1bSJed Brown } 192c4762a1bSJed Brown break; 193d71ae5a4SJacob Faibussowitsch default: 194d71ae5a4SJacob Faibussowitsch SETERRQ(comm, PETSC_ERR_ARG_OUTOFRANGE, "Cannot make meshes for dimension %" PetscInt_FMT, dim); 195c4762a1bSJed Brown } 196c4762a1bSJed Brown if (user->refinementLimit > 0.0) { 197c4762a1bSJed Brown DM rdm; 198c4762a1bSJed Brown const char *name; 199c4762a1bSJed Brown 2009566063dSJacob Faibussowitsch PetscCall(DMPlexSetRefinementUniform(*newdm, PETSC_FALSE)); 2019566063dSJacob Faibussowitsch PetscCall(DMPlexSetRefinementLimit(*newdm, user->refinementLimit)); 2029566063dSJacob Faibussowitsch PetscCall(DMRefine(*newdm, PETSC_COMM_SELF, &rdm)); 2039566063dSJacob Faibussowitsch PetscCall(PetscObjectGetName((PetscObject)*newdm, &name)); 2049566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)rdm, name)); 2059566063dSJacob Faibussowitsch PetscCall(DMDestroy(newdm)); 206c4762a1bSJed Brown *newdm = rdm; 207c4762a1bSJed Brown } 208c4762a1bSJed Brown if (user->interpolate) { 209c4762a1bSJed Brown DM idm; 210c4762a1bSJed Brown 2119566063dSJacob Faibussowitsch PetscCall(DMPlexInterpolate(*newdm, &idm)); 2129566063dSJacob Faibussowitsch PetscCall(DMDestroy(newdm)); 213c4762a1bSJed Brown *newdm = idm; 214c4762a1bSJed Brown } 2159566063dSJacob Faibussowitsch PetscCall(DMSetFromOptions(*newdm)); 2163ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 217c4762a1bSJed Brown } 218c4762a1bSJed Brown 219d71ae5a4SJacob Faibussowitsch static PetscErrorCode TestCone(DM dm, AppCtx *user) 220d71ae5a4SJacob Faibussowitsch { 221c4762a1bSJed Brown PetscInt numRuns, cStart, cEnd, c, i; 222c4762a1bSJed Brown PetscReal maxTimePerRun = user->maxConeTime; 223c4762a1bSJed Brown PetscLogStage stage; 224c4762a1bSJed Brown PetscLogEvent event; 225c4762a1bSJed Brown PetscEventPerfInfo eventInfo; 226c4762a1bSJed Brown MPI_Comm comm; 227c4762a1bSJed Brown PetscMPIInt rank; 228c4762a1bSJed Brown 229c4762a1bSJed Brown PetscFunctionBegin; 2309566063dSJacob Faibussowitsch PetscCall(PetscObjectGetComm((PetscObject)dm, &comm)); 2319566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(comm, &rank)); 2329566063dSJacob Faibussowitsch PetscCall(PetscLogStageRegister("DMPlex Cone Test", &stage)); 2339566063dSJacob Faibussowitsch PetscCall(PetscLogEventRegister("Cone", PETSC_OBJECT_CLASSID, &event)); 2349566063dSJacob Faibussowitsch PetscCall(PetscLogStagePush(stage)); 2359566063dSJacob Faibussowitsch PetscCall(DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd)); 2369566063dSJacob Faibussowitsch PetscCall(PetscLogEventBegin(event, 0, 0, 0, 0)); 237c4762a1bSJed Brown for (i = 0; i < user->iterations; ++i) { 238c4762a1bSJed Brown for (c = cStart; c < cEnd; ++c) { 239c4762a1bSJed Brown const PetscInt *cone; 240c4762a1bSJed Brown 2419566063dSJacob Faibussowitsch PetscCall(DMPlexGetCone(dm, c, &cone)); 242c4762a1bSJed Brown } 243c4762a1bSJed Brown } 2449566063dSJacob Faibussowitsch PetscCall(PetscLogEventEnd(event, 0, 0, 0, 0)); 2459566063dSJacob Faibussowitsch PetscCall(PetscLogStagePop()); 246c4762a1bSJed Brown 2479566063dSJacob Faibussowitsch PetscCall(PetscLogEventGetPerfInfo(stage, event, &eventInfo)); 248c4762a1bSJed Brown numRuns = (cEnd - cStart) * user->iterations; 24963a3b9bcSJacob Faibussowitsch PetscCheck(eventInfo.count == 1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of event calls %d should be 1", eventInfo.count); 25063a3b9bcSJacob Faibussowitsch PetscCheck((PetscInt)eventInfo.flops == 0, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of event flops %" PetscInt_FMT " should be 0", (PetscInt)eventInfo.flops); 251c4762a1bSJed Brown if (user->printTimes) { 25263a3b9bcSJacob Faibussowitsch PetscCall(PetscSynchronizedPrintf(comm, "[%d] Cones: %" PetscInt_FMT " Total time: %.3es Average time per cone: %.3es\n", rank, numRuns, eventInfo.time, eventInfo.time / numRuns)); 2539566063dSJacob Faibussowitsch PetscCall(PetscSynchronizedFlush(comm, PETSC_STDOUT)); 254c4762a1bSJed Brown } else if (eventInfo.time > maxTimePerRun * numRuns) { 25563a3b9bcSJacob Faibussowitsch PetscCall(PetscSynchronizedPrintf(comm, "[%d] Cones: %" PetscInt_FMT " Average time per cone: %gs standard: %gs\n", rank, numRuns, (double)(eventInfo.time / numRuns), (double)maxTimePerRun)); 2569566063dSJacob Faibussowitsch PetscCall(PetscSynchronizedFlush(comm, PETSC_STDOUT)); 25763a3b9bcSJacob Faibussowitsch PetscCheck(!user->errors, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Average time for cone %g > standard %g", (double)(eventInfo.time / numRuns), (double)maxTimePerRun); 258c4762a1bSJed Brown } 2593ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 260c4762a1bSJed Brown } 261c4762a1bSJed Brown 262d71ae5a4SJacob Faibussowitsch static PetscErrorCode TestTransitiveClosure(DM dm, AppCtx *user) 263d71ae5a4SJacob Faibussowitsch { 264c4762a1bSJed Brown PetscInt numRuns, cStart, cEnd, c, i; 265c4762a1bSJed Brown PetscReal maxTimePerRun = user->maxClosureTime; 266c4762a1bSJed Brown PetscLogStage stage; 267c4762a1bSJed Brown PetscLogEvent event; 268c4762a1bSJed Brown PetscEventPerfInfo eventInfo; 269c4762a1bSJed Brown MPI_Comm comm; 270c4762a1bSJed Brown PetscMPIInt rank; 271c4762a1bSJed Brown 272c4762a1bSJed Brown PetscFunctionBegin; 2739566063dSJacob Faibussowitsch PetscCall(PetscObjectGetComm((PetscObject)dm, &comm)); 2749566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(comm, &rank)); 2759566063dSJacob Faibussowitsch PetscCall(PetscLogStageRegister("DMPlex Transitive Closure Test", &stage)); 2769566063dSJacob Faibussowitsch PetscCall(PetscLogEventRegister("TransitiveClosure", PETSC_OBJECT_CLASSID, &event)); 2779566063dSJacob Faibussowitsch PetscCall(PetscLogStagePush(stage)); 2789566063dSJacob Faibussowitsch PetscCall(DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd)); 2799566063dSJacob Faibussowitsch PetscCall(PetscLogEventBegin(event, 0, 0, 0, 0)); 280c4762a1bSJed Brown for (i = 0; i < user->iterations; ++i) { 281c4762a1bSJed Brown for (c = cStart; c < cEnd; ++c) { 282c4762a1bSJed Brown PetscInt *closure = NULL; 283c4762a1bSJed Brown PetscInt closureSize; 284c4762a1bSJed Brown 2859566063dSJacob Faibussowitsch PetscCall(DMPlexGetTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure)); 2869566063dSJacob Faibussowitsch PetscCall(DMPlexRestoreTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure)); 287c4762a1bSJed Brown } 288c4762a1bSJed Brown } 2899566063dSJacob Faibussowitsch PetscCall(PetscLogEventEnd(event, 0, 0, 0, 0)); 2909566063dSJacob Faibussowitsch PetscCall(PetscLogStagePop()); 291c4762a1bSJed Brown 2929566063dSJacob Faibussowitsch PetscCall(PetscLogEventGetPerfInfo(stage, event, &eventInfo)); 293c4762a1bSJed Brown numRuns = (cEnd - cStart) * user->iterations; 29463a3b9bcSJacob Faibussowitsch PetscCheck(eventInfo.count == 1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of event calls %d should be 1", eventInfo.count); 29563a3b9bcSJacob Faibussowitsch PetscCheck((PetscInt)eventInfo.flops == 0, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of event flops %" PetscInt_FMT " should be 0", (PetscInt)eventInfo.flops); 296c4762a1bSJed Brown if (user->printTimes) { 29763a3b9bcSJacob Faibussowitsch PetscCall(PetscSynchronizedPrintf(comm, "[%d] Closures: %" PetscInt_FMT " Total time: %.3es Average time per cone: %.3es\n", rank, numRuns, eventInfo.time, eventInfo.time / numRuns)); 2989566063dSJacob Faibussowitsch PetscCall(PetscSynchronizedFlush(comm, PETSC_STDOUT)); 299c4762a1bSJed Brown } else if (eventInfo.time > maxTimePerRun * numRuns) { 30063a3b9bcSJacob Faibussowitsch PetscCall(PetscSynchronizedPrintf(comm, "[%d] Closures: %" PetscInt_FMT " Average time per cone: %gs standard: %gs\n", rank, numRuns, (double)(eventInfo.time / numRuns), (double)maxTimePerRun)); 3019566063dSJacob Faibussowitsch PetscCall(PetscSynchronizedFlush(comm, PETSC_STDOUT)); 30263a3b9bcSJacob Faibussowitsch PetscCheck(!user->errors, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Average time for closure %g > standard %g", (double)(eventInfo.time / numRuns), (double)maxTimePerRun); 303c4762a1bSJed Brown } 3043ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 305c4762a1bSJed Brown } 306c4762a1bSJed Brown 307d71ae5a4SJacob Faibussowitsch static PetscErrorCode TestVecClosure(DM dm, PetscBool useIndex, PetscBool useSpectral, AppCtx *user) 308d71ae5a4SJacob Faibussowitsch { 309c4762a1bSJed Brown PetscSection s; 310c4762a1bSJed Brown Vec v; 311c4762a1bSJed Brown PetscInt numRuns, cStart, cEnd, c, i; 312c4762a1bSJed Brown PetscScalar tmpArray[64]; 313c4762a1bSJed Brown PetscScalar *userArray = user->reuseArray ? tmpArray : NULL; 314c4762a1bSJed Brown PetscReal maxTimePerRun = user->maxVecClosureTime; 315c4762a1bSJed Brown PetscLogStage stage; 316c4762a1bSJed Brown PetscLogEvent event; 317c4762a1bSJed Brown PetscEventPerfInfo eventInfo; 318c4762a1bSJed Brown MPI_Comm comm; 319c4762a1bSJed Brown PetscMPIInt rank; 320c4762a1bSJed Brown 321c4762a1bSJed Brown PetscFunctionBegin; 3229566063dSJacob Faibussowitsch PetscCall(PetscObjectGetComm((PetscObject)dm, &comm)); 3239566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(comm, &rank)); 324c4762a1bSJed Brown if (useIndex) { 325c4762a1bSJed Brown if (useSpectral) { 3269566063dSJacob Faibussowitsch PetscCall(PetscLogStageRegister("DMPlex Vector Closure with Index Test", &stage)); 3279566063dSJacob Faibussowitsch PetscCall(PetscLogEventRegister("VecClosureInd", PETSC_OBJECT_CLASSID, &event)); 328c4762a1bSJed Brown } else { 3299566063dSJacob Faibussowitsch PetscCall(PetscLogStageRegister("DMPlex Vector Spectral Closure with Index Test", &stage)); 3309566063dSJacob Faibussowitsch PetscCall(PetscLogEventRegister("VecClosureSpecInd", PETSC_OBJECT_CLASSID, &event)); 331c4762a1bSJed Brown } 332c4762a1bSJed Brown } else { 333c4762a1bSJed Brown if (useSpectral) { 3349566063dSJacob Faibussowitsch PetscCall(PetscLogStageRegister("DMPlex Vector Spectral Closure Test", &stage)); 3359566063dSJacob Faibussowitsch PetscCall(PetscLogEventRegister("VecClosureSpec", PETSC_OBJECT_CLASSID, &event)); 336c4762a1bSJed Brown } else { 3379566063dSJacob Faibussowitsch PetscCall(PetscLogStageRegister("DMPlex Vector Closure Test", &stage)); 3389566063dSJacob Faibussowitsch PetscCall(PetscLogEventRegister("VecClosure", PETSC_OBJECT_CLASSID, &event)); 339c4762a1bSJed Brown } 340c4762a1bSJed Brown } 3419566063dSJacob Faibussowitsch PetscCall(PetscLogStagePush(stage)); 3429566063dSJacob Faibussowitsch PetscCall(DMSetNumFields(dm, user->numFields)); 3439566063dSJacob Faibussowitsch PetscCall(DMPlexCreateSection(dm, NULL, user->numComponents, user->numDof, 0, NULL, NULL, NULL, NULL, &s)); 3449566063dSJacob Faibussowitsch PetscCall(DMSetLocalSection(dm, s)); 3459566063dSJacob Faibussowitsch if (useIndex) PetscCall(DMPlexCreateClosureIndex(dm, s)); 3469566063dSJacob Faibussowitsch if (useSpectral) PetscCall(DMPlexSetClosurePermutationTensor(dm, PETSC_DETERMINE, s)); 3479566063dSJacob Faibussowitsch PetscCall(PetscSectionDestroy(&s)); 3489566063dSJacob Faibussowitsch PetscCall(DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd)); 3499566063dSJacob Faibussowitsch PetscCall(DMGetLocalVector(dm, &v)); 3509566063dSJacob Faibussowitsch PetscCall(PetscLogEventBegin(event, 0, 0, 0, 0)); 351c4762a1bSJed Brown for (i = 0; i < user->iterations; ++i) { 352c4762a1bSJed Brown for (c = cStart; c < cEnd; ++c) { 353c4762a1bSJed Brown PetscScalar *closure = userArray; 354c4762a1bSJed Brown PetscInt closureSize = 64; 355c4762a1bSJed Brown 3569566063dSJacob Faibussowitsch PetscCall(DMPlexVecGetClosure(dm, s, v, c, &closureSize, &closure)); 3579566063dSJacob Faibussowitsch if (!user->reuseArray) PetscCall(DMPlexVecRestoreClosure(dm, s, v, c, &closureSize, &closure)); 358c4762a1bSJed Brown } 359c4762a1bSJed Brown } 3609566063dSJacob Faibussowitsch PetscCall(PetscLogEventEnd(event, 0, 0, 0, 0)); 3619566063dSJacob Faibussowitsch PetscCall(DMRestoreLocalVector(dm, &v)); 3629566063dSJacob Faibussowitsch PetscCall(PetscLogStagePop()); 363c4762a1bSJed Brown 3649566063dSJacob Faibussowitsch PetscCall(PetscLogEventGetPerfInfo(stage, event, &eventInfo)); 365c4762a1bSJed Brown numRuns = (cEnd - cStart) * user->iterations; 36663a3b9bcSJacob Faibussowitsch PetscCheck(eventInfo.count == 1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of event calls %d should be 1", eventInfo.count); 36763a3b9bcSJacob Faibussowitsch PetscCheck((PetscInt)eventInfo.flops == 0, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of event flops %" PetscInt_FMT " should be 0", (PetscInt)eventInfo.flops); 368c4762a1bSJed Brown if (user->printTimes || eventInfo.time > maxTimePerRun * numRuns) { 369c4762a1bSJed Brown const char *title = "VecClosures"; 370c4762a1bSJed Brown const char *titleIndex = "VecClosures with Index"; 371c4762a1bSJed Brown const char *titleSpec = "VecClosures Spectral"; 372c4762a1bSJed Brown const char *titleSpecIndex = "VecClosures Spectral with Index"; 373c4762a1bSJed Brown 374c4762a1bSJed Brown if (user->printTimes) { 3759371c9d4SSatish Balay PetscCall(PetscSynchronizedPrintf(comm, "[%d] %s: %" PetscInt_FMT " Total time: %.3es Average time per vector closure: %.3es\n", rank, useIndex ? (useSpectral ? titleSpecIndex : titleIndex) : (useSpectral ? titleSpec : title), numRuns, 3769371c9d4SSatish Balay eventInfo.time, eventInfo.time / numRuns)); 3779566063dSJacob Faibussowitsch PetscCall(PetscSynchronizedFlush(comm, PETSC_STDOUT)); 378c4762a1bSJed Brown } else { 3799371c9d4SSatish Balay PetscCall( 3809371c9d4SSatish Balay PetscSynchronizedPrintf(comm, "[%d] %s: %" PetscInt_FMT " Average time per vector closure: %gs standard: %gs\n", rank, useIndex ? (useSpectral ? titleSpecIndex : titleIndex) : (useSpectral ? titleSpec : title), numRuns, (double)(eventInfo.time / numRuns), (double)maxTimePerRun)); 3819566063dSJacob Faibussowitsch PetscCall(PetscSynchronizedFlush(comm, PETSC_STDOUT)); 38263a3b9bcSJacob Faibussowitsch PetscCheck(!user->errors, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Average time for vector closure %g > standard %g", (double)(eventInfo.time / numRuns), (double)maxTimePerRun); 383c4762a1bSJed Brown } 384c4762a1bSJed Brown } 3853ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 386c4762a1bSJed Brown } 387c4762a1bSJed Brown 388d71ae5a4SJacob Faibussowitsch static PetscErrorCode CleanupContext(AppCtx *user) 389d71ae5a4SJacob Faibussowitsch { 390c4762a1bSJed Brown PetscFunctionBegin; 3919566063dSJacob Faibussowitsch PetscCall(PetscFree(user->numComponents)); 3929566063dSJacob Faibussowitsch PetscCall(PetscFree(user->numDof)); 3933ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 394c4762a1bSJed Brown } 395c4762a1bSJed Brown 396d71ae5a4SJacob Faibussowitsch int main(int argc, char **argv) 397d71ae5a4SJacob Faibussowitsch { 398c4762a1bSJed Brown DM dm; 399c4762a1bSJed Brown AppCtx user; 400c4762a1bSJed Brown 401327415f7SBarry Smith PetscFunctionBeginUser; 4029566063dSJacob Faibussowitsch PetscCall(PetscInitialize(&argc, &argv, NULL, help)); 4039566063dSJacob Faibussowitsch PetscCall(ProcessOptions(&user)); 4049566063dSJacob Faibussowitsch PetscCall(PetscLogDefaultBegin()); 4059566063dSJacob Faibussowitsch PetscCall(CreateMesh(PETSC_COMM_WORLD, &user, &dm)); 4069566063dSJacob Faibussowitsch PetscCall(TestCone(dm, &user)); 4079566063dSJacob Faibussowitsch PetscCall(TestTransitiveClosure(dm, &user)); 4089566063dSJacob Faibussowitsch PetscCall(TestVecClosure(dm, PETSC_FALSE, PETSC_FALSE, &user)); 4099566063dSJacob Faibussowitsch PetscCall(TestVecClosure(dm, PETSC_TRUE, PETSC_FALSE, &user)); 410c4762a1bSJed Brown if (!user.cellSimplex && user.spectral) { 4119566063dSJacob Faibussowitsch PetscCall(TestVecClosure(dm, PETSC_FALSE, PETSC_TRUE, &user)); 4129566063dSJacob Faibussowitsch PetscCall(TestVecClosure(dm, PETSC_TRUE, PETSC_TRUE, &user)); 413c4762a1bSJed Brown } 4149566063dSJacob Faibussowitsch PetscCall(DMDestroy(&dm)); 4159566063dSJacob Faibussowitsch PetscCall(CleanupContext(&user)); 4169566063dSJacob Faibussowitsch PetscCall(PetscFinalize()); 417b122ec5aSJacob Faibussowitsch return 0; 418c4762a1bSJed Brown } 419c4762a1bSJed Brown 420c4762a1bSJed Brown /*TEST 421c4762a1bSJed Brown 422c4762a1bSJed Brown build: 423dfd57a17SPierre Jolivet requires: defined(PETSC_USE_LOG) 424c4762a1bSJed Brown 425c4762a1bSJed Brown # 2D Simplex P_1 scalar tests 426c4762a1bSJed Brown testset: 427c4762a1bSJed Brown args: -num_dof 1,0,0 -iterations 2 -print_times 428c4762a1bSJed Brown test: 429c4762a1bSJed Brown suffix: correctness_0 430c4762a1bSJed Brown test: 431c4762a1bSJed Brown suffix: correctness_1 432c4762a1bSJed Brown args: -interpolate -dm_refine 2 433c4762a1bSJed Brown test: 434c4762a1bSJed Brown suffix: correctness_2 435c4762a1bSJed Brown requires: triangle 436*56245cf9SMatthew G. Knepley args: -interpolate -dm_refine 5 437c4762a1bSJed Brown test: 438c4762a1bSJed Brown suffix: 0 439c4762a1bSJed Brown TODO: Only for performance testing 440c4762a1bSJed Brown args: -num_dof 1,0,0 -iterations 10000 -max_cone_time 1.1e-8 -max_closure_time 1.3e-7 -max_vec_closure_time 3.6e-7 441c4762a1bSJed Brown test: 442c4762a1bSJed Brown suffix: 1 443c4762a1bSJed Brown requires: triangle 444c4762a1bSJed Brown TODO: Only for performance testing 445c4762a1bSJed Brown args: -refinement_limit 1.0e-5 -num_dof 1,0,0 -iterations 2 -max_cone_time 2.1e-8 -max_closure_time 1.5e-7 -max_vec_closure_time 3.6e-7 446c4762a1bSJed Brown test: 447c4762a1bSJed Brown suffix: 2 448c4762a1bSJed Brown TODO: Only for performance testing 449c4762a1bSJed Brown args: -num_fields 1 -num_components 1 -num_dof 1,0,0 -iterations 10000 -max_cone_time 1.1e-8 -max_closure_time 1.3e-7 -max_vec_closure_time 4.5e-7 450c4762a1bSJed Brown test: 451c4762a1bSJed Brown suffix: 3 452c4762a1bSJed Brown requires: triangle 453c4762a1bSJed Brown TODO: Only for performance testing 454c4762a1bSJed Brown args: -refinement_limit 1.0e-5 -num_fields 1 -num_components 1 -num_dof 1,0,0 -iterations 2 -max_cone_time 2.1e-8 -max_closure_time 1.5e-7 -max_vec_closure_time 4.7e-7 455c4762a1bSJed Brown test: 456c4762a1bSJed Brown suffix: 4 457c4762a1bSJed Brown TODO: Only for performance testing 458c4762a1bSJed Brown args: -interpolate -num_dof 1,0,0 -iterations 10000 -max_cone_time 1.1e-8 -max_closure_time 6.5e-7 -max_vec_closure_time 1.0e-6 459c4762a1bSJed Brown test: 460c4762a1bSJed Brown suffix: 5 461c4762a1bSJed Brown requires: triangle 462c4762a1bSJed Brown TODO: Only for performance testing 463c4762a1bSJed Brown args: -interpolate -refinement_limit 1.0e-4 -num_dof 1,0,0 -iterations 2 -max_cone_time 2.1e-8 -max_closure_time 6.5e-7 -max_vec_closure_time 1.0e-6 464c4762a1bSJed Brown test: 465c4762a1bSJed Brown suffix: 6 466c4762a1bSJed Brown TODO: Only for performance testing 467c4762a1bSJed Brown args: -interpolate -num_fields 1 -num_components 1 -num_dof 1,0,0 -iterations 10000 -max_cone_time 1.1e-8 -max_closure_time 6.5e-7 -max_vec_closure_time 1.1e-6 468c4762a1bSJed Brown test: 469c4762a1bSJed Brown suffix: 7 470c4762a1bSJed Brown requires: triangle 471c4762a1bSJed Brown TODO: Only for performance testing 472c4762a1bSJed Brown args: -interpolate -refinement_limit 1.0e-4 -num_fields 1 -num_components 1 -num_dof 1,0,0 -iterations 2 -max_cone_time 2.1e-8 -max_closure_time 6.5e-7 -max_vec_closure_time 1.2e-6 473c4762a1bSJed Brown 474c4762a1bSJed Brown # 2D Simplex P_1 vector tests 475c4762a1bSJed Brown # 2D Simplex P_2 scalar tests 476c4762a1bSJed Brown # 2D Simplex P_2 vector tests 477c4762a1bSJed Brown # 2D Simplex P_2/P_1 vector/scalar tests 478c4762a1bSJed Brown # 2D Quad P_1 scalar tests 479c4762a1bSJed Brown # 2D Quad P_1 vector tests 480c4762a1bSJed Brown # 2D Quad P_2 scalar tests 481c4762a1bSJed Brown # 2D Quad P_2 vector tests 482c4762a1bSJed Brown # 3D Simplex P_1 scalar tests 483c4762a1bSJed Brown # 3D Simplex P_1 vector tests 484c4762a1bSJed Brown # 3D Simplex P_2 scalar tests 485c4762a1bSJed Brown # 3D Simplex P_2 vector tests 486c4762a1bSJed Brown # 3D Hex P_1 scalar tests 487c4762a1bSJed Brown # 3D Hex P_1 vector tests 488c4762a1bSJed Brown # 3D Hex P_2 scalar tests 489c4762a1bSJed Brown # 3D Hex P_2 vector tests 490c4762a1bSJed Brown 491c4762a1bSJed Brown TEST*/ 492