xref: /petsc/src/sys/memory/mtr.c (revision 2c9581d29997335d805cf04b9d7ec09e8f4441b2)
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