blas.c

Go to the documentation of this file.

/*
 * This work is licensed under a Creative Commons 
 * Attribution-Noncommercial-Share Alike 3.0 United States License.
 * 
 *    http://creativecommons.org/licenses/by-nc-sa/3.0/us/
 * 
 * You are free:
 * 
 *    to Share - to copy, distribute, display, and perform the work
 *    to Remix - to make derivative works
 * 
 * Under the following conditions:
 * 
 *    Attribution. You must attribute the work in the manner specified by the
 *    author or licensor (but not in any way that suggests that they endorse you
 *    or your use of the work).
 * 
 *    Noncommercial. You may not use this work for commercial purposes.
 * 
 *    Share Alike. If you alter, transform, or build upon this work, you may
 *    distribute the resulting work only under the same or similar license to
 *    this one.
 * 
 * For any reuse or distribution, you must make clear to others the license
 * terms of this work. The best way to do this is by including this header.
 * 
 * Any of the above conditions can be waived if you get permission from the
 * copyright holder.
 * 
 * Apart from the remix rights granted under this license, nothing in this
 * license impairs or restricts the author's moral rights.
 */


#include <jwsc/config.h>

#include <stdlib.h>
#include <assert.h>

#include <Accelerate/Accelerate.h>

#include "jwsc/math/complex.h"
#include "jwsc/math/blas.h"


/* ===============  LEVEL 1 ==============*/


void blas_sswap(int n, float* x, int inc_x, float* y, int inc_y)
{
    cblas_sswap(n, x, inc_x, y, inc_y);
}

void blas_dswap(int n, double* x, int inc_x, double* y, int inc_y)
{
    cblas_dswap(n, x, inc_x, y, inc_y);
}

void blas_cswap(int n, Complex_f* x, int inc_x, Complex_f* y, int inc_y)
{
    cblas_cswap(n, x, inc_x, y, inc_y);
}

void blas_zswap(int n, Complex_d* x, int inc_x, Complex_d* y, int inc_y)
{
    cblas_zswap(n, x, inc_x, y, inc_y);
}

void blas_sscal(int n, float a, float* x, int inc_x)
{
    cblas_sscal(n, a, x, inc_x);
}

void blas_dscal(int n, double a, double* x, int inc_x)
{
    cblas_dscal(n, a, x, inc_x);
}

void blas_cscal(int n, Complex_f a, Complex_f* x, int inc_x)
{
    cblas_cscal(n, &a, x, inc_x);
}

void blas_zscal(int n, Complex_d a, Complex_d* x, int inc_x)
{
    cblas_zscal(n, &a, x, inc_x);
}

void blas_scopy(int n, const float* x, int inc_x, float* y, int inc_y)
{
    cblas_scopy(n, (float*)x, inc_x, y, inc_y);
}

void blas_dcopy(int n, const double* x, int inc_x, double* y, int inc_y)
{
    cblas_dcopy(n, (double*)x, inc_x, y, inc_y);
}

void blas_ccopy(int n, const Complex_f* x, int inc_x, Complex_f* y, int inc_y)
{
    cblas_ccopy(n, (Complex_f*)x, inc_x, y, inc_y);
}

void blas_zcopy(int n, const Complex_d* x, int inc_x, Complex_d* y, int inc_y)
{
    cblas_zcopy(n, (Complex_d*)x, inc_x, y, inc_y);
}

void blas_saxpy
(
    int          n, 
    float        a, 
    const float* x, 
    int          inc_x, 
    float*       y, 
    int          inc_y
)
{
    cblas_saxpy(n, a, (float*)x, inc_x, y, inc_y);
}

void blas_daxpy
(
    int           n, 
    double        a, 
    const double* x, 
    int           inc_x, 
    double*       y, 
    int           inc_y
)
{
    cblas_daxpy(n, a, (double*)x, inc_x, y, inc_y);
}

void blas_caxpy
(
    int              n, 
    Complex_f        a, 
    const Complex_f* x, 
    int              inc_x, 
    Complex_f*       y, 
    int              inc_y
)
{
    cblas_caxpy(n, &a, (Complex_f*)x, inc_x, y, inc_y);
}

void blas_zaxpy
(
    int              n, 
    Complex_d        a, 
    const Complex_d* x, 
    int              inc_x, 
    Complex_d*       y, 
    int              inc_y
)
{
    cblas_zaxpy(n, &a, (Complex_d*)x, inc_x, y, inc_y);
}

float blas_sdot(int n, const float* x, int inc_x, const float* y, int inc_y)
{
    return cblas_sdot(n, (float*)x, inc_x, (float*)y, inc_y);
}

double blas_ddot(int n, const double* x, int inc_x, const double* y, int inc_y)
{
    return cblas_ddot(n, (double*)x, inc_x, (double*)y, inc_y);
}

float blas_snrm2(int n, const float* x, int inc_x)
{
    return cblas_snrm2(n, (float*)x, inc_x);
}

double blas_dnrm2(int n, const double* x, int inc_x)
{
    return cblas_dnrm2(n, (double*)x, inc_x);
}

float blas_sasum(int n, const float* x, int inc_x)
{
    return cblas_sasum(n, (float*)x, inc_x);
}

double blas_dasum(int n, const double* x, int inc_x)
{
    return cblas_dasum(n, (double*)x, inc_x);
}

int blas_isamax(int n, const float* x, int inc_x)
{
    return cblas_isamax(n, (float*)x, inc_x);
}

int blas_idamax(int n, const double* x, int inc_x)
{
    return cblas_idamax(n, (double*)x, inc_x);
}

