1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
/**
* Copyright (c) 2013 Mellanox Technologies, Inc.
* All rights reserved.
* $COPYRIGHT$
*
* Additional copyrights may follow
*
* $HEADER$
*/
/**
* @file
* Description of the Registration Cache framework
*/
#ifndef MCA_MEMHEAP_PTMALLOC_H
#define MCA_MEMHEAP_PTMALLOC_H
#include "oshmem_config.h"
#include "oshmem/mca/mca.h"
#include "opal/class/opal_list.h"
#include "opal/mca/threads/mutex.h"
#include "oshmem/mca/memheap/memheap.h"
#include "oshmem/mca/memheap/base/base.h"
#include "oshmem/mca/spml/spml.h"
#include "oshmem/util/oshmem_util.h"
#include "opal/class/opal_hash_table.h"
#include "opal/mca/btl/btl.h"
#include <string.h>
#include <sys/types.h>
#include <math.h>
BEGIN_C_DECLS
#include "malloc_defs.h"
/*
* At the moment we use only dlmalloc part of the ptmalloc3. Thread safety is implemented by using locks on
* alloc operations. Since all shmem alloc ops are collectives, malloc performance is not a problem. So it makes
* sense to use simpler algorithm.
*
* Heap is allocate in one chunk, and we implement our on sbrk like function that serves portions of the memory
* to malloc.
*
* At the moment we do not support growing/returning heap based memory to OS.
*/
/* Structure for managing shmem symmetric heap */
struct mca_memheap_ptmalloc_module_t {
mca_memheap_base_module_t super;
int priority; /** Module's Priority */
void *base;
size_t cur_size;
size_t max_size;
size_t max_alloc_size;
opal_mutex_t lock; /** Part of the allocator */
};
typedef struct mca_memheap_ptmalloc_module_t mca_memheap_ptmalloc_module_t;
OSHMEM_DECLSPEC extern mca_memheap_ptmalloc_module_t memheap_ptmalloc;
/*
* Buddy interface.
* Please pay attention to the new differences in the interface.
*/
OSHMEM_DECLSPEC extern int mca_memheap_ptmalloc_module_init(memheap_context_t *);
OSHMEM_DECLSPEC extern int mca_memheap_ptmalloc_alloc(size_t, void**);
OSHMEM_DECLSPEC extern int mca_memheap_ptmalloc_realloc(size_t, void*, void **);
OSHMEM_DECLSPEC extern int mca_memheap_ptmalloc_align(size_t, size_t, void**);
OSHMEM_DECLSPEC extern int mca_memheap_ptmalloc_free(void*);
OSHMEM_DECLSPEC extern int mca_memheap_ptmalloc_finalize(void);
END_C_DECLS
#endif /* MCA_MEMHEAP_BUDDY_H */
|
