020efdf1f9
This commit essentially caches the invoking comm/win/file on the ompi_request_t. This, paired with the req_type field, allows us to retrieve the invoking MPI object and invoke the proper errhandler. The patch is missing most updates for the MPI-2 one-sided stuff (i.e., the patch mainly fixes comms and files); I didn't really understand that code and didn't want to hazard trying to figure it out when Brian can probably do it much more quickly. So #250 will still stay open, pending MPI-2 one-sided updates for this stuff. This commit was SVN r12339. The following Trac tickets were found above: Ticket 250 --> https://svn.open-mpi.org/trac/ompi/ticket/250
128 строки
4.7 KiB
C
128 строки
4.7 KiB
C
/*
|
|
* 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-2006 The Regents of the University of California.
|
|
* All rights reserved.
|
|
* Copyright (c) 2006 Cisco Systems, Inc. All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "ompi_config.h"
|
|
|
|
#include "pml_dr.h"
|
|
#include "pml_dr_recvreq.h"
|
|
#include "pml_dr_sendreq.h"
|
|
|
|
|
|
int mca_pml_dr_start(size_t count, ompi_request_t** requests)
|
|
{
|
|
int rc;
|
|
size_t i;
|
|
for(i=0; i<count; i++) {
|
|
mca_pml_base_request_t *pml_request = (mca_pml_base_request_t*)requests[i];
|
|
if (OMPI_REQUEST_PML != requests[i]->req_type) {
|
|
continue;
|
|
}
|
|
if(NULL == pml_request)
|
|
continue;
|
|
|
|
/* If the persistent request is currebtly active - obtain the
|
|
* request lock and verify the status is incomplete. if the
|
|
* pml layer has not completed the request - mark the request
|
|
* as free called - so that it will be freed when the request
|
|
* completes - and create a new request.
|
|
*/
|
|
|
|
switch(pml_request->req_ompi.req_state) {
|
|
case OMPI_REQUEST_INACTIVE:
|
|
if(pml_request->req_pml_complete == true)
|
|
break;
|
|
/* otherwise fall through */
|
|
case OMPI_REQUEST_ACTIVE: {
|
|
|
|
ompi_request_t *request;
|
|
OPAL_THREAD_LOCK(&ompi_request_lock);
|
|
if (pml_request->req_pml_complete == false) {
|
|
/* free request after it completes */
|
|
pml_request->req_free_called = true;
|
|
} else {
|
|
/* can reuse the existing request */
|
|
OPAL_THREAD_UNLOCK(&ompi_request_lock);
|
|
break;
|
|
}
|
|
|
|
/* allocate a new request */
|
|
switch(pml_request->req_type) {
|
|
case MCA_PML_REQUEST_SEND: {
|
|
mca_pml_base_send_mode_t sendmode =
|
|
((mca_pml_base_send_request_t*)pml_request)->req_send_mode;
|
|
rc = mca_pml_dr_isend_init(
|
|
pml_request->req_addr,
|
|
pml_request->req_count,
|
|
pml_request->req_datatype,
|
|
pml_request->req_peer,
|
|
pml_request->req_tag,
|
|
sendmode,
|
|
pml_request->req_comm,
|
|
&request);
|
|
break;
|
|
}
|
|
case MCA_PML_REQUEST_RECV:
|
|
rc = mca_pml_dr_irecv_init(
|
|
pml_request->req_addr,
|
|
pml_request->req_count,
|
|
pml_request->req_datatype,
|
|
pml_request->req_peer,
|
|
pml_request->req_tag,
|
|
pml_request->req_comm,
|
|
&request);
|
|
break;
|
|
default:
|
|
rc = OMPI_ERR_REQUEST;
|
|
break;
|
|
}
|
|
OPAL_THREAD_UNLOCK(&ompi_request_lock);
|
|
if(OMPI_SUCCESS != rc)
|
|
return rc;
|
|
pml_request = (mca_pml_base_request_t*)request;
|
|
requests[i] = request;
|
|
break;
|
|
}
|
|
default:
|
|
return OMPI_ERR_REQUEST;
|
|
}
|
|
|
|
/* start the request */
|
|
switch(pml_request->req_type) {
|
|
case MCA_PML_REQUEST_SEND:
|
|
{
|
|
mca_pml_dr_send_request_t* sendreq = (mca_pml_dr_send_request_t*)pml_request;
|
|
MCA_PML_DR_SEND_REQUEST_START(sendreq, rc);
|
|
if(rc != OMPI_SUCCESS)
|
|
return rc;
|
|
break;
|
|
}
|
|
case MCA_PML_REQUEST_RECV:
|
|
{
|
|
mca_pml_dr_recv_request_t* recvreq = (mca_pml_dr_recv_request_t*)pml_request;
|
|
MCA_PML_DR_RECV_REQUEST_START(recvreq);
|
|
break;
|
|
}
|
|
default:
|
|
return OMPI_ERR_REQUEST;
|
|
}
|
|
}
|
|
return OMPI_SUCCESS;
|
|
}
|
|
|