90fb58de4f
allocated from mpool memory (which is registered memory for RDMA transports) This is not a problem for a small jobs, but for a big number of ranks an amount of waisted memory is big. This commit was SVN r13921.
136 строки
4.2 KiB
C
136 строки
4.2 KiB
C
/*
|
|
* Copyright (c) 2004-2006 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2004-2005 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-2005 The Regents of the University of California.
|
|
* All rights reserved.
|
|
* Copyright (c) 2006 Sandia National Laboratories. All rights
|
|
* reserved.
|
|
* Copyright (c) 2006 Sun Microsystems, Inc. All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "btl_udapl.h"
|
|
#include "btl_udapl_frag.h"
|
|
#include "ompi/mca/mpool/rdma/mpool_rdma.h"
|
|
|
|
|
|
static void mca_btl_udapl_frag_common_constructor(mca_btl_udapl_frag_t* frag)
|
|
{
|
|
mca_btl_udapl_reg_t* reg =
|
|
(mca_btl_udapl_reg_t*)frag->base.super.registration;
|
|
|
|
#if OMPI_ENABLE_DEBUG
|
|
frag->base.des_src = NULL;
|
|
frag->base.des_src_cnt = 0;
|
|
frag->base.des_dst = NULL;
|
|
frag->base.des_dst_cnt = 0;
|
|
frag->base.des_flags = 0;
|
|
#endif
|
|
|
|
frag->registration = reg;
|
|
frag->segment.seg_addr.pval = (unsigned char*)frag->base.super.ptr;
|
|
frag->ftr = NULL;
|
|
|
|
/* Don't understand why yet, but there are cases where reg is NULL -
|
|
that is, this memory has not been registered. So be careful not
|
|
to dereference a NULL pointer. */
|
|
if(NULL != reg) {
|
|
/* Save the LMR context so we can set up LMR subset triplets later */
|
|
frag->triplet.lmr_context = reg->lmr_triplet.lmr_context;
|
|
}
|
|
}
|
|
|
|
static void mca_btl_udapl_frag_eager_constructor(mca_btl_udapl_frag_t* frag)
|
|
{
|
|
frag->segment.seg_len = mca_btl_udapl_module.super.btl_eager_limit;
|
|
frag->size = mca_btl_udapl_component.udapl_eager_frag_size;
|
|
mca_btl_udapl_frag_common_constructor(frag);
|
|
}
|
|
|
|
static void mca_btl_udapl_frag_max_constructor(mca_btl_udapl_frag_t* frag)
|
|
{
|
|
frag->segment.seg_len = mca_btl_udapl_module.super.btl_max_send_size;
|
|
frag->size = mca_btl_udapl_component.udapl_max_frag_size;
|
|
mca_btl_udapl_frag_common_constructor(frag);
|
|
}
|
|
|
|
static void mca_btl_udapl_frag_user_constructor(mca_btl_udapl_frag_t* frag)
|
|
{
|
|
mca_btl_udapl_frag_common_constructor(frag);
|
|
frag->segment.seg_len = 0;
|
|
frag->segment.seg_addr.pval = NULL;
|
|
frag->ftr = NULL;
|
|
frag->size = 0;
|
|
frag->registration = NULL;
|
|
}
|
|
|
|
static void mca_btl_udapl_frag_eager_rdma_constructor(mca_btl_udapl_frag_t* frag)
|
|
{
|
|
mca_btl_udapl_frag_eager_constructor(frag);
|
|
frag->segment.seg_len = mca_btl_udapl_module.super.btl_eager_limit;
|
|
frag->size = mca_btl_udapl_component.udapl_eager_frag_size;
|
|
frag->rdma_ftr = (mca_btl_udapl_rdma_footer_t *)
|
|
((char *)(frag->segment.seg_addr.pval) +
|
|
frag->size -
|
|
sizeof(mca_btl_udapl_rdma_footer_t));
|
|
frag->rdma_ftr->active=0;
|
|
}
|
|
|
|
static void mca_btl_udapl_frag_common_destructor(mca_btl_udapl_frag_t* frag)
|
|
{
|
|
#if OMPI_ENABLE_DEBUG
|
|
frag->ftr = NULL;
|
|
frag->size = 0;
|
|
frag->registration = NULL;
|
|
frag->segment.seg_len = 0;
|
|
frag->segment.seg_addr.pval = NULL;
|
|
|
|
frag->base.des_src = NULL;
|
|
frag->base.des_src_cnt = 0;
|
|
frag->base.des_dst = NULL;
|
|
frag->base.des_dst_cnt = 0;
|
|
frag->base.des_flags = 0;
|
|
#endif
|
|
}
|
|
|
|
|
|
OBJ_CLASS_INSTANCE(
|
|
mca_btl_udapl_frag_t,
|
|
mca_btl_base_descriptor_t,
|
|
NULL,
|
|
NULL);
|
|
|
|
OBJ_CLASS_INSTANCE(
|
|
mca_btl_udapl_frag_eager_t,
|
|
mca_btl_base_descriptor_t,
|
|
mca_btl_udapl_frag_eager_constructor,
|
|
mca_btl_udapl_frag_common_destructor);
|
|
|
|
OBJ_CLASS_INSTANCE(
|
|
mca_btl_udapl_frag_max_t,
|
|
mca_btl_base_descriptor_t,
|
|
mca_btl_udapl_frag_max_constructor,
|
|
mca_btl_udapl_frag_common_destructor);
|
|
|
|
OBJ_CLASS_INSTANCE(
|
|
mca_btl_udapl_frag_user_t,
|
|
mca_btl_base_descriptor_t,
|
|
mca_btl_udapl_frag_user_constructor,
|
|
NULL);
|
|
|
|
OBJ_CLASS_INSTANCE(
|
|
mca_btl_udapl_frag_eager_rdma_t,
|
|
mca_btl_base_descriptor_t,
|
|
mca_btl_udapl_frag_eager_rdma_constructor,
|
|
mca_btl_udapl_frag_common_destructor);
|