BLAS.h

Go to the documentation of this file.

/*************************************************************************
> File Name: BLAS.h
> Project Name: CubbyFlow
> Author: Chan-Ho Chris Ohk
> Purpose: Generic BLAS operator wrapper class.
> Created Time: 2017/08/13
> Copyright (c) 2018, Chan-Ho Chris Ohk
*************************************************************************/
#ifndef CUBBYFLOW_BLAS_H
#define CUBBYFLOW_BLAS_H

namespace CubbyFlow
{
    template <typename S, typename V, typename M>
    struct BLAS
    {
        using ScalarType = S;
        using VectorType = V;
        using MatrixType = M;

        static void Set(ScalarType s, VectorType* result);

        static void Set(const VectorType& v, VectorType* result);

        static void Set(ScalarType s, MatrixType* result);

        static void Set(const MatrixType& m, MatrixType* result);

        static ScalarType Dot(const VectorType& a, const VectorType& b);

        static void AXPlusY(ScalarType a, const VectorType& x, const VectorType& y, VectorType* result);

        static void MVM(const MatrixType& m, const VectorType& v, VectorType* result);

        static void Residual(const MatrixType& a, const VectorType& x, const VectorType& b, VectorType* result);

        static ScalarType L2Norm(const VectorType& v);

        static ScalarType LInfNorm(const VectorType& v);
    };
}

#include <Core/Math/BLAS-Impl.h>

#endif