75 строки
2.1 KiB
C
75 строки
2.1 KiB
C
|
#include "request/grequest.h"
|
||
|
|
|
||
|
|
|
||
|
|
static int ompi_grequest_query(ompi_request_t* req, ompi_status_public_t* status)
|
||
|
|
{
|
||
|
|
int rc = OMPI_SUCCESS;
|
||
|
|
ompi_grequest_t* greq = (ompi_grequest_t*)req;
|
||
|
|
if(greq->greq_query != NULL)
|
||
|
|
rc = greq->greq_query(greq->greq_state, status);
|
||
|
|
return rc;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int ompi_grequest_free(ompi_request_t* req)
|
||
|
|
{
|
||
|
|
int rc = OMPI_SUCCESS;
|
||
|
|
ompi_grequest_t* greq = (ompi_grequest_t*)req;
|
||
|
|
if(greq->greq_free != NULL)
|
||
|
|
rc = greq->greq_free(greq->greq_state);
|
||
|
|
OBJ_RELEASE(req);
|
||
|
|
return rc;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int ompi_grequest_cancel(ompi_request_t* req, int flag)
|
||
|
|
{
|
||
|
|
int rc = OMPI_SUCCESS;
|
||
|
|
ompi_grequest_t* greq = (ompi_grequest_t*)req;
|
||
|
|
if(greq->greq_cancel != NULL)
|
||
|
|
rc = greq->greq_cancel(greq->greq_state, flag);
|
||
|
|
return rc;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void ompi_grequest_construct(ompi_grequest_t* greq)
|
||
|
|
{
|
||
|
|
OMPI_REQUEST_INIT(&greq->greq_base);
|
||
|
|
greq->greq_base.req_query = ompi_grequest_query;
|
||
|
|
greq->greq_base.req_free = ompi_grequest_free;
|
||
|
|
greq->greq_base.req_cancel = ompi_grequest_cancel;
|
||
|
|
greq->greq_base.req_type = OMPI_REQUEST_GEN;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
static void ompi_grequest_destruct(ompi_grequest_t* greq)
|
||
|
|
{
|
||
|
|
OMPI_REQUEST_FINI(&greq->greq_base);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
OBJ_CLASS_INSTANCE(
|
||
|
|
ompi_grequest_t,
|
||
|
|
ompi_request_t,
|
||
|
|
ompi_grequest_construct,
|
||
|
|
ompi_grequest_destruct);
|
||
|
|
|
||
|
|
|
||
|
|
int ompi_grequest_start(
|
||
|
|
MPI_Grequest_query_function *gquery_fn,
|
||
|
|
MPI_Grequest_free_function *gfree_fn,
|
||
|
|
MPI_Grequest_cancel_function *gcancel_fn,
|
||
|
|
void* gstate,
|
||
|
|
ompi_request_t** request)
|
||
|
|
{
|
||
|
|
ompi_grequest_t *greq = OBJ_NEW(ompi_grequest_t);
|
||
|
|
if(greq == NULL) {
|
||
|
|
return OMPI_ERR_OUT_OF_RESOURCE;
|
||
|
|
}
|
||
|
|
|
||
|
|
greq->greq_state = gstate;
|
||
|
|
greq->greq_query = gquery_fn;
|
||
|
|
greq->greq_free = gfree_fn;
|
||
|
|
greq->greq_cancel = gcancel_fn;
|
||
|
|
*request = &greq->greq_base;
|
||
|
|
return OMPI_SUCCESS;
|
||
|
|
}
|
||
|
|
|