codelet_zpotrf.c

Go to the documentation of this file.

#include "morse_starpu.h"

/*
 * Codelet CPU
 */
static void cl_zpotrf_cpu_func(void *descr[], void *cl_arg)
{
    PLASMA_enum uplo;
    int N;
    PLASMA_Complex64_t *A;
    int LDA;
    int INFO = 0;

    A = (PLASMA_Complex64_t *)STARPU_MATRIX_GET_PTR(descr[0]);

    starpu_unpack_cl_args(cl_arg, &uplo, &N, &LDA);
    INFO = LAPACKE_zpotrf_work(LAPACK_COL_MAJOR, lapack_const(uplo), N, A, LDA);
}

/*
 * Codelet Multi-cores
 */
#ifdef MORSE_USE_MULTICORE
static void cl_zpotrf_mc_func(void *descr[], void *cl_arg)
{
    int uplo;
    int N;
    PLASMA_Complex64_t *A;
    int LDA;
    int INFO = 0;

    A = (PLASMA_Complex64_t *)STARPU_MATRIX_GET_PTR(descr[0]);

    starpu_unpack_cl_args(cl_arg, &uplo, &N, &LDA);
    INFO = PLASMA_zpotrf_Lapack(uplo, N, A, LDA);
}
#else
#define cl_zpotrf_mc_func cl_zpotrf_cpu_func
#endif

/*
 * Codelet GPU
 */
#ifdef MORSE_USE_CUDA
static void cl_zpotrf_cuda_func(void *descr[], void *cl_arg)
{
    int uplo;
    int N;
    cuDoubleComplex *A;
    int LDA;
    int INFO = 0;
    int  ret;

    A = (cuDoubleComplex *)STARPU_MATRIX_GET_PTR(descr[0]);

    starpu_unpack_cl_args(cl_arg, &uplo, &N, &LDA);

    ret = magma_zpotrf_gpu(
        plasma_lapack_constants[uplo][0],
        N, A, LDA, &INFO);
    if (ret != MAGMA_SUCCESS) {
        fprintf(stderr, "Error in Magma: %d\n", ret); 
        exit(-1);
    }
    cudaThreadSynchronize();
}
#endif

/*
 * Codelet definition
 */
CODELETS(zpotrf, 1, cl_zpotrf_cpu_func, cl_zpotrf_cuda_func, cl_zpotrf_mc_func)

/*
 * Wrapper
 */
void MORSE_zpotrf( MorseOption_t *option, 
                   PLASMA_enum uplo, int n, 
                   magma_desc_t *A, int Am, int An, int iinfo)
{
    starpu_codelet *zpotrf_codelet;
    void (*callback)(void*) = option->profiling ? cl_zpotrf_callback : NULL;
    int lda = BLKLDD( A, Am );

#ifdef MORSE_USE_MULTICORE
    zpotrf_codelet = option->parallel ? &cl_zpotrf_mc : &cl_zpotrf;
#else
    zpotrf_codelet = &cl_zpotrf;
#endif

    starpu_Insert_Task(zpotrf_codelet,
                       VALUE,        &uplo,  sizeof(PLASMA_enum),
                       VALUE,        &n,     sizeof(int),
                       INOUT,        BLKADDR( A, PLASMA_Complex64_t, Am, An ),
                       VALUE,        &lda,   sizeof(int),
                       PRIORITY,     option->priority,
                       CALLBACK,     callback, NULL,
                       0);

    /* TODO: take cancellation into account */
    /* if ( *info != 0 ) */
    /*     return (*info + iinfo); */
}