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>
89 строки
3.1 KiB
C
89 строки
3.1 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_BUDDY_H
|
|
#define MCA_MEMHEAP_BUDDY_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>
|
|
|
|
#define BITS_PER_BYTE 8
|
|
#define __BITOPS_WORDSIZE 64
|
|
#define DEFAULT_HASHTABLE_SIZE 100
|
|
|
|
#define BITOP_WORD(nr) ((nr) / bits_per_long())
|
|
#define DIV_ROUND_UP(n,d) (((n) + (d) - 1) / (d))
|
|
#define BITS_TO_LONGS(nr) DIV_ROUND_UP(nr, BITS_PER_BYTE * sizeof(unsigned long))
|
|
#define __BITOPS_WORDS(bits) (((bits)+__BITOPS_WORDSIZE-1)/__BITOPS_WORDSIZE)
|
|
#define clear_bit(x,y) __clear_bit((x), (y))
|
|
#define set_bit(x,y) __set_bit((x), (y))
|
|
#define find_first_bit(addr, size) find_next_bit((addr), (size), 0)
|
|
|
|
BEGIN_C_DECLS
|
|
|
|
struct mca_memheap_buddy_heap_t {
|
|
unsigned long **bits; /** Part of the buddy allocator */
|
|
unsigned *num_free; /** Part of the buddy allocator */
|
|
unsigned max_order; /** Log2 of Maximal heap size, part of the allocator */
|
|
unsigned min_order; /** min alloc order */
|
|
void* symmetric_heap; /** Symmetric Heap */
|
|
opal_hash_table_t* symmetric_heap_hashtable; /** Pointer to the Symmetric heap used for moving on it */
|
|
};
|
|
typedef struct mca_memheap_buddy_heap_t mca_memheap_buddy_heap_t;
|
|
|
|
/* Structure for managing shmem symmetric heap */
|
|
struct mca_memheap_buddy_module_t {
|
|
mca_memheap_base_module_t super;
|
|
|
|
int priority; /** Module's Priority */
|
|
mca_memheap_buddy_heap_t heap;
|
|
mca_memheap_buddy_heap_t private_heap;
|
|
opal_mutex_t lock; /** Part of the buddy allocator */
|
|
};
|
|
typedef struct mca_memheap_buddy_module_t mca_memheap_buddy_module_t;
|
|
OSHMEM_DECLSPEC extern mca_memheap_buddy_module_t memheap_buddy;
|
|
|
|
/*
|
|
* Buddy interface.
|
|
* Please pay attention to the new differences in the interface.
|
|
*/
|
|
OSHMEM_DECLSPEC extern int mca_memheap_buddy_module_init(memheap_context_t *);
|
|
OSHMEM_DECLSPEC extern int mca_memheap_buddy_alloc(size_t, void**);
|
|
OSHMEM_DECLSPEC extern int mca_memheap_buddy_realloc(size_t, void*, void **);
|
|
OSHMEM_DECLSPEC extern int mca_memheap_buddy_align(size_t, size_t, void**);
|
|
OSHMEM_DECLSPEC extern int mca_memheap_buddy_free(void*);
|
|
OSHMEM_DECLSPEC extern int mca_memheap_buddy_finalize(void);
|
|
|
|
/* private alloc/free functions */
|
|
OSHMEM_DECLSPEC extern int mca_memheap_buddy_private_alloc(size_t, void**);
|
|
OSHMEM_DECLSPEC extern int mca_memheap_buddy_private_free(void*);
|
|
|
|
/**
|
|
* static/global variables support. Consider making it a separate component
|
|
*/
|
|
|
|
END_C_DECLS
|
|
|
|
#endif /* MCA_MEMHEAP_BUDDY_H */
|