CFD/hypre
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix memory tracker with omp (#844)

Browse Source 
This PR fixes hypre memory tracker with OMP threading.
This commit is contained in:
Rui Peng Li 2023-03-22 16:01:24 -07:00 committed by GitHub
parent 1eabaf5f33
commit 2391e4703f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 57 additions and 45 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/utilities/_hypre_utilities.h
View File

@ -914,6 +914,8 @@ typedef struct
hypre_MemoryTrackerQueue queue[HYPRE_MEMORY_NUM_EVENTS];
} hypre_MemoryTracker;
extern hypre_MemoryTracker *_hypre_memory_tracker;
#define hypre_TAlloc(type, count, location) \
( \
{ \

16
src/utilities/general.c
View File

@ -8,22 +8,6 @@
#include "_hypre_utilities.h"
#include "_hypre_utilities.hpp"
#ifdef HYPRE_USING_MEMORY_TRACKER
hypre_MemoryTracker *_hypre_memory_tracker = NULL;
/* accessor to the global ``_hypre_memory_tracker'' */
hypre_MemoryTracker*
hypre_memory_tracker(void)
{
if (!_hypre_memory_tracker)
{
_hypre_memory_tracker = hypre_MemoryTrackerCreate();
}
return _hypre_memory_tracker;
}
#endif
/* global variable _hypre_handle:
* Outside this file, do NOT access it directly,
* but use hypre_handle() instead (see handle.h) */

82
src/utilities/memory_tracker.c
View File

@ -16,6 +16,25 @@
#if defined(HYPRE_USING_MEMORY_TRACKER)
hypre_MemoryTracker *_hypre_memory_tracker = NULL;
/* accessor to the global ``_hypre_memory_tracker'' */
hypre_MemoryTracker*
hypre_memory_tracker(void)
{
#ifdef HYPRE_USING_OPENMP
#pragma omp critical
#endif
{
if (!_hypre_memory_tracker)
{
_hypre_memory_tracker = hypre_MemoryTrackerCreate();
}
}
return _hypre_memory_tracker;
}
size_t hypre_total_bytes[hypre_NUM_MEMORY_LOCATION];
size_t hypre_peak_bytes[hypre_NUM_MEMORY_LOCATION];
size_t hypre_current_bytes[hypre_NUM_MEMORY_LOCATION];
@ -252,44 +271,49 @@ hypre_MemoryTrackerInsert2(const char *action,
hypre_MemoryTrackerQueue *queue = &tracker->queue[q];
/* resize if not enough space */
if (queue->alloced_size <= queue->actual_size)
#ifdef HYPRE_USING_OPENMP
#pragma omp critical
#endif
{
queue->alloced_size = 2 * queue->alloced_size + 1;
queue->data = (hypre_MemoryTrackerEntry *) realloc(queue->data,
queue->alloced_size * sizeof(hypre_MemoryTrackerEntry));
}
/* resize if not enough space */
hypre_assert(queue->actual_size < queue->alloced_size);
if (queue->alloced_size <= queue->actual_size)
{
queue->alloced_size = 2 * queue->alloced_size + 1;
queue->data = (hypre_MemoryTrackerEntry *) realloc(queue->data,
queue->alloced_size * sizeof(hypre_MemoryTrackerEntry));
}
/* insert an entry */
hypre_MemoryTrackerEntry *entry = queue->data + queue->actual_size;
hypre_assert(queue->actual_size < queue->alloced_size);
entry->index = queue->actual_size;
entry->time_step = tracker->curr_time_step;
sprintf(entry->action, "%s", action);
entry->ptr = ptr;
entry->ptr2 = ptr2;
entry->nbytes = nbytes;
entry->memory_location = memory_location;
entry->memory_location2 = memory_location2;
sprintf(entry->filename, "%s", filename);
sprintf(entry->function, "%s", function);
entry->line = line;
entry->pair = (size_t) -1;
/* insert an entry */
hypre_MemoryTrackerEntry *entry = queue->data + queue->actual_size;
entry->index = queue->actual_size;
entry->time_step = tracker->curr_time_step;
sprintf(entry->action, "%s", action);
entry->ptr = ptr;
entry->ptr2 = ptr2;
entry->nbytes = nbytes;
entry->memory_location = memory_location;
entry->memory_location2 = memory_location2;
sprintf(entry->filename, "%s", filename);
sprintf(entry->function, "%s", function);
entry->line = line;
entry->pair = (size_t) -1;
#if 0
HYPRE_Int myid;
hypre_MPI_Comm_rank(hypre_MPI_COMM_WORLD, &myid);
if (myid == 0 && entry->time_step == 28111) {assert(0);}
HYPRE_Int myid;
hypre_MPI_Comm_rank(hypre_MPI_COMM_WORLD, &myid);
if (myid == 0 && entry->time_step == 28111) {assert(0);}
#endif
/* increase the time step */
tracker->curr_time_step ++;
/* increase the time step */
tracker->curr_time_step ++;
/* increase the queue length by 1 */
queue->actual_size ++;
/* increase the queue length by 1 */
queue->actual_size ++;
}
}
HYPRE_Int

2
src/utilities/memory_tracker.h
View File

@ -69,6 +69,8 @@ typedef struct
hypre_MemoryTrackerQueue queue[HYPRE_MEMORY_NUM_EVENTS];
} hypre_MemoryTracker;
extern hypre_MemoryTracker *_hypre_memory_tracker;
#define hypre_TAlloc(type, count, location) \
( \
{ \