int blas_icamax(int n, const Complex_f* x, int inc_x)
{
    return cblas_icamax(n, (Complex_f*)x, inc_x);
}

int blas_izamax(int n, const Complex_d* x, int inc_x)
{
    return cblas_izamax(n, (Complex_d*)x, inc_x);
}

/* ===============  LEVEL 2 PROTOTYPES ==============*/


void blas_sgemv
(
    char         trans, 
    int          m, 
    int          n, 
    float        alpha, 
    const float* A, 
    int          lda,
    const float* x,
    int          inc_x,
    float        beta,
    float*       y,
    int          inc_y
)
{
    trans = (trans == 'N' || trans == 'n') ? CblasTrans : CblasNoTrans;
    cblas_sgemv(CblasColMajor, trans, n, m, alpha, (float*)A, lda, (float*)x,
            inc_x, beta, y, inc_y);
}

void blas_dgemv
(
    char          trans, 
    int           m, 
    int           n, 
    double        alpha, 
    const double* A, 
    int           lda,
    const double* x,
    int           inc_x,
    double        beta,
    double*       y,
    int           inc_y
)
{
    trans = (trans == 'N' || trans == 'n') ? CblasTrans : CblasNoTrans;
    cblas_dgemv(CblasColMajor, trans, n, m, alpha, (double*)A, lda, (double*)x,
            inc_x, beta, y, inc_y);
}

void blas_cgemv
(
    char             trans, 
    int              m, 
    int              n, 
    Complex_f        alpha, 
    const Complex_f* A, 
    int              lda,
    const Complex_f* x,
    int              inc_x,
    Complex_f        beta,
    Complex_f*       y,
    int              inc_y
)
{
    trans = (trans == 'N' || trans == 'n') ? CblasTrans : CblasNoTrans;
    cblas_cgemv(CblasColMajor, trans, n, m, &alpha, (Complex_f*)A, lda,
            (Complex_f*)x, inc_x, &beta, y, inc_y);
}

void blas_zgemv
(
    char             trans, 
    int              m, 
    int              n, 
    Complex_d        alpha, 
    const Complex_d* A, 
    int              lda,
    const Complex_d* x,
    int              inc_x,
    Complex_d        beta,
    Complex_d*       y,
    int              inc_y
)
{
    trans = (trans == 'N' || trans == 'n') ? CblasTrans : CblasNoTrans;
    cblas_zgemv(CblasColMajor, trans, n, m, &alpha, (Complex_d*)A, lda,
            (Complex_d*)x, inc_x, &beta, y, inc_y);
}

/* ===============  LEVEL 3 ==============*/


void blas_sgemm
(
    char         trans_a, 
    char         trans_b, 
    int          m, 
    int          n, 
    int          k, 
    float        alpha, 
    const float* A, 
    int          lda,
    const float* B,
    int          ldb,
    float        beta,
    float*       C,
    int          ldc
)
{
    trans_a = (trans_a == 'N' || trans_a == 'n') ? CblasNoTrans : CblasTrans;
    trans_b = (trans_b == 'N' || trans_b == 'n') ? CblasNoTrans : CblasTrans;
    cblas_sgemm(CblasColMajor, trans_a, trans_b, n, m, k, alpha, (float*) B,
            ldb, (float*) A, lda, beta, C, ldc);
}

void blas_dgemm
(
    char          trans_a, 
    char          trans_b, 
    int           m, 
    int           n, 
    int           k, 
    double        alpha, 
    const double* A, 
    int           lda,
    const double* B,
    int           ldb,
    double        beta,
    double*       C,
    int           ldc
)
{
    trans_a = (trans_a == 'N' || trans_a == 'n') ? CblasNoTrans : CblasTrans;
    trans_b = (trans_b == 'N' || trans_b == 'n') ? CblasNoTrans : CblasTrans;
    cblas_dgemm(CblasColMajor, trans_a, trans_b, n, m, k, alpha, (double*) B,
            ldb, (double*) A, lda, beta, C, ldc);
}

void blas_cgemm
(
    char             trans_a, 
    char             trans_b, 
    int              m, 
    int              n, 
    int              k, 
    Complex_f        alpha, 
    const Complex_f* A, 
    int              lda,
    const Complex_f* B,
    int              ldb,
    Complex_f        beta,
    Complex_f*       C,
    int              ldc
)
{
    trans_a = (trans_a == 'N' || trans_a == 'n') ? CblasNoTrans : CblasTrans;
    trans_b = (trans_b == 'N' || trans_b == 'n') ? CblasNoTrans : CblasTrans;
    cblas_cgemm(CblasColMajor, trans_a, trans_b, n, m, k, &alpha, (Complex_f*)
            B, ldb, (Complex_f*) A, lda, &beta, C, ldc);
}

void blas_zgemm
(
    char             trans_a, 
    char             trans_b, 
    int              m, 
    int              n, 
    int              k, 
    Complex_d        alpha, 
    const Complex_d* A, 
    int              lda,
    const Complex_d* B,
    int              ldb,
    Complex_d        beta,
    Complex_d*       C,
    int              ldc
)
{
    trans_a = (trans_a == 'N' || trans_a == 'n') ? CblasNoTrans : CblasTrans;
    trans_b = (trans_b == 'N' || trans_b == 'n') ? CblasNoTrans : CblasTrans;
    cblas_zgemm(CblasColMajor, trans_a, trans_b, n, m, k, &alpha, (Complex_d*)
            B, ldb, (Complex_d*) A, lda, &beta, C, ldc);
}