core_cswpab.c

Go to the documentation of this file.

#include <stdlib.h>
#include "common.h"
#include "quark.h"

#if defined(PLASMA_HAVE_WEAK)
#pragma weak CORE_cswpab = PCORE_cswpab
#define CORE_cswpab PCORE_cswpab
#endif
void CORE_cswpab(int i, int n1, int n2, PLASMA_Complex32_t *A, PLASMA_Complex32_t *work) {
    PLASMA_Complex32_t *A0 = &(A[i]);
    PLASMA_Complex32_t *A1 = &(A[i+n1]);
    PLASMA_Complex32_t *A2 = &(A[i+n2]);
    int j;
    
    if( n1 < n2 ) {
        memcpy(work,  A0, n1*sizeof(PLASMA_Complex32_t));
        for (j=0; j<n2; j++)
            A0[j] = A1[j];
        memcpy(A2, work,  n1*sizeof(PLASMA_Complex32_t));
    } else {
        memcpy(work,  A1, n2*sizeof(PLASMA_Complex32_t));
        for (j=n1-1; j>-1; j--)
            A2[j] = A0[j];
        memcpy(A0, work,  n2*sizeof(PLASMA_Complex32_t));
    }
}

/***************************************************************************/
void QUARK_CORE_cswpab(Quark *quark, Quark_Task_Flags *task_flags,
                       int i, int n1, int n2,
                       PLASMA_Complex32_t *A, int szeA)
{
    DAG_CORE_SWPAB;
    QUARK_Insert_Task(
        quark, CORE_cswpab_quark, task_flags,
        sizeof(int),                           &i,   VALUE,
        sizeof(int),                           &n1,  VALUE,
        sizeof(int),                           &n2,  VALUE,
        sizeof(PLASMA_Complex32_t)*szeA,       A,            INOUT,
        sizeof(PLASMA_Complex32_t)*min(n1,n2), NULL,         SCRATCH,
        0);
}

/***************************************************************************/
#if defined(PLASMA_HAVE_WEAK)
#pragma weak CORE_cswpab_quark = PCORE_cswpab_quark
#define CORE_cswpab_quark PCORE_cswpab_quark
#endif
void CORE_cswpab_quark(Quark *quark)
{
    int i;
    int n1;
    int n2;
    PLASMA_Complex32_t *A;
    PLASMA_Complex32_t *work;
    
    quark_unpack_args_5(quark, i, n1, n2, A, work);
    CORE_cswpab( i, n1, n2, A, work);
}