1143 строки
44 KiB
C
1143 строки
44 KiB
C
|
#define IB
|
||
|
#include "ib.h"
|
||
|
|
||
|
double t;
|
||
|
|
||
|
/* is IB initialized? */
|
||
|
static int IB_Ginitialized=0;
|
||
|
/* the keyval (global) */
|
||
|
static int IB_Gkeyval=MPI_KEYVAL_INVALID;
|
||
|
|
||
|
static struct {
|
||
|
VAPI_hca_hndl_t hca_hndl;
|
||
|
VAPI_pd_hndl_t pd_hndl;
|
||
|
|
||
|
hb_tree *memlist; /* this is the libdict structure to hang off the search tree */
|
||
|
} IB_Hca_info;
|
||
|
|
||
|
/* function definitions */
|
||
|
static __inline__ void IB_Memlist_memlist_delete(IB_Memlistel *entry);
|
||
|
static __inline__ void IB_Memlist_delete_key(IB_Memlistel *k);
|
||
|
static __inline__ int IB_Memlist_compare_entries(IB_Memlistel *a, IB_Memlistel *b, void *param);
|
||
|
static __inline__ void IB_Taglist_delete(IB_Taglstel *entry);
|
||
|
static __inline__ void IB_Taglist_delete_key(IB_Taglstel *k);
|
||
|
static __inline__ int IB_Taglist_compare_entries(IB_Taglstel *a, IB_Taglstel *b, void *param);
|
||
|
|
||
|
#if 0
|
||
|
static __inline__ int IB_Addtotaglst(IB_Comminfo *comminfo, int tag, IB_Req *req, int peer) {
|
||
|
IB_Taglstel *new;
|
||
|
|
||
|
new = malloc(sizeof(IB_Taglstel));
|
||
|
new->tag = tag;
|
||
|
new->peer = peer;
|
||
|
new->req = req;
|
||
|
new->next = NULL;
|
||
|
|
||
|
/* first element in list */
|
||
|
if(comminfo->taglistend == NULL) {
|
||
|
comminfo->taglisthead = new;
|
||
|
} else {
|
||
|
comminfo->taglistend->next = new;
|
||
|
}
|
||
|
//printf("added tag %i/peer %i as new element to taglist\n", tag, peer);
|
||
|
comminfo->taglistend = new;
|
||
|
|
||
|
return IB_OK;
|
||
|
}
|
||
|
|
||
|
static __inline__ IB_Req *IB_Gettagreq(IB_Comminfo *comminfo, int tag, int peer) {
|
||
|
IB_Taglstel *ptr, *tmp;
|
||
|
IB_Req *req;
|
||
|
|
||
|
/* empty list */
|
||
|
if(comminfo->taglisthead == NULL) return NULL;
|
||
|
|
||
|
ptr = comminfo->taglisthead;
|
||
|
/* tag/peer is in first element */
|
||
|
if((ptr->tag == tag) && (ptr->peer == peer)) {
|
||
|
req = comminfo->taglisthead->req;
|
||
|
comminfo->taglisthead = ptr->next;
|
||
|
/* list is empty */
|
||
|
if(comminfo->taglisthead == NULL) comminfo->taglistend = NULL;
|
||
|
free(ptr);
|
||
|
//printf("removed tag %i\n", tag);
|
||
|
return req;
|
||
|
}
|
||
|
|
||
|
while ((ptr->next != NULL)) {
|
||
|
if((ptr->next->tag == tag) && (ptr->next->peer == peer)) break;
|
||
|
ptr = ptr->next;
|
||
|
}
|
||
|
|
||
|
if(ptr->next == NULL)
|
||
|
/* we did not find it */
|
||
|
return NULL;
|
||
|
else {
|
||
|
/* we found it somewhere in the middle */
|
||
|
req = ptr->next->req;
|
||
|
tmp = ptr->next;
|
||
|
ptr->next = ptr->next->next;
|
||
|
/* it was the last element */
|
||
|
if(ptr->next == NULL) comminfo->taglistend = ptr;
|
||
|
free(tmp);
|
||
|
//printf("removed tag %i\n", tag);
|
||
|
return req;
|
||
|
}
|
||
|
}
|
||
|
#endif
|
||
|
|
||
|
static int IB_Create_qp( int rank, int remote, VAPI_hca_hndl_t *hca_hndl_p, VAPI_cq_hndl_t *sr_cq_hndl_p, VAPI_cq_hndl_t *rr_cq_hndl_p, VAPI_qp_hndl_t *qp_hndl_p, VAPI_pd_hndl_t *pd_hndp_p, MPI_Comm comm);
|
||
|
|
||
|
static int IB_Key_copy(MPI_Comm oldcomm, int keyval, void *extra_state, void *attribute_val_in, void *attribute_val_out, int *flag) {
|
||
|
/* delete the attribute in the new comm - it will be created at the
|
||
|
* first usage */
|
||
|
*flag = 0;
|
||
|
|
||
|
return MPI_SUCCESS;
|
||
|
}
|
||
|
|
||
|
static int IB_Key_delete(MPI_Comm comm, int keyval, void *attribute_val, void *extra_state) {
|
||
|
IB_Comminfo *comminfo;
|
||
|
|
||
|
if(keyval == IB_Gkeyval) {
|
||
|
comminfo=(IB_Comminfo*)attribute_val;
|
||
|
free(comminfo);
|
||
|
} else {
|
||
|
printf("Got wrong keyval!(%i)\n", keyval);
|
||
|
}
|
||
|
|
||
|
return MPI_SUCCESS;
|
||
|
}
|
||
|
|
||
|
static __inline__ void IB_stat( VAPI_ret_t ret, char *string ) {
|
||
|
int rank;
|
||
|
|
||
|
if( ret != VAPI_OK) {
|
||
|
MPI_Comm_rank( MPI_COMM_WORLD, &rank );
|
||
|
printf("[%u]: *** [ERROR] *** %s -> %s (%s)\n", rank, string, VAPI_strerror( ret ), VAPI_strerror_sym( ret ));
|
||
|
// return;
|
||
|
MPI_Finalize();
|
||
|
exit(1);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static __inline__ void IB_CQ_stat( VAPI_wc_desc_t ret, char *string ) {
|
||
|
int rank;
|
||
|
|
||
|
if(ret.status != VAPI_SUCCESS) {
|
||
|
MPI_Comm_rank( MPI_COMM_WORLD, &rank );
|
||
|
printf("[%u]: *** [ERROR] *** %s -> %s\n", rank, string, VAPI_wc_status_sym( ret.status ));
|
||
|
printf("[%u]: *** [ERROR] *** opcode -> %s\n", rank, VAPI_cqe_opcode_sym(ret.opcode));
|
||
|
// return;
|
||
|
MPI_Finalize();
|
||
|
exit(1);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static int IB_Init() {
|
||
|
u_int32_t num_of_hcas; /* actual number of hcas */
|
||
|
VAPI_hca_id_t *hca_id_buf_p; /* HCA result buffer */
|
||
|
int ret, res;
|
||
|
|
||
|
/* keyval is not initialized yet, we have to init it */
|
||
|
if(MPI_KEYVAL_INVALID == IB_Gkeyval) {
|
||
|
res = MPI_Keyval_create(IB_Key_copy, IB_Key_delete, &(IB_Gkeyval), NULL);
|
||
|
if((MPI_SUCCESS != res)) { printf("Error in MPI_Keyval_create() (%i)\n", res); return IB_OOR; }
|
||
|
}
|
||
|
|
||
|
hca_id_buf_p = malloc(sizeof(VAPI_hca_id_t) * 2);
|
||
|
|
||
|
/* get all HCAs */
|
||
|
ret = EVAPI_list_hcas( (u_int32_t)2,
|
||
|
&num_of_hcas,
|
||
|
hca_id_buf_p );
|
||
|
if( ret == VAPI_OK) {
|
||
|
fprintf(stderr, "[INFO] found %d adapter(s), first-name: %s\n", num_of_hcas, (char *)hca_id_buf_p);
|
||
|
} else {
|
||
|
IB_stat( ret, "EVAPI_list_hcas()" );
|
||
|
}
|
||
|
|
||
|
/* get handle of first HCA */
|
||
|
ret = EVAPI_get_hca_hndl( *hca_id_buf_p, &IB_Hca_info.hca_hndl );
|
||
|
IB_stat( ret, "EVAPI_get_hca_hndl()" );
|
||
|
|
||
|
// allocate PD
|
||
|
ret = VAPI_alloc_pd( IB_Hca_info.hca_hndl,
|
||
|
&IB_Hca_info.pd_hndl );
|
||
|
IB_stat( ret, "VAPI_alloc_pd()" );
|
||
|
|
||
|
IB_Hca_info.memlist = hb_tree_new((dict_cmp_func)IB_Memlist_compare_entries, (void *) IB_Memlist_delete_key, (void *)IB_Memlist_memlist_delete);
|
||
|
if(IB_Hca_info.memlist == NULL) { printf("error in hb_dict_new()\n"); return IB_OOR; }
|
||
|
|
||
|
IB_Ginitialized = 1;
|
||
|
|
||
|
return IB_OK;
|
||
|
}
|
||
|
|
||
|
static __inline__ IB_Comminfo *IB_Comm_init(MPI_Comm comm) {
|
||
|
IB_Comminfo *comminfo;
|
||
|
int res, flag;
|
||
|
|
||
|
if(!IB_Ginitialized) IB_Init();
|
||
|
|
||
|
res = MPI_Attr_get(comm, IB_Gkeyval, &comminfo, &flag);
|
||
|
if((MPI_SUCCESS != res)) { printf("Error in MPI_Attr_get() (%i)\n", res); return NULL; }
|
||
|
if (!flag) {
|
||
|
VAPI_cqe_num_t num_of_entries_p; /* # CQ entries */
|
||
|
VAPI_mrw_t req_mrw_p;
|
||
|
VAPI_mrw_t rep_mrw_p; /* responded memory region */
|
||
|
IB_Peer_info *a2abuf1, *a2abuf2, *a2abuf3;
|
||
|
int p, i, j, rank;
|
||
|
|
||
|
res = MPI_Comm_size(comm, &p);
|
||
|
res = MPI_Comm_rank(comm, &rank);
|
||
|
|
||
|
/* we have to create a new one */
|
||
|
comminfo = malloc(sizeof(IB_Comminfo));
|
||
|
if(comminfo == NULL) { printf("Error in malloc()\n"); return NULL; }
|
||
|
|
||
|
comminfo->taglisthead = NULL;
|
||
|
comminfo->taglistend = NULL;
|
||
|
|
||
|
//printf("[%i] build up %i connections in comm %p \n", rank, p-1, comm);
|
||
|
/* allocate QPs */
|
||
|
comminfo->qp_hndl_arr = malloc(p*sizeof(VAPI_qp_hndl_t));
|
||
|
if(comminfo->qp_hndl_arr == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
/* allocate SR CQs */
|
||
|
comminfo->sr_cq_hndl_arr = malloc(p*sizeof(VAPI_cq_hndl_t));
|
||
|
if(comminfo->sr_cq_hndl_arr == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
/* allocate RR CQs */
|
||
|
comminfo->rr_cq_hndl_arr = malloc(p*sizeof(VAPI_cq_hndl_t));
|
||
|
if(comminfo->rr_cq_hndl_arr == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
/* allocate a tag list for each host */
|
||
|
comminfo->taglist=malloc(p*sizeof(hb_tree*));
|
||
|
if(comminfo->taglist == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
for(i=0; i<p;i++) {
|
||
|
comminfo->taglist[i] = hb_tree_new((dict_cmp_func)IB_Taglist_compare_entries, (void *) IB_Taglist_delete_key, (void *)IB_Taglist_delete);
|
||
|
if(comminfo->taglist[i] == NULL) { printf("hb_tree_new() error\n"); return NULL; }
|
||
|
}
|
||
|
|
||
|
/* allocate rtr send queue */
|
||
|
comminfo->rtr_send=malloc(IB_RTR_SIZE*sizeof(IB_Peer_info_tag));
|
||
|
if(comminfo->rtr_send == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
for(i=0; i<IB_RTR_SIZE;i++) comminfo->rtr_send[i].tag=-1;
|
||
|
/* allocate rtr queue */
|
||
|
comminfo->rtr=malloc(p*sizeof(IB_Peer_info_tag*));
|
||
|
if(comminfo->rtr== NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
/* allocate rtr free queue */
|
||
|
comminfo->rtr_peer_free=malloc(p*sizeof(int*));
|
||
|
if(comminfo->rtr_peer_free == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
/* allocate rtr info */
|
||
|
comminfo->rtr_info=malloc(p*sizeof(IB_Peer_info));
|
||
|
if(comminfo->rtr_info == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
/* allocate rtr l_key */
|
||
|
comminfo->rtr_l_key=malloc(p*sizeof(VAPI_lkey_t));
|
||
|
if(comminfo->rtr_l_key == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
/* allocate rtr memory region handle */
|
||
|
comminfo->rtr_mr_hndl_p=malloc(p*sizeof(VAPI_mr_hndl_t));
|
||
|
if(comminfo->rtr_mr_hndl_p == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
|
||
|
/* allocate eager send queue */
|
||
|
comminfo->eager_send=malloc(IB_EAGER_SIZE*sizeof(IB_Eager_data));
|
||
|
if(comminfo->eager_send == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
for(i=0; i<IB_EAGER_SIZE;i++) comminfo->eager_send[i].tag=-1;
|
||
|
/* allocate eager queue */
|
||
|
comminfo->eager=malloc(p*sizeof(IB_Eager_data*));
|
||
|
if(comminfo->eager== NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
/* allocate eager free queue */
|
||
|
comminfo->eager_peer_free=malloc(p*sizeof(int*));
|
||
|
if(comminfo->eager_peer_free == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
/* allocate eager info */
|
||
|
comminfo->eager_info=malloc(p*sizeof(IB_Peer_info));
|
||
|
if(comminfo->eager_info == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
/* allocate eager l_key */
|
||
|
comminfo->eager_l_key=malloc(p*sizeof(VAPI_lkey_t));
|
||
|
if(comminfo->eager_l_key == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
/* allocate eager memory region handle */
|
||
|
comminfo->eager_mr_hndl_p=malloc(p*sizeof(VAPI_mr_hndl_t));
|
||
|
if(comminfo->eager_mr_hndl_p == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
|
||
|
/* allocate eager free info */
|
||
|
comminfo->eager_free_info=malloc(p*sizeof(IB_Peer_info));
|
||
|
if(comminfo->eager_free_info == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
/* allocate eager free memory region handle */
|
||
|
comminfo->eager_free_mr_hndl_p=malloc(p*sizeof(VAPI_mr_hndl_t));
|
||
|
if(comminfo->eager_free_mr_hndl_p == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
|
||
|
/* allocate a2abuf1 */
|
||
|
a2abuf1=malloc(p*sizeof(IB_Peer_info));
|
||
|
if(a2abuf1 == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
/* allocate a2abuf2 */
|
||
|
a2abuf2=malloc(p*sizeof(IB_Peer_info));
|
||
|
if(a2abuf2 == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
/* allocate a2abuf3 */
|
||
|
a2abuf3=malloc(p*sizeof(IB_Peer_info));
|
||
|
if(a2abuf3 == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
|
||
|
for(i = 0; i < p; i++) {
|
||
|
if(i == rank) continue;
|
||
|
res = VAPI_create_cq(IB_Hca_info.hca_hndl, 100000, &comminfo->sr_cq_hndl_arr[i], &num_of_entries_p );
|
||
|
IB_stat( res, "VAPI_create_cq()" );
|
||
|
res = VAPI_create_cq(IB_Hca_info.hca_hndl, 100000, &comminfo->rr_cq_hndl_arr[i], &num_of_entries_p );
|
||
|
IB_stat( res, "VAPI_create_cq()" );
|
||
|
|
||
|
res = IB_Create_qp(rank, i, &IB_Hca_info.hca_hndl, &(comminfo->sr_cq_hndl_arr[i]), &(comminfo->rr_cq_hndl_arr[i]), &(comminfo->qp_hndl_arr[i]), &IB_Hca_info.pd_hndl, comm );
|
||
|
//printf("[%i] rank %i has sr_cq: %i and rr_cq: %i and qp_hndl: %i\n", rank, i, (int)comminfo->sr_cq_hndl_arr[i], (int)comminfo->rr_cq_hndl_arr[i], (int)comminfo->qp_hndl_arr[i]);
|
||
|
if(res != 0) { printf("Error in IB_Create_qp (%i)\n", res); return NULL; }
|
||
|
|
||
|
/* allocate rtr element */
|
||
|
comminfo->rtr[i] = malloc(sizeof(IB_Peer_info_tag)*IB_RTR_SIZE);
|
||
|
if(comminfo->rtr[i] == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
for(j=0; j<IB_RTR_SIZE; j++) comminfo->rtr[i][j].tag = -1;
|
||
|
/* allocate rtr free queue */
|
||
|
comminfo->rtr_peer_free[i]=malloc(sizeof(int)*IB_RTR_SIZE);
|
||
|
if(comminfo->rtr_peer_free[i] == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
/* set free rtr queue to free :) */
|
||
|
for(j=0; j<IB_RTR_SIZE; j++) comminfo->rtr_peer_free[i][j] = -1;
|
||
|
|
||
|
/* allocate eager element */
|
||
|
comminfo->eager[i] = malloc(sizeof(IB_Eager_data)*IB_EAGER_SIZE);
|
||
|
if(comminfo->eager[i] == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
for(j=0; j<IB_EAGER_SIZE; j++) { memset((void*)(&comminfo->eager[i][j]), 0, sizeof(IB_Eager_data)); }
|
||
|
/* allocate free queue */
|
||
|
comminfo->eager_peer_free[i]=malloc(sizeof(int)*IB_EAGER_SIZE);
|
||
|
if(comminfo->eager_peer_free[i] == NULL) { printf("malloc() error\n"); return NULL; }
|
||
|
/* set free queue to free :) */
|
||
|
for(j=0; j<IB_EAGER_SIZE; j++) comminfo->eager_peer_free[i][j] = -1;
|
||
|
}
|
||
|
|
||
|
for(i=0; i<p;i++) {
|
||
|
if (rank == i) continue;
|
||
|
/* register rtr buffer */
|
||
|
memset(&req_mrw_p, 0, sizeof(VAPI_mrw_t));
|
||
|
req_mrw_p.type = VAPI_MR;
|
||
|
req_mrw_p.start = (VAPI_virt_addr_t)(comminfo->rtr[i]);
|
||
|
req_mrw_p.size = sizeof(IB_Peer_info_tag)*IB_RTR_SIZE;
|
||
|
req_mrw_p.pd_hndl = IB_Hca_info.pd_hndl;
|
||
|
req_mrw_p.acl = VAPI_EN_LOCAL_WRITE |
|
||
|
VAPI_EN_REMOTE_WRITE |
|
||
|
VAPI_EN_REMOTE_READ;
|
||
|
|
||
|
res = VAPI_register_mr( IB_Hca_info.hca_hndl,
|
||
|
&req_mrw_p,
|
||
|
&comminfo->rtr_mr_hndl_p[i],
|
||
|
&rep_mrw_p );
|
||
|
IB_stat( res, "VAPI_register_mr() for RTR buffers" );
|
||
|
|
||
|
comminfo->rtr_l_key[i] = rep_mrw_p.l_key;
|
||
|
|
||
|
a2abuf1[i].r_key = (unsigned long)rep_mrw_p.r_key;
|
||
|
a2abuf1[i].addr = (unsigned long)comminfo->rtr[i];
|
||
|
//printf("[%i] my info - r_key: %lu addr for rank %i: %lu\n", rank, a2abuf[i].r_key, i, a2abuf[i].addr);
|
||
|
|
||
|
/* register eager buffer */
|
||
|
memset(&req_mrw_p, 0, sizeof(VAPI_mrw_t));
|
||
|
req_mrw_p.type = VAPI_MR;
|
||
|
req_mrw_p.start = (VAPI_virt_addr_t)(comminfo->eager[i]);
|
||
|
req_mrw_p.size = sizeof(IB_Eager_data)*IB_EAGER_SIZE;
|
||
|
req_mrw_p.pd_hndl = IB_Hca_info.pd_hndl;
|
||
|
req_mrw_p.acl = VAPI_EN_LOCAL_WRITE |
|
||
|
VAPI_EN_REMOTE_WRITE |
|
||
|
VAPI_EN_REMOTE_READ;
|
||
|
|
||
|
res = VAPI_register_mr( IB_Hca_info.hca_hndl,
|
||
|
&req_mrw_p,
|
||
|
&comminfo->eager_mr_hndl_p[i],
|
||
|
&rep_mrw_p );
|
||
|
IB_stat( res, "VAPI_register_mr() for eager buffers" );
|
||
|
|
||
|
comminfo->eager_l_key[i] = rep_mrw_p.l_key;
|
||
|
|
||
|
a2abuf2[i].r_key = (unsigned long)rep_mrw_p.r_key;
|
||
|
a2abuf2[i].addr = (unsigned long)comminfo->eager[i];
|
||
|
|
||
|
/* register eager free buffer */
|
||
|
memset(&req_mrw_p, 0, sizeof(VAPI_mrw_t));
|
||
|
req_mrw_p.type = VAPI_MR;
|
||
|
req_mrw_p.start = (VAPI_virt_addr_t)(comminfo->eager_peer_free[i]);
|
||
|
req_mrw_p.size = sizeof(int)*IB_EAGER_SIZE;
|
||
|
req_mrw_p.pd_hndl = IB_Hca_info.pd_hndl;
|
||
|
req_mrw_p.acl = VAPI_EN_LOCAL_WRITE |
|
||
|
VAPI_EN_REMOTE_WRITE |
|
||
|
VAPI_EN_REMOTE_READ;
|
||
|
|
||
|
res = VAPI_register_mr( IB_Hca_info.hca_hndl,
|
||
|
&req_mrw_p,
|
||
|
&comminfo->eager_free_mr_hndl_p[i],
|
||
|
&rep_mrw_p );
|
||
|
IB_stat( res, "VAPI_register_mr() for eager free buffer" );
|
||
|
|
||
|
a2abuf3[i].r_key = (unsigned long)rep_mrw_p.r_key;
|
||
|
a2abuf3[i].addr = (unsigned long)comminfo->eager_peer_free[i];
|
||
|
//printf("[%i] my info - r_key: %lu addr for rank %i: %lu\n", rank, a2abuf1[i].r_key, i, a2abuf1[i].addr);
|
||
|
//printf("[%i] my info - r_key: %lu addr for rank %i: %lu\n", rank, a2abuf2[i].r_key, i, a2abuf2[i].addr);
|
||
|
//printf("[%i] my info - r_key: %lu addr for rank %i: %lu\n", rank, a2abuf3[i].r_key, i, a2abuf3[i].addr);
|
||
|
}
|
||
|
MPI_Alltoall(a2abuf1, 2, MPI_UNSIGNED_LONG, comminfo->rtr_info, 2, MPI_UNSIGNED_LONG, comm);
|
||
|
free(a2abuf1);
|
||
|
MPI_Alltoall(a2abuf2, 2, MPI_UNSIGNED_LONG, comminfo->eager_info, 2, MPI_UNSIGNED_LONG, comm);
|
||
|
free(a2abuf2);
|
||
|
MPI_Alltoall(a2abuf3, 2, MPI_UNSIGNED_LONG, comminfo->eager_free_info, 2, MPI_UNSIGNED_LONG, comm);
|
||
|
free(a2abuf3);
|
||
|
for(i=0; i<p;i++) {
|
||
|
if (rank == i) continue;
|
||
|
//printf("[%i] rtr rem info - r_key: %lu addr for me at node %i: %lu\n", rank, comminfo->rtr_info[i].r_key, i, comminfo->rtr_info[i].addr);
|
||
|
//printf("[%i] eager rem info - r_key: %lu addr for me at node %i: %lu\n", rank, comminfo->eager_info[i].r_key, i, comminfo->eager_info[i].addr);
|
||
|
//printf("[%i] eager_free rem info - r_key: %lu addr for me at node %i: %lu\n", rank, comminfo->eager_free_info[i].r_key, i, comminfo->eager_free_info[i].addr);
|
||
|
}
|
||
|
|
||
|
/* register rtr send buffer */
|
||
|
memset(&req_mrw_p, 0, sizeof(VAPI_mrw_t));
|
||
|
req_mrw_p.type = VAPI_MR;
|
||
|
req_mrw_p.start = (VAPI_virt_addr_t)(comminfo->rtr_send);
|
||
|
req_mrw_p.size = sizeof(IB_Peer_info_tag)*IB_RTR_SIZE;
|
||
|
req_mrw_p.pd_hndl = IB_Hca_info.pd_hndl;
|
||
|
req_mrw_p.acl = VAPI_EN_LOCAL_WRITE |
|
||
|
VAPI_EN_REMOTE_WRITE |
|
||
|
VAPI_EN_REMOTE_READ;
|
||
|
|
||
|
res = VAPI_register_mr( IB_Hca_info.hca_hndl,
|
||
|
&req_mrw_p,
|
||
|
&comminfo->rtr_send_mr_hndl_p,
|
||
|
&rep_mrw_p );
|
||
|
IB_stat( res, "VAPI_register_mr()" );
|
||
|
|
||
|
comminfo->rtr_send_l_key = rep_mrw_p.l_key;
|
||
|
|
||
|
/* register eager send buffer */
|
||
|
memset(&req_mrw_p, 0, sizeof(VAPI_mrw_t));
|
||
|
req_mrw_p.type = VAPI_MR;
|
||
|
req_mrw_p.start = (VAPI_virt_addr_t)(comminfo->eager_send);
|
||
|
req_mrw_p.size = sizeof(IB_Eager_data)*IB_EAGER_SIZE;
|
||
|
req_mrw_p.pd_hndl = IB_Hca_info.pd_hndl;
|
||
|
req_mrw_p.acl = VAPI_EN_LOCAL_WRITE |
|
||
|
VAPI_EN_REMOTE_WRITE |
|
||
|
VAPI_EN_REMOTE_READ;
|
||
|
|
||
|
res = VAPI_register_mr( IB_Hca_info.hca_hndl,
|
||
|
&req_mrw_p,
|
||
|
&comminfo->eager_send_mr_hndl_p,
|
||
|
&rep_mrw_p );
|
||
|
IB_stat( res, "VAPI_register_mr()" );
|
||
|
|
||
|
comminfo->eager_send_l_key = rep_mrw_p.l_key;
|
||
|
|
||
|
/* register empty send buffer */
|
||
|
memset(&req_mrw_p, 0, sizeof(VAPI_mrw_t));
|
||
|
req_mrw_p.type = VAPI_MR;
|
||
|
req_mrw_p.start = (VAPI_virt_addr_t)(&comminfo->empty);
|
||
|
req_mrw_p.size = sizeof(int32_t);
|
||
|
req_mrw_p.pd_hndl = IB_Hca_info.pd_hndl;
|
||
|
req_mrw_p.acl = VAPI_EN_LOCAL_WRITE |
|
||
|
VAPI_EN_REMOTE_WRITE |
|
||
|
VAPI_EN_REMOTE_READ;
|
||
|
|
||
|
res = VAPI_register_mr( IB_Hca_info.hca_hndl,
|
||
|
&req_mrw_p,
|
||
|
&comminfo->eager_send_mr_hndl_p,
|
||
|
&rep_mrw_p );
|
||
|
IB_stat( res, "VAPI_register_mr()" );
|
||
|
|
||
|
comminfo->empty_l_key = rep_mrw_p.l_key;
|
||
|
|
||
|
comminfo->empty = -1;
|
||
|
|
||
|
/* put the new attribute to the comm */
|
||
|
res = MPI_Attr_put(comm, IB_Gkeyval, comminfo);
|
||
|
if((MPI_SUCCESS != res)) { printf("Error in MPI_Attr_put() (%i)\n", res); return NULL; }
|
||
|
}
|
||
|
|
||
|
return comminfo;
|
||
|
}
|
||
|
|
||
|
static __inline__ int IB_Register_mem(void *buf, int size, VAPI_mr_hndl_t *mr, VAPI_rkey_t *r_key, VAPI_lkey_t *l_key) {
|
||
|
VAPI_mrw_t req_mrw_p;
|
||
|
VAPI_mrw_t rep_mrw_p; /* responded memory region */
|
||
|
int res, rank;
|
||
|
IB_Memlistel *memel, *newel, keyel;
|
||
|
|
||
|
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
|
||
|
//printf("[%i] in IB_Register_mem\n", rank);
|
||
|
|
||
|
keyel.buf = buf;
|
||
|
keyel.size = size;
|
||
|
memel = hb_tree_search(IB_Hca_info.memlist, &keyel);
|
||
|
if(memel != NULL) {
|
||
|
//printf("[%i] we found a region from %lu to %lu :-)\n", rank, (unsigned long)memel->buf,(unsigned long)(memel->buf+memel->size));
|
||
|
if(r_key != NULL) *r_key = memel->r_key;
|
||
|
if(l_key != NULL) *l_key = memel->l_key;
|
||
|
return IB_OK;
|
||
|
}
|
||
|
|
||
|
//printf("[%i] we did not find a region - registering %i bytes from addr %lu to %lu :-(\n", rank, size, (unsigned long)buf, (unsigned long)(buf+size));
|
||
|
memset(&req_mrw_p, 0, sizeof(VAPI_mrw_t));
|
||
|
req_mrw_p.type = VAPI_MR;
|
||
|
req_mrw_p.start = (VAPI_virt_addr_t)buf;
|
||
|
req_mrw_p.size = size;
|
||
|
req_mrw_p.pd_hndl = IB_Hca_info.pd_hndl;
|
||
|
req_mrw_p.acl = VAPI_EN_LOCAL_WRITE |
|
||
|
VAPI_EN_REMOTE_WRITE |
|
||
|
VAPI_EN_REMOTE_READ;
|
||
|
|
||
|
res = VAPI_register_mr( IB_Hca_info.hca_hndl,
|
||
|
&req_mrw_p,
|
||
|
mr,
|
||
|
&rep_mrw_p );
|
||
|
IB_stat( res, "VAPI_register_mr()" );
|
||
|
|
||
|
/* TODO: we should react to "Resources temporary unavailable" (EAGAIN)
|
||
|
* and free some MRs ... to continue :) */
|
||
|
newel = malloc(sizeof(IB_Memlistel));
|
||
|
newel->buf = buf;
|
||
|
newel->size = size;
|
||
|
newel->mr=mr;
|
||
|
newel->r_key=rep_mrw_p.r_key;
|
||
|
newel->l_key=rep_mrw_p.l_key;
|
||
|
res = hb_tree_insert (IB_Hca_info.memlist, newel, newel, 0);
|
||
|
if(res != 0) printf("[%i] error in dict_insert() (%i) while inserting region from %lu to %lu\n", rank, res, (unsigned long)newel->buf, (unsigned long)(newel->buf+newel->size));
|
||
|
//if(res == 0) printf("[%i] inserted region from %lu to %lu\n", rank, (unsigned long)newel->buf, (unsigned long)newel->top);
|
||
|
|
||
|
if(r_key != NULL) *r_key = rep_mrw_p.r_key;
|
||
|
if(l_key != NULL) *l_key = rep_mrw_p.l_key;
|
||
|
|
||
|
return IB_OK;
|
||
|
}
|
||
|
|
||
|
static __inline__ int IB_Do_send(IB_Req *req) {
|
||
|
int res;
|
||
|
|
||
|
//printf("[%i] posting data SR from %p size: %i tag: %i to %i (addr: %lu, r_key: %u)\n", req->rank, (void*)req->sr_sg_lst.addr, req->sr_sg_lst.len, req->sr_desc.imm_data, req->peer, (unsigned long)req->sr_desc.remote_addr, req->sr_desc.r_key);
|
||
|
res = VAPI_post_sr( IB_Hca_info.hca_hndl, req->comminfo->qp_hndl_arr[req->peer], &req->sr_desc );
|
||
|
IB_stat( res, "VAPI_post_sr()" );
|
||
|
|
||
|
req->status = SEND_POSTED_SR;
|
||
|
|
||
|
return IB_OK;
|
||
|
}
|
||
|
|
||
|
int IB_Isend(void *buf, int count, MPI_Datatype type, int dst, int tag, MPI_Comm comm, IB_Request *request) {
|
||
|
int res,sendentry,i;
|
||
|
MPI_Aint ext;
|
||
|
IB_Req *req;
|
||
|
char *flag;
|
||
|
VAPI_sg_lst_entry_t sr_sg_lst; /* the IB SG list */
|
||
|
VAPI_sr_desc_t sr_desc; /* the IB SR descr. */
|
||
|
|
||
|
*request = malloc(sizeof(IB_Req));
|
||
|
req = *request;
|
||
|
|
||
|
if(!count) {
|
||
|
req->status = SEND_DONE;
|
||
|
return IB_OK;
|
||
|
}
|
||
|
|
||
|
MPI_Type_extent(type, &ext);
|
||
|
|
||
|
req->comminfo = IB_Comm_init(comm);
|
||
|
if(req->comminfo == NULL) { printf("Error in IB_Comm_init()\n"); return IB_OOR; }
|
||
|
|
||
|
if(count*ext <= IB_EAGER_LIMIT) {
|
||
|
/* we send this as eager message */
|
||
|
|
||
|
/* find free eager send buffer */
|
||
|
for(sendentry=0; sendentry<IB_EAGER_SIZE; sendentry++) {
|
||
|
if(req->comminfo->eager_send[sendentry].tag == -1) break;
|
||
|
}
|
||
|
if(sendentry==IB_EAGER_SIZE) { printf("*** eager send list full - we should retry later but crash\n"); exit(1); }
|
||
|
//printf("[%i] found local eager sendbuffer entry %i\n", req->rank, sendentry);
|
||
|
req->sendel = sendentry;
|
||
|
|
||
|
/* copy data into buffer */
|
||
|
memcpy(&req->comminfo->eager_send[sendentry].buf, buf, count*ext);
|
||
|
/* set header */
|
||
|
req->comminfo->eager_send[sendentry].size=count*ext;
|
||
|
req->comminfo->eager_send[sendentry].tag=tag;
|
||
|
/* set next byte after message to '1' */
|
||
|
flag = (char*)&req->comminfo->eager_send[sendentry].buf + req->comminfo->eager_send[sendentry].size;
|
||
|
*flag = (char)1;
|
||
|
|
||
|
/* prepare eager send request */
|
||
|
sr_sg_lst.addr = (VAPI_virt_addr_t)(&req->comminfo->eager_send[sendentry]);
|
||
|
sr_sg_lst.len = sizeof(int32_t)+2*sizeof(int16_t)+sizeof(char)+req->comminfo->eager_send[sendentry].size; /* TODO: dangerous - datatype */
|
||
|
sr_sg_lst.lkey = req->comminfo->eager_send_l_key;
|
||
|
sr_desc.id = (u_int64_t)req;
|
||
|
sr_desc.opcode = VAPI_RDMA_WRITE;
|
||
|
sr_desc.set_se = 0;
|
||
|
sr_desc.comp_type = VAPI_SIGNALED;
|
||
|
sr_desc.sg_lst_p = &sr_sg_lst;
|
||
|
sr_desc.sg_lst_len = 1;
|
||
|
sr_desc.imm_data = (u_int64_t)0; /* not used */
|
||
|
sr_desc.fence = 0;
|
||
|
sr_desc.compare_add = 0;
|
||
|
sr_desc.r_key = (VAPI_rkey_t)(req->comminfo->eager_info[dst].r_key);
|
||
|
/* get offset in the receiver's rtr eager array */
|
||
|
for(i=0; i<IB_EAGER_SIZE; i++) {
|
||
|
if(req->comminfo->eager_peer_free[dst][i] == -1) break;
|
||
|
}
|
||
|
if(i==IB_EAGER_SIZE) { printf("******************* [%i] eager list on peer %i full - we should retry later but crash\n", req->rank, dst); }
|
||
|
req->comminfo->eager_peer_free[dst][i] = tag;
|
||
|
sr_desc.remote_addr = req->comminfo->eager_info[dst].addr+i*sizeof(IB_Eager_data); /* TODO: 64 Bit */
|
||
|
req->comminfo->eager_send[sendentry].index=i;
|
||
|
//printf("[%i] found free eager slot index %i on node %i (addr: %lu) free)\n", req->rank, i, dst, sr_desc.remote_addr);
|
||
|
|
||
|
res = VAPI_post_sr( IB_Hca_info.hca_hndl, req->comminfo->qp_hndl_arr[dst], &sr_desc );
|
||
|
IB_stat( res, "VAPI_post_sr()" );
|
||
|
//printf("[%i] post EAGER SR from %lu to node %i addr %lu rkey: %u, len: %o\n", req->rank, sr_sg_lst.addr, req->peer, (unsigned long)sr_desc.remote_addr, sr_desc.r_key, sr_sg_lst.len);
|
||
|
|
||
|
|
||
|
req->status = SEND_SENT_EAGER;
|
||
|
//while(IB_Test(&req) != IB_OK);
|
||
|
|
||
|
} else {
|
||
|
/* we send this as rendezvous */
|
||
|
|
||
|
/* register memory region for send */
|
||
|
//printf("[%i] register send memory %lu size: %i\n", req->rank, (unsigned long)buf, (int)(count*ext));
|
||
|
res = IB_Register_mem(buf, count*ext, &req->mr_hndl_p, NULL, &req->sr_sg_lst.lkey); /* TODO: count*ext Danger for datatypes ... */
|
||
|
|
||
|
/* initialize sr_desc as far as we can (remote r_key and addr are
|
||
|
* missing set after we received RTR */
|
||
|
req->sr_sg_lst.addr = (VAPI_virt_addr_t)buf;
|
||
|
req->sr_sg_lst.len = count*ext; /* TODO: count*ext Danger for datatypes ... */
|
||
|
req->sr_desc.id = (u_int64_t)req;
|
||
|
req->sr_desc.opcode = VAPI_SEND_WITH_IMM;
|
||
|
req->sr_desc.set_se = 0;
|
||
|
req->sr_desc.comp_type = VAPI_SIGNALED;
|
||
|
req->sr_desc.sg_lst_p = &req->sr_sg_lst;
|
||
|
req->sr_desc.sg_lst_len = 1;
|
||
|
req->sr_desc.imm_data = (u_int64_t)tag;
|
||
|
req->sr_desc.fence = 0;
|
||
|
req->sr_desc.compare_add= 0;
|
||
|
|
||
|
req->status = SEND_WAITING_RTR;
|
||
|
}
|
||
|
|
||
|
//IB_Test(&req);
|
||
|
MPI_Comm_rank(comm, &req->rank);
|
||
|
MPI_Comm_size(comm, &req->p);
|
||
|
req->tag = tag;
|
||
|
req->peer= dst;
|
||
|
|
||
|
return IB_OK;
|
||
|
}
|
||
|
|
||
|
int IB_Irecv(void *buf, int count, MPI_Datatype type, int src, int tag, MPI_Comm comm, IB_Request *request) {
|
||
|
int res, i, sendentry;
|
||
|
MPI_Aint ext;
|
||
|
VAPI_sg_lst_entry_t sr_sg_lst; /* the IB SG list */
|
||
|
VAPI_sr_desc_t sr_desc; /* the IB SR descr. */
|
||
|
VAPI_sg_lst_entry_t rr_sg_lst;
|
||
|
VAPI_rr_desc_t rr_desc;
|
||
|
IB_Req *req;
|
||
|
IB_Taglstel *newel;
|
||
|
|
||
|
*request = malloc(sizeof(IB_Req));
|
||
|
req = *request;
|
||
|
|
||
|
if(count == 0) {
|
||
|
req->status = RECV_DONE;
|
||
|
return IB_OK;
|
||
|
}
|
||
|
|
||
|
MPI_Type_extent(type, &ext);
|
||
|
MPI_Comm_size(comm, &req->p);
|
||
|
MPI_Comm_rank(comm, &req->rank);
|
||
|
req->tag = tag;
|
||
|
req->peer= src;
|
||
|
req->buf=buf;
|
||
|
|
||
|
req->comminfo = IB_Comm_init(comm);
|
||
|
if(req->comminfo == NULL) { printf("Error in IB_Comm_init()\n"); return IB_OOR; }
|
||
|
|
||
|
if(count*ext <= IB_EAGER_LIMIT) {
|
||
|
/* do nothing, wait for eager message */
|
||
|
req->status = RECV_WAITING_EAGER;
|
||
|
/* we take a look if we received it already */
|
||
|
//IB_Test(&req);
|
||
|
} else {
|
||
|
/* find a new empty sendentry in the comminfo->send array which is
|
||
|
* pre-registered to send RTR messages from */
|
||
|
for(sendentry=0; sendentry<IB_RTR_SIZE; sendentry++) {
|
||
|
if(req->comminfo->rtr_send[sendentry].tag == -1) break;
|
||
|
}
|
||
|
if(sendentry==IB_RTR_SIZE) { printf("*** rtr send list full - we should retry later but crash\n"); exit(1); }
|
||
|
/* fill selected send entry */
|
||
|
req->comminfo->rtr_send[sendentry].tag = tag;
|
||
|
req->comminfo->rtr_send[sendentry].addr = (unsigned long)buf;
|
||
|
/* remember index in sendlist to free it fast after sending */
|
||
|
req->sendel = sendentry;
|
||
|
|
||
|
/* register memory region for recv */
|
||
|
res = IB_Register_mem(buf, count*ext, &req->mr_hndl_p, (VAPI_rkey_t*)&req->comminfo->rtr_send[sendentry].r_key, &rr_sg_lst.lkey); /* TODO: count*ext Danger for datatypes ... */
|
||
|
|
||
|
/* prepare data receive request */
|
||
|
rr_sg_lst.addr = (VAPI_virt_addr_t)buf;
|
||
|
rr_sg_lst.len = count*ext;
|
||
|
rr_desc.sg_lst_p = &rr_sg_lst;
|
||
|
rr_desc.sg_lst_len = 1;
|
||
|
rr_desc.id = (u_int64_t)req;
|
||
|
rr_desc.opcode = VAPI_RECEIVE;
|
||
|
rr_desc.comp_type = VAPI_SIGNALED;
|
||
|
|
||
|
res = VAPI_post_rr( IB_Hca_info.hca_hndl, req->comminfo->qp_hndl_arr[src], &rr_desc );
|
||
|
IB_stat( res, "VAPI_post_rr()" );
|
||
|
|
||
|
/* add the tag and the peer to the taglist */
|
||
|
/* legacy old crappy taglist :) */
|
||
|
//res = IB_Addtotaglst(comminfo, tag, req, src);
|
||
|
|
||
|
/* new fancy taglist */
|
||
|
newel = malloc(sizeof(IB_Taglstel));
|
||
|
newel->tag=tag;
|
||
|
newel->req=req;
|
||
|
res = hb_tree_insert(req->comminfo->taglist[src], newel, newel, 0);
|
||
|
//printf("[%i] inserted request %lu with tag %i and src %i\n", req->rank, (unsigned long)req, tag, src);
|
||
|
|
||
|
/* prepare RTR send request */
|
||
|
sr_sg_lst.addr = (VAPI_virt_addr_t)(&req->comminfo->rtr_send[sendentry]);
|
||
|
sr_sg_lst.len = sizeof(IB_Peer_info_tag);
|
||
|
sr_sg_lst.lkey = req->comminfo->rtr_send_l_key;
|
||
|
sr_desc.id = (u_int64_t)req;
|
||
|
sr_desc.opcode = VAPI_RDMA_WRITE;
|
||
|
sr_desc.set_se = 0;
|
||
|
sr_desc.comp_type = VAPI_SIGNALED;
|
||
|
sr_desc.sg_lst_p = &sr_sg_lst;
|
||
|
sr_desc.sg_lst_len = 1;
|
||
|
sr_desc.imm_data = (u_int64_t)0; /* not used */
|
||
|
sr_desc.fence = 0;
|
||
|
sr_desc.compare_add = 0;
|
||
|
sr_desc.r_key = (VAPI_rkey_t)(req->comminfo->rtr_info[src].r_key);
|
||
|
/* get offset in the receiver's rtr RTR array */
|
||
|
for(i=0; i<IB_RTR_SIZE; i++) {
|
||
|
if(req->comminfo->rtr_peer_free[src][i] == -1) break;
|
||
|
}
|
||
|
if(i==IB_RTR_SIZE) { printf("*** unexpected list full - we should retry later but crash\n"); }
|
||
|
req->comminfo->rtr_peer_free[src][i] = tag;
|
||
|
sr_desc.remote_addr = req->comminfo->rtr_info[src].addr+i*sizeof(IB_Peer_info_tag); /* TODO: 64 Bit */
|
||
|
//printf("[%i] found free RTR slot index %i on node %i (addr: %lu) free)\n", req->rank, i, src, sr_desc.remote_addr);
|
||
|
|
||
|
/* post RTR request */
|
||
|
res = VAPI_post_sr( IB_Hca_info.hca_hndl, req->comminfo->qp_hndl_arr[req->peer], &sr_desc );
|
||
|
IB_stat( res, "VAPI_post_sr()" );
|
||
|
//printf("[%i] post RTR SR from %lu to node %i addr %lu rkey: %u\n", req->rank, sr_sg_lst.addr, req->peer, (unsigned long)sr_desc.remote_addr, sr_desc.r_key);
|
||
|
|
||
|
req->status = RECV_SENDING_RTR;
|
||
|
}
|
||
|
|
||
|
//IB_Test(&req);
|
||
|
|
||
|
return IB_OK;
|
||
|
}
|
||
|
|
||
|
int IB_Test(IB_Request *request) {
|
||
|
int i, j, res;
|
||
|
VAPI_wc_desc_t comp_desc_p; /* work completion descriptor */
|
||
|
IB_Req *tmpreq, *req;
|
||
|
IB_Taglstel *memel, keyel;
|
||
|
char *flag;
|
||
|
|
||
|
req = *request;
|
||
|
|
||
|
if((req->status == SEND_DONE) || (req->status == RECV_DONE) || (req->status == RECV_EAGER_DONE))
|
||
|
return IB_OK;
|
||
|
|
||
|
/* if I wait for RTR - search rtr array for my tag ... */
|
||
|
if(req->status == SEND_WAITING_RTR) {
|
||
|
for(i=0; i<IB_RTR_SIZE; i++) {
|
||
|
if(req->comminfo->rtr[req->peer][i].tag == req->tag) {
|
||
|
//printf("[%i] found RTR from peer %i at addr %lu (tag: %i, r_key: %lu, addr: %lu)\n", req->rank, req->peer, (unsigned long)(&req->comminfo->rtr[req->peer][i]), req->comminfo->rtr[req->peer][i].tag, req->comminfo->rtr[req->peer][i].r_key, (unsigned long)req->comminfo->rtr[req->peer][i].addr);
|
||
|
req->sr_desc.r_key = (VAPI_rkey_t)req->comminfo->rtr[req->peer][i].r_key;
|
||
|
req->sr_desc.remote_addr = req->comminfo->rtr[req->peer][i].addr; /* TODO: 64 Bit */
|
||
|
/* 'free' rtr element */
|
||
|
req->comminfo->rtr[req->peer][i].tag = -1;
|
||
|
IB_Do_send(req);
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/* I wait for an eager message */
|
||
|
if(req->status == RECV_WAITING_EAGER) {
|
||
|
/* ok, poll all eager slots we have from the peer we wait for */
|
||
|
for(i=0; i<IB_EAGER_SIZE; i++) {
|
||
|
if(req->comminfo->eager[req->peer][i].tag == req->tag) {
|
||
|
VAPI_sg_lst_entry_t sr_sg_lst; /* the IB SG list */
|
||
|
VAPI_sr_desc_t sr_desc; /* the IB SR descr. */
|
||
|
int index;
|
||
|
|
||
|
/* poll last byte until we can be sure that we have the *full*
|
||
|
* message in the buffer */
|
||
|
flag = (char*)&req->comminfo->eager[req->peer][i].buf + req->comminfo->eager[req->peer][i].size;
|
||
|
while(*flag != (char)1);
|
||
|
//printf("[%i] found eager message from peer %i at addr %lu (tag: %i)\n", req->rank, req->peer, (unsigned long)(&req->comminfo->eager[req->peer][i]), (int)req->comminfo->eager[req->peer][i].tag);
|
||
|
/* copy message to recv buffer */
|
||
|
memcpy(req->buf, &req->comminfo->eager[req->peer][i].buf, req->comminfo->eager[req->peer][i].size);
|
||
|
index = req->comminfo->eager[req->peer][i].index;
|
||
|
/* set the buffer to '0' to make flag-polling useful */
|
||
|
memset((void*)(&req->comminfo->eager[req->peer][i]), 0, sizeof(int32_t)+2*sizeof(int16_t)+sizeof(char)+req->comminfo->eager[req->peer][i].size);
|
||
|
/* RDMA into the free-buffer on the sender to indicate that my
|
||
|
* buffer can be reused */
|
||
|
|
||
|
//printf("[%i] RDMA EAGER_RECVD (%i) to node %i in buffer %lu at index %i\n", req->rank, req->comminfo->empty, req->peer, (unsigned long)req->comminfo->eager_free_info[req->peer].addr, index);
|
||
|
/* prepare EAGER_RECVD send request */
|
||
|
sr_sg_lst.addr = (VAPI_virt_addr_t)(&req->comminfo->empty);
|
||
|
sr_sg_lst.len = sizeof(int);
|
||
|
sr_sg_lst.lkey = req->comminfo->empty_l_key;
|
||
|
sr_desc.id = (u_int64_t)0;
|
||
|
sr_desc.opcode = VAPI_RDMA_WRITE;
|
||
|
sr_desc.set_se = 0;
|
||
|
sr_desc.comp_type = VAPI_SIGNALED;
|
||
|
sr_desc.sg_lst_p = &sr_sg_lst;
|
||
|
sr_desc.sg_lst_len = 1;
|
||
|
sr_desc.imm_data = (u_int64_t)0; /* not used */
|
||
|
sr_desc.fence = 0;
|
||
|
sr_desc.compare_add = 0;
|
||
|
sr_desc.r_key = (VAPI_rkey_t)(req->comminfo->eager_free_info[req->peer].r_key);
|
||
|
sr_desc.remote_addr = req->comminfo->eager_free_info[req->peer].addr+sizeof(int)*index; /* TODO: 64 Bit */
|
||
|
|
||
|
/* post EAGER_RECVD request */
|
||
|
res = VAPI_post_sr( IB_Hca_info.hca_hndl, req->comminfo->qp_hndl_arr[req->peer], &sr_desc );
|
||
|
IB_stat( res, "VAPI_post_sr()" );
|
||
|
/* mark receive as done */
|
||
|
req->status = RECV_EAGER_DONE;
|
||
|
/* leave loop */
|
||
|
break;
|
||
|
// return IB_OK;
|
||
|
// -> we need to poll CQs ...
|
||
|
}
|
||
|
}
|
||
|
// return IB_CONTINUE;
|
||
|
}
|
||
|
|
||
|
//t=0-MPI_Wtime();
|
||
|
/* poll all CQs of the comm related to req (we should probably only poll those where we wait for something */
|
||
|
//for(i=0; i<req->p; i++) {
|
||
|
//for(i=req->peer; i<=req->peer; i++) {
|
||
|
//if(i == req->rank) continue;
|
||
|
i=req->peer;
|
||
|
/************************************** SEND QUEUE handling ************************************/
|
||
|
res = VAPI_poll_cq( IB_Hca_info.hca_hndl, req->comminfo->sr_cq_hndl_arr[i], &comp_desc_p );
|
||
|
if((res != VAPI_EBUSY) && (res != VAPI_CQ_EMPTY)) {
|
||
|
//printf("[%i] have SR CQE from host %i\n", req->rank, i);
|
||
|
IB_CQ_stat( comp_desc_p, "VAPI_poll_cq(SR)" );
|
||
|
/* id == 0 for eager recvd messages - we do not need to wait for them ... */
|
||
|
if(comp_desc_p.id != 0) {
|
||
|
tmpreq = (IB_Req*)(comp_desc_p.id);
|
||
|
if(tmpreq->status == SEND_POSTED_SR) {
|
||
|
/* we sent the message and are ready */
|
||
|
/* TODO: free *all* request resources here */
|
||
|
tmpreq->status = SEND_DONE;
|
||
|
//printf("[%i] req %lu send to %i with tag %i is done ...\n", tmpreq->rank, (unsigned long)tmpreq, tmpreq->peer, tmpreq->tag);
|
||
|
} else if (tmpreq->status == SEND_SENT_EAGER) {
|
||
|
/* set rtr sendlist element to free */
|
||
|
tmpreq->comminfo->eager_send[tmpreq->sendel].tag = -1;
|
||
|
tmpreq->status = SEND_DONE;
|
||
|
//printf("[%i] eager req %lu send to %i with tag %i is done ...\n", tmpreq->rank, (unsigned long)tmpreq, tmpreq->peer, tmpreq->tag);
|
||
|
} else if ((tmpreq->status == RECV_SENDING_RTR) || (tmpreq->status == RECV_DONE)) {
|
||
|
/* set rtr sendlist element to free */
|
||
|
tmpreq->comminfo->rtr_send[tmpreq->sendel].tag = -1;
|
||
|
/* do not change DONE requests back :o) */
|
||
|
if(tmpreq->status != RECV_DONE) tmpreq->status = RECV_SENT_RTR;
|
||
|
} else {
|
||
|
printf("[%i] req %lu unexpected status (%i) for send to %i (tag: %i) after poll sr_cq \n", tmpreq->rank, (unsigned long)tmpreq, tmpreq->status, tmpreq->peer, tmpreq->tag);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
/************************************** RECEIVE QUEUE handling ************************************/
|
||
|
res = VAPI_poll_cq( IB_Hca_info.hca_hndl, req->comminfo->rr_cq_hndl_arr[i], &comp_desc_p );
|
||
|
if((res != VAPI_EBUSY) && (res != VAPI_CQ_EMPTY)) {
|
||
|
//printf("[%i] have RR CQE from host %i\n", req->rank, i);
|
||
|
IB_CQ_stat( comp_desc_p, "VAPI_poll_cq(RR)" );
|
||
|
|
||
|
/* we received real data - match it (tag, peer)*/
|
||
|
keyel.tag=comp_desc_p.imm_data;
|
||
|
memel = hb_tree_search(req->comminfo->taglist[i], &keyel);
|
||
|
if(memel == NULL) {
|
||
|
printf("[%i] got unexpected packet with tag: %i, peer: %i\n", req->rank, comp_desc_p.imm_data, i);
|
||
|
printf(" this CANNOT happen!!!\n");
|
||
|
return IB_CONTINUE;
|
||
|
}
|
||
|
/* delete element - TODO: this should actually be done together with
|
||
|
* the find ... we should extend libdict */
|
||
|
res = hb_tree_remove(req->comminfo->taglist[i], &keyel, 0);
|
||
|
if(res != 0) { printf("error deleting tag element in hb_tree_remove()\n"); }
|
||
|
tmpreq = memel->req;
|
||
|
//printf("[%i] found request %lu for tag %i and src %i\n", tmpreq->rank, (unsigned long)tmpreq, comp_desc_p.imm_data, i);
|
||
|
if((tmpreq->status == RECV_SENDING_RTR) || (tmpreq->status == RECV_SENT_RTR)) {
|
||
|
/* TODO: free *all* req resources here */
|
||
|
//printf("[%i] req %lu we received data message from host %i for tag %i-> done \n", req->rank, (unsigned long)tmpreq, req->peer, req->tag);
|
||
|
|
||
|
/* get offset in the receiver's array */
|
||
|
for(j=0; j<IB_RTR_SIZE; j++) {
|
||
|
if(tmpreq->comminfo->rtr_peer_free[tmpreq->peer][j] == tmpreq->tag) break;
|
||
|
}
|
||
|
if(j==IB_RTR_SIZE) { printf("[%i] we did not find tag %i to delete - should not HAPPEN!!!\n", tmpreq->rank, tmpreq->tag); }
|
||
|
tmpreq->comminfo->rtr_peer_free[tmpreq->peer][j] = -1;
|
||
|
|
||
|
tmpreq->status = RECV_DONE;
|
||
|
} else {
|
||
|
printf("[%i] req %lu (tag: %i) unexpected status (%i) after poll rr_cq \n", tmpreq->rank, (unsigned long)tmpreq, tmpreq->tag, tmpreq->status);
|
||
|
}
|
||
|
}
|
||
|
//}
|
||
|
//t+=MPI_Wtime();
|
||
|
//printf("time: %lf\n", t*1e6);
|
||
|
|
||
|
if((req->status == SEND_DONE) || (req->status == RECV_DONE) || (req->status == RECV_EAGER_DONE))
|
||
|
return IB_OK;
|
||
|
else
|
||
|
return IB_CONTINUE;
|
||
|
}
|
||
|
|
||
|
int IB_Wait(IB_Request *request) {
|
||
|
|
||
|
while(IB_Test(request) != IB_OK) {};
|
||
|
|
||
|
return IB_OK;
|
||
|
}
|
||
|
|
||
|
int IB_Testall(int count, IB_Request *requests, int *flag) {
|
||
|
int i, res;
|
||
|
|
||
|
*flag = 1;
|
||
|
for(i=0; i<count; i++) {
|
||
|
res = IB_Test(&requests[i]);
|
||
|
if(res != IB_OK) *flag = 0;
|
||
|
}
|
||
|
|
||
|
return IB_OK;
|
||
|
}
|
||
|
|
||
|
int IB_Waitall(int count, IB_Request *requests) {
|
||
|
int i, res, done;
|
||
|
|
||
|
done = IB_CONTINUE;
|
||
|
do {
|
||
|
done = IB_OK;
|
||
|
for(i=0; i<count; i++) {
|
||
|
if((requests[i]->status != SEND_DONE) && (requests[i]->status != RECV_DONE) && (requests[i]->status != RECV_EAGER_DONE))
|
||
|
{
|
||
|
res = IB_Test(&requests[i]); /* we shouldn't test ready requests ... */
|
||
|
/* in case of error */
|
||
|
if((res != IB_OK) && (res != IB_CONTINUE)) break;
|
||
|
/* we have at least one unfinished request ... */
|
||
|
if(res == IB_CONTINUE) done = IB_CONTINUE;
|
||
|
}
|
||
|
}
|
||
|
} while(done == IB_CONTINUE);
|
||
|
|
||
|
return done;
|
||
|
}
|
||
|
|
||
|
static int IB_Create_qp( int rank, int remote,
|
||
|
VAPI_hca_hndl_t *hca_hndl_p,
|
||
|
VAPI_cq_hndl_t *sr_cq_hndl_p,
|
||
|
VAPI_cq_hndl_t *rr_cq_hndl_p,
|
||
|
VAPI_qp_hndl_t *qp_hndl_p,
|
||
|
VAPI_pd_hndl_t *pd_hndp_p,
|
||
|
MPI_Comm comm) {
|
||
|
// variables
|
||
|
VAPI_qp_init_attr_t qp_init_attr_p; // QP init attribs
|
||
|
VAPI_ret_t ret; // VAPI return value
|
||
|
VAPI_qp_prop_t qp_prop_p, rem_qp_prop_p; // QP properties
|
||
|
VAPI_hca_port_t hca_port_p, rem_hca_port_p; // port properties
|
||
|
VAPI_qp_attr_mask_t qp_attr_mask_p; // QP attribute mask
|
||
|
VAPI_qp_attr_t qp_attr_p; // QP attributes
|
||
|
VAPI_qp_cap_t qp_cap_p; // QP capabilities
|
||
|
|
||
|
MPI_Status stat; // mpi status
|
||
|
|
||
|
// get my LID
|
||
|
ret = VAPI_query_hca_port_prop( *hca_hndl_p,
|
||
|
(IB_port_t)1,
|
||
|
&hca_port_p );
|
||
|
IB_stat( ret, "EVAPI_get_hca_hndl()" );
|
||
|
|
||
|
|
||
|
// set QP Properties ...
|
||
|
qp_init_attr_p.sq_cq_hndl = *sr_cq_hndl_p; // cq associated with sr's
|
||
|
qp_init_attr_p.rq_cq_hndl = *rr_cq_hndl_p; // cq associated with rr's
|
||
|
qp_init_attr_p.cap.max_oust_wr_sq = 1000; // TODO guessed
|
||
|
qp_init_attr_p.cap.max_oust_wr_rq = 1000; // TODO guessed
|
||
|
qp_init_attr_p.cap.max_sg_size_sq = 1; // TODO guessed
|
||
|
qp_init_attr_p.cap.max_sg_size_rq = 1; // TODO guessed
|
||
|
qp_init_attr_p.cap.max_inline_data_sq = 1; // TODO guessed
|
||
|
// qp_init_attr_p.rdd_hndl = no rdd (Reliable Datagram Domain) used
|
||
|
qp_init_attr_p.sq_sig_type = VAPI_SIGNAL_ALL_WR; // signal all submitted WR's
|
||
|
qp_init_attr_p.rq_sig_type = VAPI_SIGNAL_ALL_WR; // signal all submitted WR's
|
||
|
qp_init_attr_p.pd_hndl = *pd_hndp_p; // the PD
|
||
|
// QP Transport Service Type
|
||
|
qp_init_attr_p.ts_type = VAPI_TS_RC;
|
||
|
|
||
|
// create QP
|
||
|
ret = VAPI_create_qp( *hca_hndl_p,
|
||
|
&qp_init_attr_p,
|
||
|
qp_hndl_p,
|
||
|
&qp_prop_p );
|
||
|
IB_stat( ret, "VAPI_create_qp()" );
|
||
|
|
||
|
MPI_Send(&hca_port_p.lid, 1, MPI_INT, remote, 0, comm);
|
||
|
MPI_Recv(&rem_hca_port_p.lid, 1, MPI_INT, remote, 0, comm, &stat);
|
||
|
MPI_Send(&qp_prop_p.qp_num, 1, MPI_INT, remote, 0, comm);
|
||
|
MPI_Recv(&rem_qp_prop_p.qp_num, 1, MPI_INT, remote, 0, comm, &stat);
|
||
|
|
||
|
//fprintf(stderr, "[INFO] (lid:qp) - %x:%u -> %x:%u\n", hca_port_p.lid, qp_prop_p.qp_num, rem_hca_port_p.lid, rem_qp_prop_p.qp_num);
|
||
|
|
||
|
// see page 222 for details ...
|
||
|
/* Transition RST to INIT */
|
||
|
QP_ATTR_MASK_CLR_ALL(qp_attr_mask_p);
|
||
|
// QP State
|
||
|
QP_ATTR_MASK_SET( qp_attr_mask_p, QP_ATTR_QP_STATE);
|
||
|
qp_attr_p.qp_state = VAPI_INIT;
|
||
|
|
||
|
// partition key index (see page 319)
|
||
|
QP_ATTR_MASK_SET( qp_attr_mask_p, QP_ATTR_PKEY_IX);
|
||
|
qp_attr_p.pkey_ix = 0; // first partition key
|
||
|
|
||
|
// queue key -> only for datagram (RD, UD) TODO set this!
|
||
|
// QP_ATTR_MASK_SET( qp_attr_mask_p, QP_ATTR_QKEY);
|
||
|
// qp_attr_p.qkey = 0;
|
||
|
|
||
|
// enable/disable RDMA R / Atomic -> allow all :)
|
||
|
QP_ATTR_MASK_SET( qp_attr_mask_p, QP_ATTR_REMOTE_ATOMIC_FLAGS);
|
||
|
qp_attr_p.remote_atomic_flags = VAPI_EN_REM_WRITE;
|
||
|
|
||
|
// physical port
|
||
|
QP_ATTR_MASK_SET( qp_attr_mask_p, QP_ATTR_PORT);
|
||
|
qp_attr_p.port = 1; // TODO should not be static!
|
||
|
|
||
|
ret = VAPI_modify_qp( *hca_hndl_p,
|
||
|
*qp_hndl_p,
|
||
|
&qp_attr_p,
|
||
|
&qp_attr_mask_p,
|
||
|
&qp_cap_p );
|
||
|
IB_stat( ret, "VAPI_modify_qp() (RST->INIT)" );
|
||
|
|
||
|
// Transition INIT to RTR
|
||
|
QP_ATTR_MASK_CLR_ALL(qp_attr_mask_p);
|
||
|
|
||
|
// new state
|
||
|
QP_ATTR_MASK_SET( qp_attr_mask_p, QP_ATTR_QP_STATE);
|
||
|
qp_attr_p.qp_state = VAPI_RTR;
|
||
|
|
||
|
// remote node address vector
|
||
|
QP_ATTR_MASK_SET( qp_attr_mask_p, QP_ATTR_AV);
|
||
|
qp_attr_p.av.sl = 0; // TODO guessed
|
||
|
qp_attr_p.av.dlid = rem_hca_port_p.lid; // partners lid
|
||
|
qp_attr_p.av.src_path_bits = 0; // TODO guessed
|
||
|
qp_attr_p.av.static_rate = 0; // TODO guessed
|
||
|
qp_attr_p.av.grh_flag = 0; // non set ...
|
||
|
qp_attr_p.av.traffic_class = 0; // TODO guessed
|
||
|
qp_attr_p.av.hop_limit = 255; // should be ok - only global routing
|
||
|
qp_attr_p.av.flow_label = 0; // only global routing
|
||
|
qp_attr_p.av.sgid_index = 0; // only global routing
|
||
|
qp_attr_p.av.port = 1; // TODO guessed
|
||
|
|
||
|
// packet sequence number
|
||
|
QP_ATTR_MASK_SET(qp_attr_mask_p, QP_ATTR_RQ_PSN);
|
||
|
qp_attr_p.rq_psn = 0;
|
||
|
|
||
|
// number of responder resources for RDMA R + Atomic
|
||
|
QP_ATTR_MASK_SET(qp_attr_mask_p, QP_ATTR_QP_OUS_RD_ATOM);
|
||
|
qp_attr_p.qp_ous_rd_atom = 5; //TODO tune here?
|
||
|
|
||
|
// minimum rnr nak timer
|
||
|
QP_ATTR_MASK_SET(qp_attr_mask_p, QP_ATTR_MIN_RNR_TIMER);
|
||
|
qp_attr_p.min_rnr_timer = 1; // TODO guessed
|
||
|
|
||
|
// dest QP number
|
||
|
QP_ATTR_MASK_SET(qp_attr_mask_p, QP_ATTR_DEST_QP_NUM);
|
||
|
qp_attr_p.dest_qp_num = rem_qp_prop_p.qp_num; // partners qp num
|
||
|
|
||
|
QP_ATTR_MASK_SET(qp_attr_mask_p, QP_ATTR_PATH_MTU);
|
||
|
qp_attr_p.path_mtu = MTU1024; // TODO tune here
|
||
|
|
||
|
ret = VAPI_modify_qp( *hca_hndl_p,
|
||
|
*qp_hndl_p,
|
||
|
&qp_attr_p,
|
||
|
&qp_attr_mask_p,
|
||
|
&qp_cap_p );
|
||
|
IB_stat( ret, "VAPI_modify_qp() (INIT->RTR)" );
|
||
|
|
||
|
|
||
|
// Transition RTR to RTS
|
||
|
QP_ATTR_MASK_CLR_ALL(qp_attr_mask_p);
|
||
|
// new state
|
||
|
QP_ATTR_MASK_SET( qp_attr_mask_p, QP_ATTR_QP_STATE);
|
||
|
qp_attr_p.qp_state = VAPI_RTS;
|
||
|
|
||
|
// SQ sequence number
|
||
|
QP_ATTR_MASK_SET(qp_attr_mask_p, QP_ATTR_SQ_PSN);
|
||
|
qp_attr_p.sq_psn = 0;
|
||
|
|
||
|
// local ACK timeout
|
||
|
QP_ATTR_MASK_SET(qp_attr_mask_p, QP_ATTR_TIMEOUT);
|
||
|
qp_attr_p.timeout = 10; // TODO tune here
|
||
|
|
||
|
// retry count
|
||
|
QP_ATTR_MASK_SET(qp_attr_mask_p, QP_ATTR_RNR_RETRY);
|
||
|
qp_attr_p.rnr_retry = 255; // increased due to VAPI_RETRY_EXC_ERR if 30 nodes fire 0.5MB at one :) -> maximum :)
|
||
|
|
||
|
// number of outstanding RDMA R / atomic at destination
|
||
|
QP_ATTR_MASK_SET(qp_attr_mask_p, QP_ATTR_OUS_DST_RD_ATOM);
|
||
|
qp_attr_p.ous_dst_rd_atom = 10; // TODO tune here
|
||
|
|
||
|
// retry count
|
||
|
QP_ATTR_MASK_SET(qp_attr_mask_p, QP_ATTR_RETRY_COUNT);
|
||
|
qp_attr_p.retry_count = 255; // guessed
|
||
|
|
||
|
ret = VAPI_modify_qp( *hca_hndl_p,
|
||
|
*qp_hndl_p,
|
||
|
&qp_attr_p,
|
||
|
&qp_attr_mask_p,
|
||
|
&qp_cap_p );
|
||
|
IB_stat( ret, "VAPI_modify_qp() (RTR->RTS)" );
|
||
|
|
||
|
return IB_OK;
|
||
|
}
|
||
|
|
||
|
static __inline__ int IB_Memlist_compare_entries(IB_Memlistel *a, IB_Memlistel *b,void *param) {
|
||
|
|
||
|
/* two memory regions are defined as equal if they have some common
|
||
|
* memory - more is not possible, because we have to ensure
|
||
|
* reflexibility (a=b includes b=a) */
|
||
|
|
||
|
if( (a->buf == b->buf) && (a->size == b->size) ) {
|
||
|
return 0;
|
||
|
}
|
||
|
if ( (a->buf < b->buf)) {
|
||
|
return -1;
|
||
|
}
|
||
|
return +1;
|
||
|
}
|
||
|
|
||
|
static __inline__ void IB_Memlist_delete_key(IB_Memlistel *k) {
|
||
|
/* do nothing because the key and the data element are identical :-)
|
||
|
* both (the single one :) is freed in IB_Memlist_memlist_delete() */
|
||
|
}
|
||
|
|
||
|
static __inline__ void IB_Memlist_memlist_delete(IB_Memlistel *entry) {
|
||
|
/* free entry and deregister MR here ... */
|
||
|
}
|
||
|
|
||
|
static __inline__ int IB_Taglist_compare_entries(IB_Taglstel *a, IB_Taglstel *b,void *param) {
|
||
|
|
||
|
if( a->tag == b->tag ) {
|
||
|
return 0;
|
||
|
}
|
||
|
if( a->tag < b->tag ) {
|
||
|
return -1;
|
||
|
}
|
||
|
return +1;
|
||
|
}
|
||
|
|
||
|
static __inline__ void IB_Taglist_delete_key(IB_Taglstel *k) {
|
||
|
/* do nothing because the key and the data element are identical :-)
|
||
|
* both (the single one :) is freed in IB_Memlist_memlist_delete() */
|
||
|
}
|
||
|
|
||
|
static __inline__ void IB_Taglist_delete(IB_Taglstel *entry) {
|
||
|
/* free taglistentry */
|
||
|
free(entry);
|
||
|
}
|