1 static char help[] = "Exhaustive memory tracking for DMPlex.\n\n\n"; 2 3 #include <petscdmplex.h> 4 5 static PetscErrorCode EstimateMemory(DM dm, PetscLogDouble *est) 6 { 7 DMLabel marker; 8 PetscInt cdim, depth, d, pStart, pEnd, p, Nd[4] = {0, 0, 0, 0}, lsize = 0, rmem = 0, imem = 0; 9 PetscInt coneSecMem = 0, coneMem = 0, supportSecMem = 0, supportMem = 0, labelMem = 0; 10 11 PetscFunctionBeginUser; 12 PetscCall(PetscPrintf(PETSC_COMM_SELF, "Memory Estimates\n")); 13 PetscCall(DMGetCoordinateDim(dm, &cdim)); 14 PetscCall(DMPlexGetDepth(dm, &depth)); 15 PetscCall(DMPlexGetChart(dm, &pStart, &pEnd)); 16 for (d = 0; d <= depth; ++d) { 17 PetscInt start, end; 18 19 PetscCall(DMPlexGetDepthStratum(dm, d, &start, &end)); 20 Nd[d] = end - start; 21 } 22 /* Coordinates: 3 Nv reals + 2*Nv + 2*Nv ints */ 23 rmem += cdim*Nd[0]; 24 imem += 2*Nd[0] + 2*Nd[0]; 25 PetscCall(PetscPrintf(PETSC_COMM_SELF, " Coordinate mem: %" PetscInt_FMT " %" PetscInt_FMT "\n", (PetscInt)(cdim*Nd[0]*sizeof(PetscReal)), (PetscInt)(4*Nd[0]*sizeof(PetscInt)))); 26 /* Depth: Nc+Nf+Ne+Nv ints */ 27 for (d = 0; d <= depth; ++d) labelMem += Nd[d]; 28 /* Cell Type: Nc+Nf+Ne+Nv ints */ 29 for (d = 0; d <= depth; ++d) labelMem += Nd[d]; 30 /* Marker */ 31 PetscCall(DMGetLabel(dm, "marker", &marker)); 32 if (marker) PetscCall(DMLabelGetStratumSize(marker, 1, &lsize)); 33 labelMem += lsize; 34 PetscCall(PetscPrintf(PETSC_COMM_SELF, " Label mem: %" PetscInt_FMT "\n", (PetscInt)(labelMem*sizeof(PetscInt)))); 35 //imem += labelMem; 36 /* Cones and Orientations: 4 Nc + 3 Nf + 2 Ne ints + (Nc+Nf+Ne) ints no separate orientation section */ 37 for (d = 0; d <= depth; ++d) coneSecMem += 2*Nd[d]; 38 for (p = pStart; p < pEnd; ++p) { 39 PetscInt csize; 40 41 PetscCall(DMPlexGetConeSize(dm, p, &csize)); 42 coneMem += csize; 43 } 44 PetscCall(PetscPrintf(PETSC_COMM_SELF, " Cone mem: %" PetscInt_FMT " %" PetscInt_FMT " (%" PetscInt_FMT ")\n", (PetscInt)(coneMem*sizeof(PetscInt)), (PetscInt)(coneSecMem*sizeof(PetscInt)), (PetscInt)(coneMem*sizeof(PetscInt)))); 45 imem += 2*coneMem + coneSecMem; 46 /* Supports: 4 Nc + 3 Nf + 2 Ne ints + Nc+Nf+Ne ints */ 47 for (d = 0; d <= depth; ++d) supportSecMem += 2*Nd[d]; 48 for (p = pStart; p < pEnd; ++p) { 49 PetscInt ssize; 50 51 PetscCall(DMPlexGetSupportSize(dm, p, &ssize)); 52 supportMem += ssize; 53 } 54 PetscCall(PetscPrintf(PETSC_COMM_SELF, " Support mem: %" PetscInt_FMT " %" PetscInt_FMT "\n", (PetscInt)(supportMem*sizeof(PetscInt)), (PetscInt)(supportSecMem*sizeof(PetscInt)))); 55 imem += supportMem + supportSecMem; 56 *est = ((PetscLogDouble) imem)*sizeof(PetscInt) + ((PetscLogDouble) rmem)*sizeof(PetscReal); 57 PetscCall(PetscPrintf(PETSC_COMM_WORLD, " Estimated memory %" PetscInt_FMT "\n", (PetscInt) *est)); 58 PetscFunctionReturn(0); 59 } 60 61 int main(int argc, char **argv) 62 { 63 DM dm; 64 PetscBool trace = PETSC_FALSE, checkMemory = PETSC_TRUE, auxMemory = PETSC_FALSE; 65 PetscLogDouble before, after, est = 0, clean, max; 66 67 PetscCall(PetscInitialize(&argc, &argv, NULL, help)); 68 PetscCall(PetscOptionsGetBool(NULL, NULL, "-trace", &trace, NULL)); 69 PetscCall(PetscOptionsGetBool(NULL, NULL, "-check_memory", &checkMemory, NULL)); 70 PetscCall(PetscOptionsGetBool(NULL, NULL, "-aux_memory", &auxMemory, NULL)); 71 PetscCall(PetscMemorySetGetMaximumUsage()); 72 PetscCall(PetscMallocGetCurrentUsage(&before)); 73 if (trace) PetscCall(PetscMallocTraceSet(NULL, PETSC_TRUE, 5000.)); 74 PetscCall(DMCreate(PETSC_COMM_WORLD, &dm)); 75 PetscCall(DMSetType(dm, DMPLEX)); 76 PetscCall(DMSetFromOptions(dm)); 77 if (trace) PetscCall(PetscMallocTraceSet(NULL, PETSC_FALSE, 5000)); 78 PetscCall(DMViewFromOptions(dm, NULL, "-dm_view")); 79 PetscCall(PetscMallocGetCurrentUsage(&after)); 80 PetscCall(PetscMemoryGetMaximumUsage(&max)); 81 PetscCall(EstimateMemory(dm, &est)); 82 PetscCall(DMDestroy(&dm)); 83 PetscCall(PetscMallocGetCurrentUsage(&clean)); 84 PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Measured Memory\n")); 85 if (auxMemory) { 86 PetscCall(PetscPrintf(PETSC_COMM_WORLD, " Initial memory %" PetscInt_FMT "\n Extra memory for build %" PetscInt_FMT "\n Memory after destroy %" PetscInt_FMT "\n", 87 (PetscInt) before, (PetscInt) (max-after), (PetscInt) clean)); 88 } 89 if (checkMemory) { 90 PetscCall(PetscPrintf(PETSC_COMM_WORLD, " Memory for mesh %" PetscInt_FMT "\n", (PetscInt) (after-before))); 91 PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Discrepancy %" PetscInt_FMT "\n", (PetscInt) PetscAbsReal(after-before-est))); 92 } 93 PetscCall(PetscFinalize()); 94 return 0; 95 } 96 97 /*TEST 98 build: 99 requires: !defined(PETSC_USE_64BIT_INDICES) double !complex !defined(PETSCTEST_VALGRIND) 100 101 # Memory checks cannot be included in tests because the allocated memory differs among environments 102 testset: 103 args: -malloc_requested_size -dm_plex_box_faces 5,5 -check_memory 0 104 test: 105 suffix: tri 106 requires: triangle 107 args: -dm_plex_simplex 1 -dm_plex_interpolate 0 108 109 test: 110 suffix: tri_interp 111 requires: triangle 112 args: -dm_plex_simplex 1 -dm_plex_interpolate 1 113 114 test: 115 suffix: quad 116 args: -dm_plex_simplex 0 -dm_plex_interpolate 0 117 118 test: 119 suffix: quad_interp 120 args: -dm_plex_simplex 0 -dm_plex_interpolate 1 121 122 # Memory checks cannot be included in tests because the allocated memory differs among environments 123 testset: 124 args: -malloc_requested_size -dm_plex_dim 3 -dm_plex_box_faces 5,5,5 -check_memory 0 125 test: 126 suffix: tet 127 requires: ctetgen 128 args: -dm_plex_simplex 1 -dm_plex_interpolate 0 129 130 test: 131 suffix: tet_interp 132 requires: ctetgen 133 args: -dm_plex_simplex 1 -dm_plex_interpolate 1 134 135 test: 136 suffix: hex 137 args: -dm_plex_simplex 0 -dm_plex_interpolate 0 138 139 test: 140 suffix: hex_interp 141 args: -dm_plex_simplex 0 -dm_plex_interpolate 1 142 TEST*/ 143