17d0a6c19SBarry Smith 2e5c89e4eSSatish Balay /* 3e5c89e4eSSatish Balay Interface to malloc() and free(). This code allows for 4e5c89e4eSSatish Balay logging of memory usage and some error checking 5e5c89e4eSSatish Balay */ 6c6db04a5SJed Brown #include <petscsys.h> /*I "petscsys.h" I*/ 7e5c89e4eSSatish Balay #if defined(PETSC_HAVE_STDLIB_H) 8e5c89e4eSSatish Balay #include <stdlib.h> 9e5c89e4eSSatish Balay #endif 10e5c89e4eSSatish Balay #if defined(PETSC_HAVE_MALLOC_H) 11e5c89e4eSSatish Balay #include <malloc.h> 12e5c89e4eSSatish Balay #endif 13e5c89e4eSSatish Balay 14e5c89e4eSSatish Balay 15e5c89e4eSSatish Balay /* 16e5c89e4eSSatish Balay These are defined in mal.c and ensure that malloced space is PetscScalar aligned 17e5c89e4eSSatish Balay */ 187087cfbeSBarry Smith extern PetscErrorCode PetscMallocAlign(size_t,int,const char[],const char[],const char[],void**); 197087cfbeSBarry Smith extern PetscErrorCode PetscFreeAlign(void*,int,const char[],const char[],const char[]); 207087cfbeSBarry Smith extern PetscErrorCode PetscTrMallocDefault(size_t,int,const char[],const char[],const char[],void**); 217087cfbeSBarry Smith extern PetscErrorCode PetscTrFreeDefault(void*,int,const char[],const char[],const char[]); 22e5c89e4eSSatish Balay 23e5c89e4eSSatish Balay 240700a824SBarry Smith #define CLASSID_VALUE ((PetscClassId) 0xf0e0d0c9) 250700a824SBarry Smith #define ALREADY_FREED ((PetscClassId) 0x0f0e0d9c) 26e5c89e4eSSatish Balay 27e5c89e4eSSatish Balay typedef struct _trSPACE { 28e5c89e4eSSatish Balay size_t size; 29e5c89e4eSSatish Balay int id; 30e5c89e4eSSatish Balay int lineno; 31e5c89e4eSSatish Balay const char *filename; 32e5c89e4eSSatish Balay const char *functionname; 33e5c89e4eSSatish Balay const char *dirname; 340700a824SBarry Smith PetscClassId classid; 358bf1f09cSShri Abhyankar #if defined(PETSC_USE_DEBUG) 36e5c89e4eSSatish Balay PetscStack stack; 37e5c89e4eSSatish Balay #endif 38e5c89e4eSSatish Balay struct _trSPACE *next,*prev; 39e5c89e4eSSatish Balay } TRSPACE; 40e5c89e4eSSatish Balay 4125b53cc9SJed Brown /* HEADER_BYTES is the number of bytes in a PetscMalloc() header. 4225b53cc9SJed Brown It is sizeof(TRSPACE) padded to be a multiple of PETSC_MEMALIGN. 4325b53cc9SJed Brown */ 44e5c89e4eSSatish Balay 45a64a8e02SBarry Smith #define HEADER_BYTES ((sizeof(TRSPACE)+(PETSC_MEMALIGN-1)) & ~(PETSC_MEMALIGN-1)) 46e5c89e4eSSatish Balay 47e5c89e4eSSatish Balay 4825b53cc9SJed Brown /* This union is used to insure that the block passed to the user retains 4925b53cc9SJed Brown a minimum alignment of PETSC_MEMALIGN. 5025b53cc9SJed Brown */ 51e5c89e4eSSatish Balay typedef union { 52e5c89e4eSSatish Balay TRSPACE sp; 5325b53cc9SJed Brown char v[HEADER_BYTES]; 54e5c89e4eSSatish Balay } TrSPACE; 55e5c89e4eSSatish Balay 56b022a5c1SBarry Smith 57e5c89e4eSSatish Balay static size_t TRallocated = 0; 58e5c89e4eSSatish Balay static int TRfrags = 0; 59e5c89e4eSSatish Balay static TRSPACE *TRhead = 0; 60e5c89e4eSSatish Balay static int TRid = 0; 61ace3abfcSBarry Smith static PetscBool TRdebugLevel = PETSC_FALSE; 62e5c89e4eSSatish Balay static size_t TRMaxMem = 0; 63e5c89e4eSSatish Balay /* 64e5c89e4eSSatish Balay Arrays to log information on all Mallocs 65e5c89e4eSSatish Balay */ 66e5c89e4eSSatish Balay static int PetscLogMallocMax = 10000,PetscLogMalloc = -1; 67574034a9SJed Brown static size_t PetscLogMallocThreshold = 0; 68e5c89e4eSSatish Balay static size_t *PetscLogMallocLength; 69e5c89e4eSSatish Balay static const char **PetscLogMallocDirectory,**PetscLogMallocFile,**PetscLogMallocFunction; 70e5c89e4eSSatish Balay 71e5c89e4eSSatish Balay #undef __FUNCT__ 72b022a5c1SBarry Smith #define __FUNCT__ "PetscSetUseTrMalloc_Private" 73b022a5c1SBarry Smith PetscErrorCode PetscSetUseTrMalloc_Private(void) 74b022a5c1SBarry Smith { 75b022a5c1SBarry Smith PetscErrorCode ierr; 76b022a5c1SBarry Smith 77b022a5c1SBarry Smith PetscFunctionBegin; 78b022a5c1SBarry Smith ierr = PetscMallocSet(PetscTrMallocDefault,PetscTrFreeDefault);CHKERRQ(ierr); 79b022a5c1SBarry Smith TRallocated = 0; 80b022a5c1SBarry Smith TRfrags = 0; 81b022a5c1SBarry Smith TRhead = 0; 82b022a5c1SBarry Smith TRid = 0; 83b022a5c1SBarry Smith TRdebugLevel = PETSC_FALSE; 84b022a5c1SBarry Smith TRMaxMem = 0; 85b022a5c1SBarry Smith PetscLogMallocMax = 10000; 86b022a5c1SBarry Smith PetscLogMalloc = -1; 87b022a5c1SBarry Smith PetscFunctionReturn(0); 88b022a5c1SBarry Smith } 89b022a5c1SBarry Smith 90b022a5c1SBarry Smith #undef __FUNCT__ 91e5c89e4eSSatish Balay #define __FUNCT__ "PetscMallocValidate" 92e5c89e4eSSatish Balay /*@C 93e5c89e4eSSatish Balay PetscMallocValidate - Test the memory for corruption. This can be used to 94e5c89e4eSSatish Balay check for memory overwrites. 95e5c89e4eSSatish Balay 96e5c89e4eSSatish Balay Input Parameter: 97e5c89e4eSSatish Balay + line - line number where call originated. 98e5c89e4eSSatish Balay . function - name of function calling 99e5c89e4eSSatish Balay . file - file where function is 100e5c89e4eSSatish Balay - dir - directory where function is 101e5c89e4eSSatish Balay 102e5c89e4eSSatish Balay Return value: 103e5c89e4eSSatish Balay The number of errors detected. 104e5c89e4eSSatish Balay 105e5c89e4eSSatish Balay Output Effect: 106e5c89e4eSSatish Balay Error messages are written to stdout. 107e5c89e4eSSatish Balay 108e5c89e4eSSatish Balay Level: advanced 109e5c89e4eSSatish Balay 110e5c89e4eSSatish Balay Notes: 111e5c89e4eSSatish Balay You should generally use CHKMEMQ as a short cut for calling this 112e5c89e4eSSatish Balay routine. 113e5c89e4eSSatish Balay 114e5c89e4eSSatish Balay The line, function, file and dir are given by the C preprocessor as 115e5c89e4eSSatish Balay __LINE__, __FUNCT__, __FILE__, and __DIR__ 116e5c89e4eSSatish Balay 117e5c89e4eSSatish Balay The Fortran calling sequence is simply PetscMallocValidate(ierr) 118e5c89e4eSSatish Balay 119e5c89e4eSSatish Balay No output is generated if there are no problems detected. 120e5c89e4eSSatish Balay 121e5c89e4eSSatish Balay .seealso: CHKMEMQ 122e5c89e4eSSatish Balay 123e5c89e4eSSatish Balay @*/ 1247087cfbeSBarry Smith PetscErrorCode PetscMallocValidate(int line,const char function[],const char file[],const char dir[]) 125e5c89e4eSSatish Balay { 1266c093d5bSvictor TRSPACE *head,*lasthead; 127e5c89e4eSSatish Balay char *a; 1280700a824SBarry Smith PetscClassId *nend; 129e5c89e4eSSatish Balay 130e5c89e4eSSatish Balay PetscFunctionBegin; 1316c093d5bSvictor head = TRhead; lasthead = NULL; 132e5c89e4eSSatish Balay while (head) { 1330700a824SBarry Smith if (head->classid != CLASSID_VALUE) { 134e5c89e4eSSatish Balay (*PetscErrorPrintf)("PetscMallocValidate: error detected at %s() line %d in %s%s\n",function,line,dir,file); 135e5c89e4eSSatish Balay (*PetscErrorPrintf)("Memory at address %p is corrupted\n",head); 136e5c89e4eSSatish Balay (*PetscErrorPrintf)("Probably write past beginning or end of array\n"); 1376c093d5bSvictor if (lasthead) 1386c093d5bSvictor (*PetscErrorPrintf)("Last intact block allocated in %s() line %d in %s%s\n",lasthead->functionname,lasthead->lineno,lasthead->dirname,lasthead->filename); 139e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEMC," "); 140e5c89e4eSSatish Balay } 141e5c89e4eSSatish Balay a = (char *)(((TrSPACE*)head) + 1); 1420700a824SBarry Smith nend = (PetscClassId *)(a + head->size); 1430700a824SBarry Smith if (*nend != CLASSID_VALUE) { 144e5c89e4eSSatish Balay (*PetscErrorPrintf)("PetscMallocValidate: error detected at %s() line %d in %s%s\n",function,line,dir,file); 145e5c89e4eSSatish Balay if (*nend == ALREADY_FREED) { 146e5c89e4eSSatish Balay (*PetscErrorPrintf)("Memory [id=%d(%.0f)] at address %p already freed\n",head->id,(PetscLogDouble)head->size,a); 147e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEMC," "); 148e5c89e4eSSatish Balay } else { 149e5c89e4eSSatish Balay (*PetscErrorPrintf)("Memory [id=%d(%.0f)] at address %p is corrupted (probably write past end of array)\n",head->id,(PetscLogDouble)head->size,a); 150e5c89e4eSSatish Balay (*PetscErrorPrintf)("Memory originally allocated in %s() line %d in %s%s\n",head->functionname,head->lineno,head->dirname,head->filename); 151e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEMC," "); 152e5c89e4eSSatish Balay } 153e5c89e4eSSatish Balay } 1546c093d5bSvictor lasthead = head; 155e5c89e4eSSatish Balay head = head->next; 156e5c89e4eSSatish Balay } 157e5c89e4eSSatish Balay PetscFunctionReturn(0); 158e5c89e4eSSatish Balay } 159e5c89e4eSSatish Balay 160e5c89e4eSSatish Balay #undef __FUNCT__ 161e5c89e4eSSatish Balay #define __FUNCT__ "PetscTrMallocDefault" 162e5c89e4eSSatish Balay /* 163e5c89e4eSSatish Balay PetscTrMallocDefault - Malloc with tracing. 164e5c89e4eSSatish Balay 165e5c89e4eSSatish Balay Input Parameters: 166e5c89e4eSSatish Balay + a - number of bytes to allocate 167e5c89e4eSSatish Balay . lineno - line number where used. Use __LINE__ for this 168e5c89e4eSSatish Balay . function - function calling routine. Use __FUNCT__ for this 169e5c89e4eSSatish Balay . filename - file name where used. Use __FILE__ for this 170e5c89e4eSSatish Balay - dir - directory where file is. Use __SDIR__ for this 171e5c89e4eSSatish Balay 172e5c89e4eSSatish Balay Returns: 173e5c89e4eSSatish Balay double aligned pointer to requested storage, or null if not 174e5c89e4eSSatish Balay available. 175e5c89e4eSSatish Balay */ 1767087cfbeSBarry Smith PetscErrorCode PetscTrMallocDefault(size_t a,int lineno,const char function[],const char filename[],const char dir[],void**result) 177e5c89e4eSSatish Balay { 178e5c89e4eSSatish Balay TRSPACE *head; 179e5c89e4eSSatish Balay char *inew; 180e5c89e4eSSatish Balay size_t nsize; 181e5c89e4eSSatish Balay PetscErrorCode ierr; 182e5c89e4eSSatish Balay 183e5c89e4eSSatish Balay PetscFunctionBegin; 18417186662SBarry Smith if (!a) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Trying to malloc zero size array"); 1852a6744ebSBarry Smith 186e5c89e4eSSatish Balay if (TRdebugLevel) { 187e5c89e4eSSatish Balay ierr = PetscMallocValidate(lineno,function,filename,dir); if (ierr) PetscFunctionReturn(ierr); 188e5c89e4eSSatish Balay } 189e5c89e4eSSatish Balay 19025b53cc9SJed Brown nsize = (a + (PETSC_MEMALIGN-1)) & ~(PETSC_MEMALIGN-1); 1910700a824SBarry Smith ierr = PetscMallocAlign(nsize+sizeof(TrSPACE)+sizeof(PetscClassId),lineno,function,filename,dir,(void**)&inew);CHKERRQ(ierr); 192e5c89e4eSSatish Balay 193e5c89e4eSSatish Balay head = (TRSPACE *)inew; 194e5c89e4eSSatish Balay inew += sizeof(TrSPACE); 195e5c89e4eSSatish Balay 196e5c89e4eSSatish Balay if (TRhead) TRhead->prev = head; 197e5c89e4eSSatish Balay head->next = TRhead; 198e5c89e4eSSatish Balay TRhead = head; 199e5c89e4eSSatish Balay head->prev = 0; 200e5c89e4eSSatish Balay head->size = nsize; 201e5c89e4eSSatish Balay head->id = TRid; 202e5c89e4eSSatish Balay head->lineno = lineno; 203e5c89e4eSSatish Balay 204e5c89e4eSSatish Balay head->filename = filename; 205e5c89e4eSSatish Balay head->functionname = function; 206e5c89e4eSSatish Balay head->dirname = dir; 2070700a824SBarry Smith head->classid = CLASSID_VALUE; 2080700a824SBarry Smith *(PetscClassId *)(inew + nsize) = CLASSID_VALUE; 209e5c89e4eSSatish Balay 210e5c89e4eSSatish Balay TRallocated += nsize; 211e5c89e4eSSatish Balay if (TRallocated > TRMaxMem) { 212e5c89e4eSSatish Balay TRMaxMem = TRallocated; 213e5c89e4eSSatish Balay } 214e5c89e4eSSatish Balay TRfrags++; 215e5c89e4eSSatish Balay 2168bf1f09cSShri Abhyankar #if defined(PETSC_USE_DEBUG) 2171f46d60fSShri Abhyankar ierr = PetscStackCopy((PetscStack*)PetscThreadLocalGetValue(petscstack),&head->stack);CHKERRQ(ierr); 218*2c9581d2SBarry Smith /* fix the line number to where the malloc() was called, not the PetscFunctionBegin; */ 219*2c9581d2SBarry Smith head->stack.line[head->stack.currentsize-2] = lineno; 220e5c89e4eSSatish Balay #endif 221e5c89e4eSSatish Balay 222e5c89e4eSSatish Balay /* 223e5c89e4eSSatish Balay Allow logging of all mallocs made 224e5c89e4eSSatish Balay */ 225574034a9SJed Brown if (PetscLogMalloc > -1 && PetscLogMalloc < PetscLogMallocMax && a >= PetscLogMallocThreshold) { 226e5c89e4eSSatish Balay if (!PetscLogMalloc) { 227e5c89e4eSSatish Balay PetscLogMallocLength = (size_t*)malloc(PetscLogMallocMax*sizeof(size_t)); 228e32f2f54SBarry Smith if (!PetscLogMallocLength) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM," "); 229e5c89e4eSSatish Balay PetscLogMallocDirectory = (const char**)malloc(PetscLogMallocMax*sizeof(char**)); 230e32f2f54SBarry Smith if (!PetscLogMallocDirectory) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM," "); 231e5c89e4eSSatish Balay PetscLogMallocFile = (const char**)malloc(PetscLogMallocMax*sizeof(char**)); 232e32f2f54SBarry Smith if (!PetscLogMallocFile) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM," "); 233e5c89e4eSSatish Balay PetscLogMallocFunction = (const char**)malloc(PetscLogMallocMax*sizeof(char**)); 234e32f2f54SBarry Smith if (!PetscLogMallocFunction) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM," "); 235e5c89e4eSSatish Balay } 236e5c89e4eSSatish Balay PetscLogMallocLength[PetscLogMalloc] = nsize; 237e5c89e4eSSatish Balay PetscLogMallocDirectory[PetscLogMalloc] = dir; 238e5c89e4eSSatish Balay PetscLogMallocFile[PetscLogMalloc] = filename; 239e5c89e4eSSatish Balay PetscLogMallocFunction[PetscLogMalloc++] = function; 240e5c89e4eSSatish Balay } 241e5c89e4eSSatish Balay *result = (void*)inew; 242e5c89e4eSSatish Balay PetscFunctionReturn(0); 243e5c89e4eSSatish Balay } 244e5c89e4eSSatish Balay 245e5c89e4eSSatish Balay 246e5c89e4eSSatish Balay #undef __FUNCT__ 247e5c89e4eSSatish Balay #define __FUNCT__ "PetscTrFreeDefault" 248e5c89e4eSSatish Balay /* 249e5c89e4eSSatish Balay PetscTrFreeDefault - Free with tracing. 250e5c89e4eSSatish Balay 251e5c89e4eSSatish Balay Input Parameters: 252e5c89e4eSSatish Balay . a - pointer to a block allocated with PetscTrMalloc 253e5c89e4eSSatish Balay . lineno - line number where used. Use __LINE__ for this 254e5c89e4eSSatish Balay . function - function calling routine. Use __FUNCT__ for this 255e5c89e4eSSatish Balay . file - file name where used. Use __FILE__ for this 256e5c89e4eSSatish Balay . dir - directory where file is. Use __SDIR__ for this 257e5c89e4eSSatish Balay */ 2587087cfbeSBarry Smith PetscErrorCode PetscTrFreeDefault(void *aa,int line,const char function[],const char file[],const char dir[]) 259e5c89e4eSSatish Balay { 260e5c89e4eSSatish Balay char *a = (char*)aa; 261e5c89e4eSSatish Balay TRSPACE *head; 262e5c89e4eSSatish Balay char *ahead; 263e5c89e4eSSatish Balay PetscErrorCode ierr; 2640700a824SBarry Smith PetscClassId *nend; 265e5c89e4eSSatish Balay 266e5c89e4eSSatish Balay PetscFunctionBegin; 267e5c89e4eSSatish Balay /* Do not try to handle empty blocks */ 268e5c89e4eSSatish Balay if (!a) { 269e5c89e4eSSatish Balay (*PetscErrorPrintf)("PetscTrFreeDefault called from %s() line %d in %s%s\n",function,line,dir,file); 270e32f2f54SBarry Smith SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Trying to free null block: Free called from %s() line %d in %s%s\n",function,line,dir,file); 271e5c89e4eSSatish Balay } 272e5c89e4eSSatish Balay 273e5c89e4eSSatish Balay if (TRdebugLevel) { 274e5c89e4eSSatish Balay ierr = PetscMallocValidate(line,function,file,dir);CHKERRQ(ierr); 275e5c89e4eSSatish Balay } 276e5c89e4eSSatish Balay 277e5c89e4eSSatish Balay ahead = a; 278e5c89e4eSSatish Balay a = a - sizeof(TrSPACE); 279e5c89e4eSSatish Balay head = (TRSPACE *)a; 280e5c89e4eSSatish Balay 2810700a824SBarry Smith if (head->classid != CLASSID_VALUE) { 282e5c89e4eSSatish Balay (*PetscErrorPrintf)("PetscTrFreeDefault() called from %s() line %d in %s%s\n",function,line,dir,file); 283e5c89e4eSSatish Balay (*PetscErrorPrintf)("Block at address %p is corrupted; cannot free;\nmay be block not allocated with PetscMalloc()\n",a); 284e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEMC,"Bad location or corrupted memory"); 285e5c89e4eSSatish Balay } 2860700a824SBarry Smith nend = (PetscClassId *)(ahead + head->size); 2870700a824SBarry Smith if (*nend != CLASSID_VALUE) { 288e5c89e4eSSatish Balay if (*nend == ALREADY_FREED) { 289e5c89e4eSSatish Balay (*PetscErrorPrintf)("PetscTrFreeDefault() called from %s() line %d in %s%s\n",function,line,dir,file); 290e5c89e4eSSatish Balay (*PetscErrorPrintf)("Block [id=%d(%.0f)] at address %p was already freed\n",head->id,(PetscLogDouble)head->size,a + sizeof(TrSPACE)); 291e5c89e4eSSatish Balay if (head->lineno > 0 && head->lineno < 50000 /* sanity check */) { 292e5c89e4eSSatish Balay (*PetscErrorPrintf)("Block freed in %s() line %d in %s%s\n",head->functionname,head->lineno,head->dirname,head->filename); 293e5c89e4eSSatish Balay } else { 294e5c89e4eSSatish Balay (*PetscErrorPrintf)("Block allocated in %s() line %d in %s%s\n",head->functionname,-head->lineno,head->dirname,head->filename); 295e5c89e4eSSatish Balay } 296e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Memory already freed"); 297e5c89e4eSSatish Balay } else { 298e5c89e4eSSatish Balay /* Damaged tail */ 299e5c89e4eSSatish Balay (*PetscErrorPrintf)("PetscTrFreeDefault() called from %s() line %d in %s%s\n",function,line,dir,file); 300e5c89e4eSSatish Balay (*PetscErrorPrintf)("Block [id=%d(%.0f)] at address %p is corrupted (probably write past end of array)\n",head->id,(PetscLogDouble)head->size,a); 301e5c89e4eSSatish Balay (*PetscErrorPrintf)("Block allocated in %s() line %d in %s%s\n",head->functionname,head->lineno,head->dirname,head->filename); 302e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEMC,"Corrupted memory"); 303e5c89e4eSSatish Balay } 304e5c89e4eSSatish Balay } 305e5c89e4eSSatish Balay /* Mark the location freed */ 306e5c89e4eSSatish Balay *nend = ALREADY_FREED; 307e5c89e4eSSatish Balay /* Save location where freed. If we suspect the line number, mark as allocated location */ 308e5c89e4eSSatish Balay if (line > 0 && line < 50000) { 309e5c89e4eSSatish Balay head->lineno = line; 310e5c89e4eSSatish Balay head->filename = file; 311e5c89e4eSSatish Balay head->functionname = function; 312e5c89e4eSSatish Balay head->dirname = dir; 313e5c89e4eSSatish Balay } else { 314e5c89e4eSSatish Balay head->lineno = - head->lineno; 315e5c89e4eSSatish Balay } 316e5c89e4eSSatish Balay /* zero out memory - helps to find some reuse of already freed memory */ 317e5c89e4eSSatish Balay ierr = PetscMemzero(aa,head->size);CHKERRQ(ierr); 318e5c89e4eSSatish Balay 319e5c89e4eSSatish Balay TRallocated -= head->size; 320e5c89e4eSSatish Balay TRfrags --; 321e5c89e4eSSatish Balay if (head->prev) head->prev->next = head->next; 322e5c89e4eSSatish Balay else TRhead = head->next; 323e5c89e4eSSatish Balay 324e5c89e4eSSatish Balay if (head->next) head->next->prev = head->prev; 325e5c89e4eSSatish Balay ierr = PetscFreeAlign(a,line,function,file,dir);CHKERRQ(ierr); 326e5c89e4eSSatish Balay PetscFunctionReturn(0); 327e5c89e4eSSatish Balay } 328e5c89e4eSSatish Balay 329e5c89e4eSSatish Balay 330e5c89e4eSSatish Balay #undef __FUNCT__ 331e5c89e4eSSatish Balay #define __FUNCT__ "PetscMemoryShowUsage" 332fe7fb379SMatthew Knepley /*@C 333e5c89e4eSSatish Balay PetscMemoryShowUsage - Shows the amount of memory currently being used 334e5c89e4eSSatish Balay in a communicator. 335e5c89e4eSSatish Balay 336e5c89e4eSSatish Balay Collective on PetscViewer 337e5c89e4eSSatish Balay 338e5c89e4eSSatish Balay Input Parameter: 339e5c89e4eSSatish Balay + viewer - the viewer that defines the communicator 340e5c89e4eSSatish Balay - message - string printed before values 341e5c89e4eSSatish Balay 342e5c89e4eSSatish Balay Level: intermediate 343e5c89e4eSSatish Balay 344e5c89e4eSSatish Balay Concepts: memory usage 345e5c89e4eSSatish Balay 3461bb97394SVictor Eijkhout .seealso: PetscMallocDump(), PetscMemoryGetCurrentUsage() 347e5c89e4eSSatish Balay @*/ 3487087cfbeSBarry Smith PetscErrorCode PetscMemoryShowUsage(PetscViewer viewer,const char message[]) 349e5c89e4eSSatish Balay { 350e5c89e4eSSatish Balay PetscLogDouble allocated,maximum,resident,residentmax; 351e5c89e4eSSatish Balay PetscErrorCode ierr; 352e5c89e4eSSatish Balay PetscMPIInt rank; 353e5c89e4eSSatish Balay MPI_Comm comm; 354e5c89e4eSSatish Balay 355e5c89e4eSSatish Balay PetscFunctionBegin; 356e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_WORLD; 357e5c89e4eSSatish Balay ierr = PetscMallocGetCurrentUsage(&allocated);CHKERRQ(ierr); 358e5c89e4eSSatish Balay ierr = PetscMallocGetMaximumUsage(&maximum);CHKERRQ(ierr); 359e5c89e4eSSatish Balay ierr = PetscMemoryGetCurrentUsage(&resident);CHKERRQ(ierr); 360e5c89e4eSSatish Balay ierr = PetscMemoryGetMaximumUsage(&residentmax);CHKERRQ(ierr); 361e5c89e4eSSatish Balay if (residentmax > 0) residentmax = PetscMax(resident,residentmax); 362e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 363e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 364e5c89e4eSSatish Balay ierr = PetscViewerASCIIPrintf(viewer,message);CHKERRQ(ierr); 3657b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr); 366e5c89e4eSSatish Balay if (resident && residentmax && allocated) { 367e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d]Current space PetscMalloc()ed %g, max space PetscMalloced() %g\n[%d]Current process memory %g max process memory %g\n",rank,allocated,maximum,rank,resident,residentmax);CHKERRQ(ierr); 368e5c89e4eSSatish Balay } else if (resident && residentmax) { 369e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d]Run with -malloc to get statistics on PetscMalloc() calls\n[%d]Current process memory %g max process memory %g\n",rank,rank,resident,residentmax);CHKERRQ(ierr); 370e5c89e4eSSatish Balay } else if (resident && allocated) { 371e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d]Current space PetscMalloc()ed %g, max space PetscMalloced() %g\n[%d]Current process memory %g, run with -memory_info to get max memory usage\n",rank,allocated,maximum,rank,resident);CHKERRQ(ierr); 372e5c89e4eSSatish Balay } else if (allocated) { 373e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d]Current space PetscMalloc()ed %g, max space PetscMalloced() %g\n[%d]OS cannot compute process memory\n",rank,allocated,maximum,rank);CHKERRQ(ierr); 374e5c89e4eSSatish Balay } else { 375e5c89e4eSSatish Balay ierr = PetscViewerASCIIPrintf(viewer,"Run with -malloc to get statistics on PetscMalloc() calls\nOS cannot compute process memory\n");CHKERRQ(ierr); 376e5c89e4eSSatish Balay } 377e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 3787b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr); 379e5c89e4eSSatish Balay PetscFunctionReturn(0); 380e5c89e4eSSatish Balay } 381e5c89e4eSSatish Balay 382e5c89e4eSSatish Balay #undef __FUNCT__ 383e5c89e4eSSatish Balay #define __FUNCT__ "PetscMallocGetCurrentUsage" 384e5c89e4eSSatish Balay /*@C 385e5c89e4eSSatish Balay PetscMallocGetCurrentUsage - gets the current amount of memory used that was PetscMalloc()ed 386e5c89e4eSSatish Balay 387e5c89e4eSSatish Balay Not Collective 388e5c89e4eSSatish Balay 389e5c89e4eSSatish Balay Output Parameters: 390e5c89e4eSSatish Balay . space - number of bytes currently allocated 391e5c89e4eSSatish Balay 392e5c89e4eSSatish Balay Level: intermediate 393e5c89e4eSSatish Balay 394e5c89e4eSSatish Balay Concepts: memory usage 395e5c89e4eSSatish Balay 396e5c89e4eSSatish Balay .seealso: PetscMallocDump(), PetscMallocDumpLog(), PetscMallocGetMaximumUsage(), PetscMemoryGetCurrentUsage(), 397e5c89e4eSSatish Balay PetscMemoryGetMaximumUsage() 398e5c89e4eSSatish Balay @*/ 3997087cfbeSBarry Smith PetscErrorCode PetscMallocGetCurrentUsage(PetscLogDouble *space) 400e5c89e4eSSatish Balay { 401e5c89e4eSSatish Balay PetscFunctionBegin; 402e5c89e4eSSatish Balay *space = (PetscLogDouble) TRallocated; 403e5c89e4eSSatish Balay PetscFunctionReturn(0); 404e5c89e4eSSatish Balay } 405e5c89e4eSSatish Balay 406e5c89e4eSSatish Balay #undef __FUNCT__ 407e5c89e4eSSatish Balay #define __FUNCT__ "PetscMallocGetMaximumUsage" 408e5c89e4eSSatish Balay /*@C 409e5c89e4eSSatish Balay PetscMallocGetMaximumUsage - gets the maximum amount of memory used that was PetscMalloc()ed at any time 410e5c89e4eSSatish Balay during this run. 411e5c89e4eSSatish Balay 412e5c89e4eSSatish Balay Not Collective 413e5c89e4eSSatish Balay 414e5c89e4eSSatish Balay Output Parameters: 415e5c89e4eSSatish Balay . space - maximum number of bytes ever allocated at one time 416e5c89e4eSSatish Balay 417e5c89e4eSSatish Balay Level: intermediate 418e5c89e4eSSatish Balay 419e5c89e4eSSatish Balay Concepts: memory usage 420e5c89e4eSSatish Balay 421e5c89e4eSSatish Balay .seealso: PetscMallocDump(), PetscMallocDumpLog(), PetscMallocGetMaximumUsage(), PetscMemoryGetCurrentUsage(), 422e5c89e4eSSatish Balay PetscMemoryGetCurrentUsage() 423e5c89e4eSSatish Balay @*/ 4247087cfbeSBarry Smith PetscErrorCode PetscMallocGetMaximumUsage(PetscLogDouble *space) 425e5c89e4eSSatish Balay { 426e5c89e4eSSatish Balay PetscFunctionBegin; 427e5c89e4eSSatish Balay *space = (PetscLogDouble) TRMaxMem; 428e5c89e4eSSatish Balay PetscFunctionReturn(0); 429e5c89e4eSSatish Balay } 430e5c89e4eSSatish Balay 431a64a8e02SBarry Smith #if defined(PETSC_USE_DEBUG) 432a64a8e02SBarry Smith #undef __FUNCT__ 433a64a8e02SBarry Smith #define __FUNCT__ "PetscMallocGetStack" 434a64a8e02SBarry Smith /*@C 435a64a8e02SBarry Smith PetscMallocGetStack - returns a pointer to the stack for the location in the program a call to PetscMalloc() was used to obtain that memory 436a64a8e02SBarry Smith 437a64a8e02SBarry Smith Collective on PETSC_COMM_WORLD 438a64a8e02SBarry Smith 439a64a8e02SBarry Smith Input Parameter: 440a64a8e02SBarry Smith . ptr - the memory location 441a64a8e02SBarry Smith 442a64a8e02SBarry Smith Output Paramter: 443a64a8e02SBarry Smith . stack - the stack indicating where the program allocated this memory 444a64a8e02SBarry Smith 445a64a8e02SBarry Smith Level: intermediate 446a64a8e02SBarry Smith 447a64a8e02SBarry Smith .seealso: PetscMallocGetCurrentUsage(), PetscMallocDumpLog() 448a64a8e02SBarry Smith @*/ 449a64a8e02SBarry Smith PetscErrorCode PetscMallocGetStack(void *ptr,PetscStack **stack) 450a64a8e02SBarry Smith { 451a64a8e02SBarry Smith TRSPACE *head; 452a64a8e02SBarry Smith 453a64a8e02SBarry Smith PetscFunctionBegin; 454a64a8e02SBarry Smith head = (TRSPACE*) (((char*)ptr) - HEADER_BYTES); 455a64a8e02SBarry Smith *stack = &head->stack; 456a64a8e02SBarry Smith PetscFunctionReturn(0); 457a64a8e02SBarry Smith } 458a64a8e02SBarry Smith #endif 459a64a8e02SBarry Smith 460e5c89e4eSSatish Balay #undef __FUNCT__ 461e5c89e4eSSatish Balay #define __FUNCT__ "PetscMallocDump" 462e5c89e4eSSatish Balay /*@C 463e5c89e4eSSatish Balay PetscMallocDump - Dumps the allocated memory blocks to a file. The information 464e5c89e4eSSatish Balay printed is: size of space (in bytes), address of space, id of space, 465e5c89e4eSSatish Balay file in which space was allocated, and line number at which it was 466e5c89e4eSSatish Balay allocated. 467e5c89e4eSSatish Balay 468e5c89e4eSSatish Balay Collective on PETSC_COMM_WORLD 469e5c89e4eSSatish Balay 470e5c89e4eSSatish Balay Input Parameter: 471e5c89e4eSSatish Balay . fp - file pointer. If fp is NULL, stdout is assumed. 472e5c89e4eSSatish Balay 473e5c89e4eSSatish Balay Options Database Key: 474e5c89e4eSSatish Balay . -malloc_dump - Dumps unfreed memory during call to PetscFinalize() 475e5c89e4eSSatish Balay 476e5c89e4eSSatish Balay Level: intermediate 477e5c89e4eSSatish Balay 478e5c89e4eSSatish Balay Fortran Note: 479e5c89e4eSSatish Balay The calling sequence in Fortran is PetscMallocDump(integer ierr) 480e5c89e4eSSatish Balay The fp defaults to stdout. 481e5c89e4eSSatish Balay 482e5c89e4eSSatish Balay Notes: uses MPI_COMM_WORLD, because this may be called in PetscFinalize() after PETSC_COMM_WORLD 483e5c89e4eSSatish Balay has been freed. 484e5c89e4eSSatish Balay 485e5c89e4eSSatish Balay Concepts: memory usage 486e5c89e4eSSatish Balay Concepts: memory bleeding 487e5c89e4eSSatish Balay Concepts: bleeding memory 488e5c89e4eSSatish Balay 4899e9a1f8fSvictor .seealso: PetscMallocGetCurrentUsage(), PetscMallocDumpLog() 490e5c89e4eSSatish Balay @*/ 4917087cfbeSBarry Smith PetscErrorCode PetscMallocDump(FILE *fp) 492e5c89e4eSSatish Balay { 493e5c89e4eSSatish Balay TRSPACE *head; 494e5c89e4eSSatish Balay PetscErrorCode ierr; 495e5c89e4eSSatish Balay PetscMPIInt rank; 496e5c89e4eSSatish Balay 497e5c89e4eSSatish Balay PetscFunctionBegin; 498e5c89e4eSSatish Balay ierr = MPI_Comm_rank(MPI_COMM_WORLD,&rank);CHKERRQ(ierr); 499da9f1d6bSBarry Smith if (!fp) fp = PETSC_STDOUT; 500e5c89e4eSSatish Balay if (TRallocated > 0) { 501e5c89e4eSSatish Balay fprintf(fp,"[%d]Total space allocated %.0f bytes\n",rank,(PetscLogDouble)TRallocated); 502e5c89e4eSSatish Balay } 503e5c89e4eSSatish Balay head = TRhead; 504e5c89e4eSSatish Balay while (head) { 505e5c89e4eSSatish Balay fprintf(fp,"[%2d]%.0f bytes %s() line %d in %s%s\n",rank,(PetscLogDouble)head->size,head->functionname,head->lineno,head->dirname,head->filename); 5068bf1f09cSShri Abhyankar #if defined(PETSC_USE_DEBUG) 507e5c89e4eSSatish Balay ierr = PetscStackPrint(&head->stack,fp);CHKERRQ(ierr); 508e5c89e4eSSatish Balay #endif 509e5c89e4eSSatish Balay head = head->next; 510e5c89e4eSSatish Balay } 511e5c89e4eSSatish Balay PetscFunctionReturn(0); 512e5c89e4eSSatish Balay } 513e5c89e4eSSatish Balay 514e5c89e4eSSatish Balay /* ---------------------------------------------------------------------------- */ 515e5c89e4eSSatish Balay 516e5c89e4eSSatish Balay #undef __FUNCT__ 517e5c89e4eSSatish Balay #define __FUNCT__ "PetscMallocSetDumpLog" 518e5c89e4eSSatish Balay /*@C 519e5c89e4eSSatish Balay PetscMallocSetDumpLog - Activates logging of all calls to PetscMalloc(). 520e5c89e4eSSatish Balay 521e5c89e4eSSatish Balay Not Collective 522e5c89e4eSSatish Balay 523e5c89e4eSSatish Balay Options Database Key: 524574034a9SJed Brown + -malloc_log <filename> - Activates PetscMallocDumpLog() 525574034a9SJed Brown - -malloc_log_threshold <min> - Activates logging and sets a minimum size 526e5c89e4eSSatish Balay 527e5c89e4eSSatish Balay Level: advanced 528e5c89e4eSSatish Balay 529574034a9SJed Brown .seealso: PetscMallocDump(), PetscMallocDumpLog(), PetscMallocSetDumpLogThreshold() 530e5c89e4eSSatish Balay @*/ 5317087cfbeSBarry Smith PetscErrorCode PetscMallocSetDumpLog(void) 532e5c89e4eSSatish Balay { 53321b680ceSJed Brown PetscErrorCode ierr; 53421b680ceSJed Brown 535e5c89e4eSSatish Balay PetscFunctionBegin; 536e5c89e4eSSatish Balay PetscLogMalloc = 0; 53721b680ceSJed Brown ierr = PetscMemorySetGetMaximumUsage();CHKERRQ(ierr); 538e5c89e4eSSatish Balay PetscFunctionReturn(0); 539e5c89e4eSSatish Balay } 540e5c89e4eSSatish Balay 541e5c89e4eSSatish Balay #undef __FUNCT__ 542574034a9SJed Brown #define __FUNCT__ "PetscMallocSetDumpLogThreshold" 543574034a9SJed Brown /*@C 544574034a9SJed Brown PetscMallocSetDumpLogThreshold - Activates logging of all calls to PetscMalloc(). 545574034a9SJed Brown 546574034a9SJed Brown Not Collective 547574034a9SJed Brown 548574034a9SJed Brown Input Arguments: 549574034a9SJed Brown . logmin - minimum allocation size to log, or PETSC_DEFAULT 550574034a9SJed Brown 551574034a9SJed Brown Options Database Key: 552574034a9SJed Brown + -malloc_log <filename> - Activates PetscMallocDumpLog() 553574034a9SJed Brown - -malloc_log_threshold <min> - Activates logging and sets a minimum size 554574034a9SJed Brown 555574034a9SJed Brown Level: advanced 556574034a9SJed Brown 557574034a9SJed Brown .seealso: PetscMallocDump(), PetscMallocDumpLog(), PetscMallocSetDumpLog() 558574034a9SJed Brown @*/ 559574034a9SJed Brown PetscErrorCode PetscMallocSetDumpLogThreshold(PetscLogDouble logmin) 560574034a9SJed Brown { 561574034a9SJed Brown PetscErrorCode ierr; 562574034a9SJed Brown 563574034a9SJed Brown PetscFunctionBegin; 564574034a9SJed Brown ierr = PetscMallocSetDumpLog();CHKERRQ(ierr); 565574034a9SJed Brown if (logmin < 0) logmin = 0.0; /* PETSC_DEFAULT or PETSC_DECIDE */ 566574034a9SJed Brown PetscLogMallocThreshold = (size_t)logmin; 567574034a9SJed Brown PetscFunctionReturn(0); 568574034a9SJed Brown } 569574034a9SJed Brown 570574034a9SJed Brown #undef __FUNCT__ 57118a2528dSJed Brown #define __FUNCT__ "PetscMallocGetDumpLog" 57218a2528dSJed Brown /*@C 57318a2528dSJed Brown PetscMallocGetDumpLog - Determine whether all calls to PetscMalloc() are being logged 57418a2528dSJed Brown 57518a2528dSJed Brown Not Collective 57618a2528dSJed Brown 57718a2528dSJed Brown Output Arguments 57818a2528dSJed Brown . logging - PETSC_TRUE if logging is active 57918a2528dSJed Brown 58018a2528dSJed Brown Options Database Key: 58118a2528dSJed Brown . -malloc_log - Activates PetscMallocDumpLog() 58218a2528dSJed Brown 58318a2528dSJed Brown Level: advanced 58418a2528dSJed Brown 58518a2528dSJed Brown .seealso: PetscMallocDump(), PetscMallocDumpLog() 58618a2528dSJed Brown @*/ 58718a2528dSJed Brown PetscErrorCode PetscMallocGetDumpLog(PetscBool *logging) 58818a2528dSJed Brown { 58918a2528dSJed Brown 59018a2528dSJed Brown PetscFunctionBegin; 59118a2528dSJed Brown *logging = (PetscBool)(PetscLogMalloc >= 0); 59218a2528dSJed Brown PetscFunctionReturn(0); 59318a2528dSJed Brown } 59418a2528dSJed Brown 59518a2528dSJed Brown #undef __FUNCT__ 596e5c89e4eSSatish Balay #define __FUNCT__ "PetscMallocDumpLog" 597e5c89e4eSSatish Balay /*@C 598e5c89e4eSSatish Balay PetscMallocDumpLog - Dumps the log of all calls to PetscMalloc(); also calls 59921b680ceSJed Brown PetscMemoryGetMaximumUsage() 600e5c89e4eSSatish Balay 601e5c89e4eSSatish Balay Collective on PETSC_COMM_WORLD 602e5c89e4eSSatish Balay 603e5c89e4eSSatish Balay Input Parameter: 604e5c89e4eSSatish Balay . fp - file pointer; or PETSC_NULL 605e5c89e4eSSatish Balay 606e5c89e4eSSatish Balay Options Database Key: 607e5c89e4eSSatish Balay . -malloc_log - Activates PetscMallocDumpLog() 608e5c89e4eSSatish Balay 609e5c89e4eSSatish Balay Level: advanced 610e5c89e4eSSatish Balay 611e5c89e4eSSatish Balay Fortran Note: 612e5c89e4eSSatish Balay The calling sequence in Fortran is PetscMallocDumpLog(integer ierr) 613e5c89e4eSSatish Balay The fp defaults to stdout. 614e5c89e4eSSatish Balay 615e5c89e4eSSatish Balay .seealso: PetscMallocGetCurrentUsage(), PetscMallocDump(), PetscMallocSetDumpLog() 616e5c89e4eSSatish Balay @*/ 6177087cfbeSBarry Smith PetscErrorCode PetscMallocDumpLog(FILE *fp) 618e5c89e4eSSatish Balay { 619e5c89e4eSSatish Balay PetscInt i,j,n,dummy,*perm; 620e5c89e4eSSatish Balay size_t *shortlength; 621f56c2debSBarry Smith int *shortcount,err; 622e5c89e4eSSatish Balay PetscMPIInt rank,size,tag = 1212 /* very bad programming */; 623ace3abfcSBarry Smith PetscBool match; 624e5c89e4eSSatish Balay const char **shortfunction; 625e5c89e4eSSatish Balay PetscLogDouble rss; 626e5c89e4eSSatish Balay MPI_Status status; 627e5c89e4eSSatish Balay PetscErrorCode ierr; 628e5c89e4eSSatish Balay 629e5c89e4eSSatish Balay PetscFunctionBegin; 630e5c89e4eSSatish Balay ierr = MPI_Comm_rank(MPI_COMM_WORLD,&rank);CHKERRQ(ierr); 631e5c89e4eSSatish Balay ierr = MPI_Comm_size(MPI_COMM_WORLD,&size);CHKERRQ(ierr); 632e5c89e4eSSatish Balay /* 633e5c89e4eSSatish Balay Try to get the data printed in order by processor. This will only sometimes work 634e5c89e4eSSatish Balay */ 635f56c2debSBarry Smith err = fflush(fp); 636e32f2f54SBarry Smith if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file"); 637f56c2debSBarry Smith 638e5c89e4eSSatish Balay ierr = MPI_Barrier(MPI_COMM_WORLD);CHKERRQ(ierr); 639e5c89e4eSSatish Balay if (rank) { 640e5c89e4eSSatish Balay ierr = MPI_Recv(&dummy,1,MPIU_INT,rank-1,tag,MPI_COMM_WORLD,&status);CHKERRQ(ierr); 641e5c89e4eSSatish Balay } 642e5c89e4eSSatish Balay 643768aa557SSatish Balay if (PetscLogMalloc < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"PetscMallocDumpLog() called without call to PetscMallocSetDumpLog() this is often due to\n setting the option -malloc_log AFTER PetscInitialize() with PetscOptionsInsert() or PetscOptionsInsertFile()"); 644768aa557SSatish Balay 645da9f1d6bSBarry Smith if (!fp) fp = PETSC_STDOUT; 646f3d65365SJed Brown ierr = PetscMemoryGetMaximumUsage(&rss);CHKERRQ(ierr); 647e5c89e4eSSatish Balay if (rss) { 648f3d65365SJed Brown ierr = PetscFPrintf(MPI_COMM_WORLD,fp,"[%d] Maximum memory PetscMalloc()ed %.0f maximum size of entire process %.0f\n",rank,(PetscLogDouble)TRMaxMem,rss);CHKERRQ(ierr); 649e5c89e4eSSatish Balay } else { 650e5c89e4eSSatish Balay ierr = PetscFPrintf(MPI_COMM_WORLD,fp,"[%d] Maximum memory PetscMalloc()ed %.0f OS cannot compute size of entire process\n",rank,(PetscLogDouble)TRMaxMem);CHKERRQ(ierr); 651e5c89e4eSSatish Balay } 652e32f2f54SBarry Smith shortcount = (int*)malloc(PetscLogMalloc*sizeof(int));if (!shortcount) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM,"Out of memory"); 653e32f2f54SBarry Smith shortlength = (size_t*)malloc(PetscLogMalloc*sizeof(size_t));if (!shortlength) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM,"Out of memory"); 654e32f2f54SBarry Smith shortfunction = (const char**)malloc(PetscLogMalloc*sizeof(char *));if (!shortfunction) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM,"Out of memory"); 65597b9d747SJed Brown for (i=0,n=0; i<PetscLogMalloc; i++) { 656e5c89e4eSSatish Balay for (j=0; j<n; j++) { 657e5c89e4eSSatish Balay ierr = PetscStrcmp(shortfunction[j],PetscLogMallocFunction[i],&match);CHKERRQ(ierr); 658e5c89e4eSSatish Balay if (match) { 659e5c89e4eSSatish Balay shortlength[j] += PetscLogMallocLength[i]; 66059ffdab8SBarry Smith shortcount[j]++; 661e5c89e4eSSatish Balay goto foundit; 662e5c89e4eSSatish Balay } 663e5c89e4eSSatish Balay } 664e5c89e4eSSatish Balay shortfunction[n] = PetscLogMallocFunction[i]; 665e5c89e4eSSatish Balay shortlength[n] = PetscLogMallocLength[i]; 66659ffdab8SBarry Smith shortcount[n] = 1; 667e5c89e4eSSatish Balay n++; 668e5c89e4eSSatish Balay foundit:; 669e5c89e4eSSatish Balay } 670e5c89e4eSSatish Balay 671e32f2f54SBarry Smith perm = (PetscInt*)malloc(n*sizeof(PetscInt));if (!perm) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM,"Out of memory"); 672e5c89e4eSSatish Balay for (i=0; i<n; i++) perm[i] = i; 673e5c89e4eSSatish Balay ierr = PetscSortStrWithPermutation(n,(const char **)shortfunction,perm);CHKERRQ(ierr); 674e5c89e4eSSatish Balay 675e5c89e4eSSatish Balay ierr = PetscFPrintf(MPI_COMM_WORLD,fp,"[%d] Memory usage sorted by function\n",rank);CHKERRQ(ierr); 676e5c89e4eSSatish Balay for (i=0; i<n; i++) { 67759ffdab8SBarry Smith ierr = PetscFPrintf(MPI_COMM_WORLD,fp,"[%d] %d %.0f %s()\n",rank,shortcount[perm[i]],(PetscLogDouble)shortlength[perm[i]],shortfunction[perm[i]]);CHKERRQ(ierr); 678e5c89e4eSSatish Balay } 679e5c89e4eSSatish Balay free(perm); 680e5c89e4eSSatish Balay free(shortlength); 68159ffdab8SBarry Smith free(shortcount); 682e5c89e4eSSatish Balay free((char **)shortfunction); 683f56c2debSBarry Smith err = fflush(fp); 684e32f2f54SBarry Smith if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file"); 685e5c89e4eSSatish Balay if (rank != size-1) { 686e5c89e4eSSatish Balay ierr = MPI_Send(&dummy,1,MPIU_INT,rank+1,tag,MPI_COMM_WORLD);CHKERRQ(ierr); 687e5c89e4eSSatish Balay } 688e5c89e4eSSatish Balay PetscFunctionReturn(0); 689e5c89e4eSSatish Balay } 690e5c89e4eSSatish Balay 691e5c89e4eSSatish Balay /* ---------------------------------------------------------------------------- */ 692e5c89e4eSSatish Balay 693e5c89e4eSSatish Balay #undef __FUNCT__ 694e5c89e4eSSatish Balay #define __FUNCT__ "PetscMallocDebug" 695e5c89e4eSSatish Balay /*@C 696e5c89e4eSSatish Balay PetscMallocDebug - Turns on/off debugging for the memory management routines. 697e5c89e4eSSatish Balay 698e5c89e4eSSatish Balay Not Collective 699e5c89e4eSSatish Balay 700e5c89e4eSSatish Balay Input Parameter: 701e5c89e4eSSatish Balay . level - PETSC_TRUE or PETSC_FALSE 702e5c89e4eSSatish Balay 703e5c89e4eSSatish Balay Level: intermediate 704e5c89e4eSSatish Balay 705e5c89e4eSSatish Balay .seealso: CHKMEMQ(), PetscMallocValidate() 706e5c89e4eSSatish Balay @*/ 7077087cfbeSBarry Smith PetscErrorCode PetscMallocDebug(PetscBool level) 708e5c89e4eSSatish Balay { 709e5c89e4eSSatish Balay PetscFunctionBegin; 710e5c89e4eSSatish Balay TRdebugLevel = level; 711e5c89e4eSSatish Balay PetscFunctionReturn(0); 712e5c89e4eSSatish Balay } 713