/* * Copyright (c) 2004-2005 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$ * * Additional copyrights may follow * * $HEADER$ */ #ifndef MCA_BTL_IB_FRAG_H #define MCA_BTL_IB_FRAG_H #define MCA_BTL_IB_FRAG_ALIGN (8) #include "ompi_config.h" #include "btl_openib.h" #include #include "ompi/mca/mpool/openib/mpool_openib.h" #if defined(c_plusplus) || defined(__cplusplus) extern "C" { #endif OMPI_DECLSPEC OBJ_CLASS_DECLARATION(mca_btl_openib_frag_t); struct mca_btl_openib_header_t { mca_btl_base_tag_t tag; int16_t credits; int16_t rdma_credits; }; typedef struct mca_btl_openib_header_t mca_btl_openib_header_t; struct mca_btl_openib_footer_t { #if OMPI_ENABLE_DEBUG uint32_t seq; #endif union { uint32_t size; uint8_t buf[4]; } u; }; typedef struct mca_btl_openib_footer_t mca_btl_openib_footer_t; typedef enum { MCA_BTL_OPENIB_CONTROL_NOOP, MCA_BTL_OPENIB_CONTROL_RDMA } mca_btl_openib_control_t; struct mca_btl_openib_control_header_t { mca_btl_openib_control_t type; }; typedef struct mca_btl_openib_control_header_t mca_btl_openib_control_header_t; struct mca_btl_openib_eager_rdma_header_t { mca_btl_openib_control_header_t control; ompi_ptr_t rdma_start; uint32_t rkey; }; typedef struct mca_btl_openib_eager_rdma_header_t mca_btl_openib_eager_rdma_header_t; enum mca_btl_openib_frag_type_t { MCA_BTL_OPENIB_FRAG_EAGER, MCA_BTL_OPENIB_FRAG_MAX, MCA_BTL_OPENIB_FRAG_FRAG, MCA_BTL_OPENIB_FRAG_EAGER_RDMA }; typedef enum mca_btl_openib_frag_type_t mca_btl_openib_frag_type_t; /** * IB send fragment derived type. */ struct mca_btl_openib_frag_t { mca_btl_base_descriptor_t base; mca_btl_base_segment_t segment; struct mca_btl_base_endpoint_t *endpoint; size_t size; int rc; mca_btl_openib_frag_type_t type; union{ struct ibv_recv_wr rd_desc; struct ibv_send_wr sr_desc; } wr_desc; struct ibv_sge sg_entry; struct ibv_mr *mr; mca_btl_openib_header_t *hdr; mca_btl_openib_footer_t *ftr; mca_mpool_openib_registration_t * openib_reg; }; typedef struct mca_btl_openib_frag_t mca_btl_openib_frag_t; OBJ_CLASS_DECLARATION(mca_btl_openib_frag_t); typedef struct mca_btl_openib_frag_t mca_btl_openib_send_frag_eager_t; OBJ_CLASS_DECLARATION(mca_btl_openib_send_frag_eager_t); typedef struct mca_btl_openib_frag_t mca_btl_openib_send_frag_max_t; OBJ_CLASS_DECLARATION(mca_btl_openib_send_frag_max_t); typedef struct mca_btl_openib_frag_t mca_btl_openib_send_frag_frag_t; OBJ_CLASS_DECLARATION(mca_btl_openib_send_frag_frag_t); typedef struct mca_btl_openib_frag_t mca_btl_openib_recv_frag_eager_t; OBJ_CLASS_DECLARATION(mca_btl_openib_recv_frag_eager_t); typedef struct mca_btl_openib_frag_t mca_btl_openib_recv_frag_max_t; OBJ_CLASS_DECLARATION(mca_btl_openib_recv_frag_max_t); /* * Allocate an IB send descriptor * */ #define MCA_BTL_IB_FRAG_ALLOC_EAGER_WAIT(btl, frag, rc) \ { \ \ ompi_free_list_item_t *item; \ OMPI_FREE_LIST_WAIT(&((mca_btl_openib_module_t*)btl)->send_free_eager, item, rc); \ frag = (mca_btl_openib_frag_t*) item; \ } #define MCA_BTL_IB_FRAG_ALLOC_EAGER(btl, frag, rc) \ { \ \ ompi_free_list_item_t *item; \ OMPI_FREE_LIST_GET(&((mca_btl_openib_module_t*)btl)->send_free_eager, item, rc); \ frag = (mca_btl_openib_frag_t*) item; \ } #define MCA_BTL_IB_FRAG_ALLOC_MAX(btl, frag, rc) \ { \ \ ompi_free_list_item_t *item; \ OMPI_FREE_LIST_GET(&((mca_btl_openib_module_t*)btl)->send_free_max, item, rc); \ frag = (mca_btl_openib_frag_t*) item; \ } #define MCA_BTL_IB_FRAG_ALLOC_FRAG(btl, frag, rc) \ { \ \ ompi_free_list_item_t *item; \ OMPI_FREE_LIST_GET(&((mca_btl_openib_module_t*)btl)->send_free_frag, item, rc); \ frag = (mca_btl_openib_frag_t*) item; \ } #define MCA_BTL_IB_FRAG_RETURN(btl, frag) \ { do { \ ompi_free_list_t* my_list; \ switch(frag->type) { \ case MCA_BTL_OPENIB_FRAG_EAGER_RDMA: \ case MCA_BTL_OPENIB_FRAG_EAGER: \ my_list = &btl->send_free_eager; \ break; \ case MCA_BTL_OPENIB_FRAG_MAX: \ my_list = &btl->send_free_max; \ break; \ case MCA_BTL_OPENIB_FRAG_FRAG: \ my_list = &btl->send_free_frag; \ } \ OMPI_FREE_LIST_RETURN(my_list, (ompi_free_list_item_t*)(frag)); \ } while(0); \ } #define MCA_BTL_IB_FRAG_PROGRESS(frag) \ do { \ switch(frag->wr_desc.sr_desc.opcode) { \ case IBV_WR_SEND: \ if(OMPI_SUCCESS != mca_btl_openib_endpoint_send(frag->endpoint, frag)) { \ BTL_ERROR(("error in posting pending send\n")); \ } \ break; \ case IBV_WR_RDMA_WRITE: \ if(OMPI_SUCCESS != mca_btl_openib_put((mca_btl_base_module_t*) openib_btl, \ frag->endpoint, \ (mca_btl_base_descriptor_t*) frag)) { \ BTL_ERROR(("error in posting pending rdma write\n")); \ } \ break; \ case IBV_WR_RDMA_READ: \ if(OMPI_SUCCESS != mca_btl_openib_get((mca_btl_base_module_t *) openib_btl, \ frag->endpoint, \ (mca_btl_base_descriptor_t*) frag)) { \ BTL_ERROR(("error in posting pending rdma read\n")); \ } \ break; \ default: \ BTL_ERROR(("error in posting pending operation, invalide opcode %d\n", frag->wr_desc.sr_desc.opcode)); \ break; \ } \ } while (0) struct mca_btl_openib_module_t; #if defined(c_plusplus) || defined(__cplusplus) } #endif #endif