BLAS-Impl.hpp

Go to the documentation of this file.

// This code is based on Jet framework.
// Copyright (c) 2018 Doyub Kim
// CubbyFlow is voxel-based fluid simulation engine for computer games.
// Copyright (c) 2020 CubbyFlow Team
// Core Part: Chris Ohk, Junwoo Hwang, Jihong Sin, Seungwoo Yoo
// AI Part: Dongheon Cho, Minseo Kim
// We are making my contributions/submissions to this project solely in our
// personal capacity and are not conveying any rights to any intellectual
// property of any third parties.

#ifndef CUBBYFLOW_BLAS_IMPL_HPP
#define CUBBYFLOW_BLAS_IMPL_HPP

#include <cmath>

namespace CubbyFlow
{
template <typename ScalarType, typename VectorType, typename MatrixType>
void BLAS<ScalarType, VectorType, MatrixType>::Set(ScalarType s,
                                                   VectorType* result)
{
    result->Fill(s);
}

template <typename ScalarType, typename VectorType, typename MatrixType>
void BLAS<ScalarType, VectorType, MatrixType>::Set(const VectorType& v,
                                                   VectorType* result)
{
    result->CopyFrom(v);
}

template <typename ScalarType, typename VectorType, typename MatrixType>
void BLAS<ScalarType, VectorType, MatrixType>::Set(ScalarType s,
                                                   MatrixType* result)
{
    result->Fill(s);
}

template <typename ScalarType, typename VectorType, typename MatrixType>
void BLAS<ScalarType, VectorType, MatrixType>::Set(const MatrixType& m,
                                                   MatrixType* result)
{
    result->CopyFrom(m);
}

template <typename ScalarType, typename VectorType, typename MatrixType>
ScalarType BLAS<ScalarType, VectorType, MatrixType>::Dot(const VectorType& a,
                                                         const VectorType& b)
{
    return a.Dot(b);
}

template <typename ScalarType, typename VectorType, typename MatrixType>
void BLAS<ScalarType, VectorType, MatrixType>::AXPlusY(ScalarType a,
                                                       const VectorType& x,
                                                       const VectorType& y,
                                                       VectorType* result)
{
    *result = a * x + y;
}

template <typename ScalarType, typename VectorType, typename MatrixType>
void BLAS<ScalarType, VectorType, MatrixType>::MVM(const MatrixType& m,
                                                   const VectorType& v,
                                                   VectorType* result)
{
    *result = m * v;
}

template <typename ScalarType, typename VectorType, typename MatrixType>
void BLAS<ScalarType, VectorType, MatrixType>::Residual(const MatrixType& a,
                                                        const VectorType& x,
                                                        const VectorType& b,
                                                        VectorType* result)
{
    *result = b - a * x;
}

template <typename ScalarType, typename VectorType, typename MatrixType>
ScalarType BLAS<ScalarType, VectorType, MatrixType>::L2Norm(const VectorType& v)
{
    return std::sqrt(v.Dot(v));
}

template <typename ScalarType, typename VectorType, typename MatrixType>
ScalarType BLAS<ScalarType, VectorType, MatrixType>::LInfNorm(
    const VectorType& v)
{
    return std::fabs(v.AbsMax());
}
}  // namespace CubbyFlow

#endif