prt_request.c

Go to the documentation of this file.

#include "prt_request.h"


prt_request_t *prt_request_new(
    prt_packet_t *packet, size_t size, int peer, int tag)
{
    prt_request_t *request = (prt_request_t*)malloc(sizeof(prt_request_t));
    prt_assert(request != NULL, "malloc failed");
    request->packet = packet;
    request->size = size;
    request->peer = peer;
    request->tag = tag;
    return request;
}


void prt_request_delete(prt_request_t *request)
{
    free(request);
}


void prt_request_send(prt_request_t *request)
{
    int retval;
    // Post the request.
    svg_trace_start_cpu(0);
    retval = MPI_Isend(
        request->packet->data,
        request->size,
        MPI_BYTE,
        request->peer,
        request->tag,
        MPI_COMM_WORLD,
        &request->request);
    svg_trace_stop_cpu(0, -DarkKhaki);
    prt_assert(retval == MPI_SUCCESS, "error in MPI_Isend");
}


void prt_request_recv(prt_request_t *request)
{
    int retval;
    // Post the request.
    svg_trace_start_cpu(0);
    retval = MPI_Irecv(
        request->packet->data,
        request->size,
        MPI_BYTE,
        request->peer,
        request->tag,
        MPI_COMM_WORLD,
        &request->request);
    svg_trace_stop_cpu(0, -OliveDrab);
    prt_assert(retval == MPI_SUCCESS, "error in MPI_Isend");
}


int prt_request_test(prt_request_t *request)
{
    int flag;
    int retval;
    // Test the request.
    // Check the status.
    // Trace if completed.
    svg_trace_start_cpu(0);
    retval = MPI_Test(&request->request, &flag, &request->status);
    if (flag == 1)
        svg_trace_stop_cpu(0, -RosyBrown);
    prt_assert(retval == MPI_SUCCESS, "error in MPI_Test");
    return flag;
}


void prt_request_cancel(prt_request_t *request)
{
    int retval;
    // Cancel the request.
    retval = MPI_Cancel(&request->request);
    prt_assert(retval == MPI_SUCCESS, "error in MPI_Cancel");

    // Release the packet.
    prt_packet_release_host(request->packet);

    // Free the request.
    free(request);
}