Home Browse Files
      1
      2
      3
      4
      5
      6
      7
      8
      9
     10
     11
     12
     13
     14
     15
     16
     17
     18
     19
     20
     21
     22
     23
     24
     25
     26
     27
     28
     29
     30
     31
     32
     33
     34
     35
     36
     37
     38
     39
     40
     41
     42
     43
     44
     45
     46
     47
     48
     49
     50
     51
     52
     53
     54
     55
     56
     57
     58
     59
     60
     61
     62
     63
     64
     65
     66
     67
     68
     69
     70
     71
     72
     73
     74
     75
     76
     77
     78
     79
     80
     81
     82
     83
     84
     85
     86
     87
     88
     89
     90
#ifndef ULMBLAS_CXXBLAS_LEVEL1_AXPY_TCC
#define ULMBLAS_CXXBLAS_LEVEL1_AXPY_TCC 1

#include <ulmblas/cxxblas/level1/axpy.h>
#include <ulmblas/impl/level1/axpy.h>
#include <ulmblas/impl/level1extensions/geaxpy.h>
#include <ulmblas/ulmblas.h>

namespace cxxblas {

template <typename IndexType, typename Alpha, typename TX, typename TY>
void
axpy(IndexType    n,
     const Alpha  &alpha,
     const TX     *x,
     IndexType    incX,
     TY           *y,
     IndexType    incY)
{
    if (incX<0) {
        x -= incX*(n-1);
    }
    if (incY<0) {
        y -= incY*(n-1);
    }
    ulmBLAS::axpy(n, alpha, x, incX, y, incY);
}

template <typename IndexType, typename Alpha, typename TX, typename TY>
void
acxpy(IndexType    n,
      const Alpha  &alpha,
      const TX     *x,
      IndexType    incX,
      TY           *y,
      IndexType    incY)
{
    if (incX<0) {
        x -= incX*(n-1);
    }
    if (incY<0) {
        y -= incY*(n-1);
    }
    ulmBLAS::acxpy(n, alpha, x, incX, y, incY);
}

template <typename IndexType, typename Alpha, typename MX, typename MY>
void
geaxpy(IndexType      m,
       IndexType      n,
       const Alpha    &alpha,
       bool           transX,
       bool           conjX,
       const MX       *X,
       IndexType      incRowX,
       IndexType      incColX,
       MY             *Y,
       IndexType      incRowY,
       IndexType      incColY)
{
    if (!transX) {
        if (!conjX) {
            ulmBLAS::geaxpy(m, n,
                            alpha,
                            X, incRowX, incColX,
                            Y, incRowY, incColY);
        } else {
            ulmBLAS::geacxpy(m, n,
                             alpha,
                             X, incRowX, incColX,
                             Y, incRowY, incColY);
        }
    } else {
        if (!conjX) {
            ulmBLAS::geaxpy(m, n,
                            alpha,
                            X, incColX, incRowX,
                            Y, incRowY, incColY);
        } else {
            ulmBLAS::geacxpy(m, n,
                             alpha,
                             X, incColX, incRowX,
                             Y, incRowY, incColY);
        }
    }
}

} // namespace cxxblas

#endif // ULMBLAS_CXXBLAS_LEVEL1_AXPY_TCC





































































































Home