codelet_ddotp.c

Go to the documentation of this file.

#include "../include/starpu_exageostat.h"
static void CORE_ddotp_starpu(void *buffers[], void *cl_arg){
        int m, m0;
        double * A;
        double * dotproduct;

        dotproduct  = (double *)STARPU_MATRIX_GET_PTR(buffers[0]);
        A       = (double *)STARPU_MATRIX_GET_PTR(buffers[1]);
        starpu_codelet_unpack_args(cl_arg, &m, &m0);
        double local_dot=cblas_ddot(m, A, 1, A, 1);
        *dotproduct += local_dot;
}

static struct starpu_codelet cl_ddotp =
{
                .where      = STARPU_CPU,
                .cpu_funcs  = {CORE_ddotp_starpu},
                .nbuffers   = 2,
                .modes      = {STARPU_RW,STARPU_R},
        .name       = "ddotp"
};

/***************************************************************************/
int MORSE_MLE_ddotp_Async(MORSE_desc_t *descA, MORSE_desc_t *descproduct, MORSE_sequence_t *sequence, MORSE_request_t  *request) {

        MORSE_context_t *morse;
        MORSE_option_t options;
        morse = morse_context_self();
        if (sequence->status != MORSE_SUCCESS)
                return -2;
        RUNTIME_options_init(&options, morse, sequence, request);

        int m, m0;
        int tempmm;
        MORSE_desc_t A = *descA;

        struct starpu_codelet *cl=&cl_ddotp;

        for (m = 0; m < A.mt; m++) {
                tempmm = m == A.mt-1 ? A.m - m * A.mb : A.mb;

                m0 = m * A.mb;


                starpu_insert_task(starpu_mpi_codelet(cl),
                                STARPU_VALUE, &tempmm, sizeof(int),
                                STARPU_VALUE, &m0,   sizeof(int),
                                STARPU_RW, RTBLKADDR(descproduct, sizeof(double), 0, 0),
                                STARPU_R, RTBLKADDR(descA, sizeof(double)*tempmm, m, 0),
                     0);
        }

    //MORSE_TASK_flush_desc( &options, MorseUpperLower, descA);
        //MORSE_TASK_flush_desc( &options, MorseUpperLower, descproduct);   
        RUNTIME_options_ws_free(&options);
        RUNTIME_options_finalize(&options, morse);
        //MORSE_TASK_dataflush_all();
        MORSE_Sequence_Wait(sequence);
        return MORSE_SUCCESS;
}