5f1254d710
Use of the old ompi_free_list_t and ompi_free_list_item_t is deprecated. These classes will be removed in a future commit. This commit updates the entire code base to use opal_free_list_t and opal_free_list_item_t. Notes: OMPI_FREE_LIST_*_MT -> opal_free_list_* (uses opal_using_threads ()) Signed-off-by: Nathan Hjelm <hjelmn@lanl.gov>
123 строки
4.4 KiB
C
123 строки
4.4 KiB
C
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil -*- */
|
|
/*
|
|
* Copyright (c) 2004-2005 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-2005 The Regents of the University of California.
|
|
* All rights reserved.
|
|
* Copyright (c) 2008 Sun Microsystems, Inc. All rights reserved.
|
|
* Copyright (c) 2009 Cisco Systems, Inc. All rights reserved.
|
|
* Copyright (c) 2012 NVIDIA Corporation. All rights reserved.
|
|
* Copyright (c) 2015 Los Alamos National Security, LLC. All rights
|
|
* reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
/**
|
|
* @file
|
|
*/
|
|
#ifndef MCA_BTL_SMCUDA_SEND_FRAG_H
|
|
#define MCA_BTL_SMCUDA_SEND_FRAG_H
|
|
|
|
#include "opal_config.h"
|
|
#include "btl_smcuda.h"
|
|
|
|
#if OPAL_CUDA_SUPPORT
|
|
#include "opal/mca/common/cuda/common_cuda.h"
|
|
#endif
|
|
|
|
#define MCA_BTL_SMCUDA_FRAG_TYPE_MASK ((uintptr_t)0x3)
|
|
#define MCA_BTL_SMCUDA_FRAG_SEND ((uintptr_t)0x0)
|
|
#define MCA_BTL_SMCUDA_FRAG_ACK ((uintptr_t)0x1)
|
|
#define MCA_BTL_SMCUDA_FRAG_PUT ((uintptr_t)0x2)
|
|
#define MCA_BTL_SMCUDA_FRAG_GET ((uintptr_t)0x3)
|
|
|
|
#define MCA_BTL_SMCUDA_FRAG_STATUS_MASK ((uintptr_t)0x4)
|
|
|
|
struct mca_btl_smcuda_frag_t;
|
|
|
|
struct mca_btl_smcuda_hdr_t {
|
|
struct mca_btl_smcuda_frag_t *frag;
|
|
size_t len;
|
|
int my_smp_rank;
|
|
mca_btl_base_tag_t tag;
|
|
};
|
|
typedef struct mca_btl_smcuda_hdr_t mca_btl_smcuda_hdr_t;
|
|
|
|
#if OPAL_CUDA_SUPPORT
|
|
struct mca_btl_base_registration_handle_t {
|
|
mca_mpool_common_cuda_reg_data_t reg_data;
|
|
};
|
|
#endif
|
|
|
|
struct mca_btl_smcuda_segment_t {
|
|
mca_btl_base_segment_t base;
|
|
#if OPAL_CUDA_SUPPORT
|
|
uint8_t key[128]; /* 64 bytes for CUDA mem handle, 64 bytes for CUDA event handle */
|
|
/** Address of the entire memory handle */
|
|
opal_ptr_t memh_seg_addr;
|
|
/** Length in bytes of entire memory handle */
|
|
uint32_t memh_seg_len;
|
|
#endif /* OPAL_CUDA_SUPPORT */
|
|
};
|
|
typedef struct mca_btl_smcuda_segment_t mca_btl_smcuda_segment_t;
|
|
|
|
/**
|
|
* shared memory send fragment derived type.
|
|
*/
|
|
struct mca_btl_smcuda_frag_t {
|
|
mca_btl_base_descriptor_t base;
|
|
mca_btl_base_segment_t segment;
|
|
struct mca_btl_base_endpoint_t *endpoint;
|
|
#if OPAL_CUDA_SUPPORT
|
|
struct mca_mpool_base_registration_t *registration;
|
|
struct mca_btl_base_registration_handle_t *local_handle;
|
|
#endif /* OPAL_CUDA_SUPPORT */
|
|
size_t size;
|
|
/* pointer written to the FIFO, this is the base of the shared memory region */
|
|
mca_btl_smcuda_hdr_t *hdr;
|
|
opal_free_list_t* my_list;
|
|
};
|
|
typedef struct mca_btl_smcuda_frag_t mca_btl_smcuda_frag_t;
|
|
typedef struct mca_btl_smcuda_frag_t mca_btl_smcuda_frag1_t;
|
|
typedef struct mca_btl_smcuda_frag_t mca_btl_smcuda_frag2_t;
|
|
typedef struct mca_btl_smcuda_frag_t mca_btl_smcuda_user_t;
|
|
|
|
|
|
OBJ_CLASS_DECLARATION(mca_btl_smcuda_frag_t);
|
|
OBJ_CLASS_DECLARATION(mca_btl_smcuda_frag1_t);
|
|
OBJ_CLASS_DECLARATION(mca_btl_smcuda_frag2_t);
|
|
OBJ_CLASS_DECLARATION(mca_btl_smcuda_user_t);
|
|
|
|
#define MCA_BTL_SMCUDA_FRAG_ALLOC_EAGER(frag) \
|
|
{ \
|
|
frag = (mca_btl_smcuda_frag_t *) \
|
|
opal_free_list_get (&mca_btl_smcuda_component.sm_frags_eager); \
|
|
}
|
|
|
|
#define MCA_BTL_SMCUDA_FRAG_ALLOC_MAX(frag) \
|
|
{ \
|
|
frag = (mca_btl_smcuda_frag_t *) \
|
|
opal_free_list_get (&mca_btl_smcuda_component.sm_frags_max); \
|
|
}
|
|
|
|
#define MCA_BTL_SMCUDA_FRAG_ALLOC_USER(frag) \
|
|
{ \
|
|
frag = (mca_btl_smcuda_frag_t *) \
|
|
opal_free_list_get (&mca_btl_smcuda_component.sm_frags_user); \
|
|
}
|
|
|
|
#define MCA_BTL_SMCUDA_FRAG_RETURN(frag) \
|
|
{ \
|
|
opal_free_list_return (frag->my_list, (opal_free_list_item_t*)(frag)); \
|
|
}
|
|
#endif
|