#include <assert.h>
#include <omp.h>
#include <stdbool.h>
#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "../offload/offload_library.h"
#include "../offload/offload_runtime.h"
#include "dbm_hyperparams.h"
#include "dbm_mempool.h"
#include "dbm_mpi.h"

Data Structures
struct	dbm_memchunk
	Private struct for storing a chunk of memory. More...

Typedefs
typedef struct dbm_memchunk	dbm_memchunk_t
	Private struct for storing a chunk of memory.

Functions
static void *	actual_malloc (size_t size, bool on_device)
	Private routine for actually allocating system memory.

static void	actual_free (const void *memory, bool on_device)
	Private routine for actually freeing system memory.

static void *	internal_mempool_malloc (dbm_memchunk_t available_head, dbm_memchunk_t allocated_head, size_t size)
	Private routine for allocating host or device memory from the pool.

void *	dbm_mempool_host_malloc (size_t size)
	Internal routine for allocating host memory from the pool.

void *	dbm_mempool_device_malloc (size_t size)
	Internal routine for allocating device memory from the pool.

static void	internal_mempool_free (dbm_memchunk_t available_head, dbm_memchunk_t allocated_head, const void *mem)
	Private routine for releasing memory back to the pool.

void	dbm_mempool_host_free (const void *memory)
	Internal routine for releasing memory back to the pool.

void	dbm_mempool_device_free (const void *memory)
	Internal routine for releasing memory back to the pool.

static void	internal_mempool_clear (dbm_memchunk_t **available_head)
	Private routine for freeing all memory in the pool.

void	dbm_mempool_clear (void)
	Internal routine for freeing all memory in the pool.

void	dbm_mempool_statistics (dbm_memstats_t *memstats)
	Internal routine to query statistics.

Variables
static dbm_memchunk_t *	mempool_device_available_head = NULL
	Private linked list of memory chunks that are available.

static dbm_memchunk_t *	mempool_host_available_head = NULL

static dbm_memchunk_t *	mempool_device_allocated_head = NULL
	Private linked list of memory chunks that are in use.

static dbm_memchunk_t *	mempool_host_allocated_head = NULL

static dbm_memstats_t	mempool_stats = {0}
	Private statistics.

Typedef Documentation

◆ dbm_memchunk_t

typedef struct dbm_memchunk dbm_memchunk_t

Private struct for storing a chunk of memory.

Author: Ole Schuett

Function Documentation

◆ actual_malloc()

static void * actual_malloc	(	size_t	size,
		bool	on_device
	)

static

Private routine for actually allocating system memory.

Author: Ole Schuett

Definition at line 56 of file dbm_mempool.c.

Here is the call graph for this function:

Here is the caller graph for this function:

◆ actual_free()

static void actual_free	(	const void *	memory,
		bool	on_device
	)

static

Private routine for actually freeing system memory.

Author: Ole Schuett

Definition at line 94 of file dbm_mempool.c.

Here is the call graph for this function:

Here is the caller graph for this function:

◆ internal_mempool_malloc()

static void * internal_mempool_malloc	(	dbm_memchunk_t **	available_head,
		dbm_memchunk_t **	allocated_head,
		size_t	size
	)

static

Private routine for allocating host or device memory from the pool.

Author: Ole Schuett

Definition at line 120 of file dbm_mempool.c.

Here is the call graph for this function:

Here is the caller graph for this function:

◆ dbm_mempool_host_malloc()

void * dbm_mempool_host_malloc ( size_t size )

Internal routine for allocating host memory from the pool.

Author: Ole Schuett

Definition at line 187 of file dbm_mempool.c.

Here is the call graph for this function:

Here is the caller graph for this function:

◆ dbm_mempool_device_malloc()

void * dbm_mempool_device_malloc ( size_t size )

Internal routine for allocating device memory from the pool.

Author: Ole Schuett

Definition at line 200 of file dbm_mempool.c.

Here is the call graph for this function:

◆ internal_mempool_free()

static void internal_mempool_free	(	dbm_memchunk_t **	available_head,
		dbm_memchunk_t **	allocated_head,
		const void *	mem
	)

static

Private routine for releasing memory back to the pool.

Author: Ole Schuett

Definition at line 218 of file dbm_mempool.c.

Here is the caller graph for this function:

◆ dbm_mempool_host_free()

void dbm_mempool_host_free ( const void * memory )

Internal routine for releasing memory back to the pool.

Author: Ole Schuett

Definition at line 246 of file dbm_mempool.c.

Here is the call graph for this function:

Here is the caller graph for this function:

◆ dbm_mempool_device_free()

void dbm_mempool_device_free ( const void * memory )

Internal routine for releasing memory back to the pool.

Author: Ole Schuett

Definition at line 259 of file dbm_mempool.c.

Here is the call graph for this function:

◆ internal_mempool_clear()

static void internal_mempool_clear ( dbm_memchunk_t ** available_head )

static

Private routine for freeing all memory in the pool.

Author: Ole Schuett

Definition at line 276 of file dbm_mempool.c.

Here is the call graph for this function:

Here is the caller graph for this function:

◆ dbm_mempool_clear()

void dbm_mempool_clear ( void )

Internal routine for freeing all memory in the pool.

Author: Ole Schuett

Definition at line 293 of file dbm_mempool.c.

Here is the call graph for this function:

Here is the caller graph for this function:

◆ dbm_mempool_statistics()

void dbm_mempool_statistics ( dbm_memstats_t * memstats )

Internal routine to query statistics.

Author: Hans Pabst

Definition at line 309 of file dbm_mempool.c.

Here is the caller graph for this function:

Variable Documentation

◆ mempool_device_available_head

dbm_memchunk_t* mempool_device_available_head = NULL

static

Private linked list of memory chunks that are available.

Author: Ole Schuett

Definition at line 36 of file dbm_mempool.c.

◆ mempool_host_available_head

dbm_memchunk_t* mempool_host_available_head = NULL

static

Definition at line 37 of file dbm_mempool.c.

◆ mempool_device_allocated_head

dbm_memchunk_t* mempool_device_allocated_head = NULL

static

Private linked list of memory chunks that are in use.

Author: Ole Schuett

Definition at line 43 of file dbm_mempool.c.

◆ mempool_host_allocated_head

dbm_memchunk_t* mempool_host_allocated_head = NULL

static

Definition at line 44 of file dbm_mempool.c.

◆ mempool_stats

dbm_memstats_t mempool_stats = {0}

static

Private statistics.

Author: Hans Pabst

Definition at line 50 of file dbm_mempool.c.

Data Structures

Typedefs

Functions

Variables

Typedef Documentation

◆ dbm_memchunk_t

Function Documentation

◆ actual_malloc()

◆ actual_free()

◆ internal_mempool_malloc()

◆ dbm_mempool_host_malloc()

◆ dbm_mempool_device_malloc()

◆ internal_mempool_free()

◆ dbm_mempool_host_free()

◆ dbm_mempool_device_free()

◆ internal_mempool_clear()

◆ dbm_mempool_clear()

◆ dbm_mempool_statistics()

Variable Documentation

◆ mempool_device_available_head

◆ mempool_host_available_head

◆ mempool_device_allocated_head

◆ mempool_host_allocated_head

◆ mempool_stats