ee3517427e
Add a framework to support different types of threading models including user space thread packages such as Qthreads and argobot: https://github.com/pmodels/argobots https://github.com/Qthreads/qthreads The default threading model is pthreads. Alternate thread models are specificed at configure time using the --with-threads=X option. The framework is static. The theading model to use is selected at Open MPI configure/build time. mca/threads: implement Argobots threading layer config: fix thread configury - Add double quotations - Change Argobot to Argobots config: implement Argobots check If the poll time is too long, MPI hangs. This quick fix just sets it to 0, but it is not good for the Pthreads version. Need to find a good way to abstract it. Note that even 1 (= 1 millisecond) causes disastrous performance degradation. rework threads MCA framework configury It now works more like the ompi/mca/rte configury, modulo some edge items that are special for threading package linking, etc. qthreads module some argobots cleanup Signed-off-by: Noah Evans <noah.evans@gmail.com> Signed-off-by: Shintaro Iwasaki <siwasaki@anl.gov> Signed-off-by: Howard Pritchard <howardp@lanl.gov>
73 строки
2.2 KiB
C
73 строки
2.2 KiB
C
/**
|
|
* 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 */
|