d4afb16f5a
This commit rewrites both the mpool and rcache frameworks. Summary of changes: - Before this change a significant portion of the rcache functionality lived in mpool components. This meant that it was impossible to add a new memory pool to use with rdma networks (ugni, openib, etc) without duplicating the functionality of an existing mpool component. All the registration functionality has been removed from the mpool and placed in the rcache framework. - All registration cache mpools components (udreg, grdma, gpusm, rgpusm) have been changed to rcache components. rcaches are allocated and released in the same way mpool components were. - It is now valid to pass NULL as the resources argument when creating an rcache. At this time the gpusm and rgpusm components support this. All other rcache components require non-NULL resources. - A new mpool component has been added: hugepage. This component supports huge page allocations on linux. - Memory pools are now allocated using "hints". Each mpool component is queried with the hints and returns a priority. The current hints supported are NULL (uses posix_memalign/malloc), page_size=x (huge page mpool), and mpool=x. - The sm mpool has been moved to common/sm. This reflects that the sm mpool is specialized and not meant for any general allocations. This mpool may be moved back into the mpool framework if there is any objection. - The opal_free_list_init arguments have been updated. The unused0 argument is not used to pass in the registration cache module. The mpool registration flags are now rcache registration flags. - All components have been updated to make use of the new framework interfaces. As this commit makes significant changes to both the mpool and rcache frameworks both versions have been bumped to 3.0.0. Signed-off-by: Nathan Hjelm <hjelmn@lanl.gov>
112 строки
5.1 KiB
C
112 строки
5.1 KiB
C
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil -*- */
|
|
/*
|
|
* Copyright (c) 2004-2006 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2004-2013 The University of Tennessee and The University
|
|
* of Tennessee Research Foundation. All rights
|
|
* reserved.
|
|
* Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
|
|
* University of Stuttgart. All rights reserved.
|
|
* Copyright (c) 2004-2006 The Regents of the University of California.
|
|
* All rights reserved.
|
|
* Copyright (c) 2011-2015 NVIDIA Corporation. All rights reserved.
|
|
* Copyright (c) 2015 Los Alamos National Security, LLC. All rights
|
|
* reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#ifndef OPAL_MCA_COMMON_CUDA_H
|
|
#define OPAL_MCA_COMMON_CUDA_H
|
|
#include "opal/mca/btl/btl.h"
|
|
#include "opal/datatype/opal_convertor.h"
|
|
|
|
#define MEMHANDLE_SIZE 8
|
|
#define EVTHANDLE_SIZE 8
|
|
|
|
struct mca_rcache_common_cuda_reg_data_t {
|
|
uint64_t memHandle[MEMHANDLE_SIZE];
|
|
uint64_t evtHandle[EVTHANDLE_SIZE];
|
|
uint64_t event;
|
|
opal_ptr_t memh_seg_addr;
|
|
size_t memh_seg_len;
|
|
};
|
|
typedef struct mca_rcache_common_cuda_reg_data_t mca_rcache_common_cuda_reg_data_t;
|
|
|
|
struct mca_rcache_common_cuda_reg_t {
|
|
mca_rcache_base_registration_t base;
|
|
mca_rcache_common_cuda_reg_data_t data;
|
|
};
|
|
typedef struct mca_rcache_common_cuda_reg_t mca_rcache_common_cuda_reg_t;
|
|
extern bool mca_common_cuda_enabled;
|
|
|
|
OPAL_DECLSPEC void mca_common_cuda_register_mca_variables(void);
|
|
|
|
OPAL_DECLSPEC void mca_common_cuda_register(void *ptr, size_t amount, char *msg);
|
|
|
|
OPAL_DECLSPEC void mca_common_cuda_unregister(void *ptr, char *msg);
|
|
|
|
OPAL_DECLSPEC void mca_common_wait_stream_synchronize(mca_rcache_common_cuda_reg_t *rget_reg);
|
|
|
|
OPAL_DECLSPEC int mca_common_cuda_memcpy(void *dst, void *src, size_t amount, char *msg,
|
|
struct mca_btl_base_descriptor_t *, int *done);
|
|
|
|
OPAL_DECLSPEC int mca_common_cuda_record_ipc_event(char *msg,
|
|
struct mca_btl_base_descriptor_t *frag);
|
|
OPAL_DECLSPEC int mca_common_cuda_record_dtoh_event(char *msg,
|
|
struct mca_btl_base_descriptor_t *frag);
|
|
OPAL_DECLSPEC int mca_common_cuda_record_htod_event(char *msg,
|
|
struct mca_btl_base_descriptor_t *frag);
|
|
|
|
OPAL_DECLSPEC void *mca_common_cuda_get_dtoh_stream(void);
|
|
OPAL_DECLSPEC void *mca_common_cuda_get_htod_stream(void);
|
|
|
|
OPAL_DECLSPEC int progress_one_cuda_ipc_event(struct mca_btl_base_descriptor_t **);
|
|
OPAL_DECLSPEC int progress_one_cuda_dtoh_event(struct mca_btl_base_descriptor_t **);
|
|
OPAL_DECLSPEC int progress_one_cuda_htod_event(struct mca_btl_base_descriptor_t **);
|
|
|
|
OPAL_DECLSPEC int mca_common_cuda_memhandle_matches(mca_rcache_common_cuda_reg_t *new_reg,
|
|
mca_rcache_common_cuda_reg_t *old_reg);
|
|
|
|
OPAL_DECLSPEC void mca_common_cuda_construct_event_and_handle(uintptr_t *event, void *handle);
|
|
OPAL_DECLSPEC void mca_common_cuda_destruct_event(uintptr_t event);
|
|
|
|
OPAL_DECLSPEC int cuda_getmemhandle(void *base, size_t, mca_rcache_base_registration_t *newreg,
|
|
mca_rcache_base_registration_t *hdrreg);
|
|
OPAL_DECLSPEC int cuda_ungetmemhandle(void *reg_data, mca_rcache_base_registration_t *reg);
|
|
OPAL_DECLSPEC int cuda_openmemhandle(void *base, size_t size, mca_rcache_base_registration_t *newreg,
|
|
mca_rcache_base_registration_t *hdrreg);
|
|
OPAL_DECLSPEC int cuda_closememhandle(void *reg_data, mca_rcache_base_registration_t *reg);
|
|
OPAL_DECLSPEC int mca_common_cuda_get_device(int *devicenum);
|
|
OPAL_DECLSPEC int mca_common_cuda_device_can_access_peer(int *access, int dev1, int dev2);
|
|
OPAL_DECLSPEC int mca_common_cuda_stage_one_init(void);
|
|
OPAL_DECLSPEC int mca_common_cuda_get_address_range(void *pbase, size_t *psize, void *base);
|
|
OPAL_DECLSPEC void mca_common_cuda_fini(void);
|
|
#if OPAL_CUDA_GDR_SUPPORT
|
|
OPAL_DECLSPEC bool mca_common_cuda_previously_freed_memory(mca_rcache_base_registration_t *reg);
|
|
OPAL_DECLSPEC void mca_common_cuda_get_buffer_id(mca_rcache_base_registration_t *reg);
|
|
#endif /* OPAL_CUDA_GDR_SUPPORT */
|
|
/**
|
|
* Return: 0 if no packing is required for sending (the upper layer
|
|
* can use directly the pointer to the contiguous user
|
|
* buffer).
|
|
* 1 if data does need to be packed, i.e. heterogeneous peers
|
|
* (source arch != dest arch) or non contiguous memory
|
|
* layout.
|
|
*/
|
|
static inline int32_t opal_convertor_cuda_need_buffers( opal_convertor_t* pConvertor )
|
|
{
|
|
int32_t retval;
|
|
uint32_t cudaflag = pConvertor->flags & CONVERTOR_CUDA; /* Save CUDA flag */
|
|
pConvertor->flags &= ~CONVERTOR_CUDA; /* Clear CUDA flag if it exists */
|
|
retval = opal_convertor_need_buffers(pConvertor);
|
|
pConvertor->flags |= cudaflag; /* Restore CUDA flag */
|
|
return retval;
|
|
}
|
|
|
|
#endif /* OPAL_MCA_COMMON_CUDA_H */
|