xmatrix5.h

Go to the documentation of this file.

#pragma once

#include "xmatrix4.h"

namespace nmx::gsl::mat {

template<class M, class V>
struct IMBlas {
    inline friend void dgemv(CBLAS_TRANSPOSE_t t, //
                             double a,
                             const M &m,
                             const V &x,
                             double b,
                             V &yinout) {
        gsl_blas_dgemv(t, a, m.gsl(), x.gsl(), b, yinout.gsl());
    }

    inline friend V amxby(CBLAS_TRANSPOSE_t t, //
                          double a,
                          const M &m,
                          const V &x,
                          double b,
                          const V &y) {
        Vector out(y);
        gsl_blas_dgemv(t, a, m.gsl(), x.gsl(), b, out.gsl());
        return out;
    }

    inline friend V amx(CBLAS_TRANSPOSE_t t, double a, const M &m,
                        const V &x) { //
        Vector out{ x };
        gsl_blas_dgemv(t, a, m.gsl(), x.gsl(), 0, out.gsl());
        return out;
    }

    inline friend V dot(const M &m, const V &v,
                        CBLAS_TRANSPOSE_t type = CblasNoTrans) { //
        return amx(type, 1, m, v);
    }

    inline friend void dgemm(CBLAS_TRANSPOSE_t t1,
                             CBLAS_TRANSPOSE_t t2,
                             double a,
                             const M &m1,
                             const M &m2,
                             double b,
                             M &minout) {
        gsl_blas_dgemm(t1, t2, a, m1.gsl(), m2.gsl(), b, minout.gsl());
    }

    inline friend M dgemm(CBLAS_TRANSPOSE_t t1, //
                          CBLAS_TRANSPOSE_t t2,
                          double a,
                          const M &m1,
                          const M &m2,
                          double b,
                          const M &m3) {
        M out{ m3 };
        gsl_blas_dgemm(t1, t2, a, m1.gsl(), m2.gsl(), b, out.gsl());
        return out;
    }

    inline friend M dgemm(CBLAS_TRANSPOSE_t t1,
                          CBLAS_TRANSPOSE_t t2, //
                          double a,
                          const M &m1,
                          const M &m2) {
        M out(m1.rows(), m2.columns());
        gsl_blas_dgemm(t1, t2, a, m1.gsl(), m2.gsl(), 0, out.gsl());
        return out;
    }

    inline friend M dot(const M &m1, //
                        const M &m2,
                        CBLAS_TRANSPOSE_t t1 = CblasNoTrans,
                        CBLAS_TRANSPOSE_t t2 = CblasNoTrans) {
        return dgemm(t1, t2, 1, m1, m2);
    }
};

} // namespace nmx::gsl::mat