#ifndef TNet_Vector_cc #define TNet_Vector_cc #include #include #include #include #include #include "Common.h" #ifdef HAVE_ATLAS extern "C"{ #include } #endif #include "Common.h" #include "Matrix.h" #include "Vector.h" namespace TNet { #ifdef HAVE_ATLAS template<> float BlasDot<>(const Vector& rA, const Vector& rB) { assert(rA.mDim == rB.mDim); return cblas_sdot(rA.mDim, rA.pData(), 1, rB.pData(), 1); } template<> double BlasDot<>(const Vector& rA, const Vector& rB) { assert(rA.mDim == rB.mDim); return cblas_ddot(rA.mDim, rA.pData(), 1, rB.pData(), 1); } template<> Vector& Vector:: BlasAxpy(const float alpha, const Vector& rV) { assert(mDim == rV.mDim); cblas_saxpy(mDim, alpha, rV.pData(), 1, mpData, 1); return *this; } template<> Vector& Vector:: BlasAxpy(const double alpha, const Vector& rV) { assert(mDim == rV.mDim); cblas_daxpy(mDim, alpha, rV.pData(), 1, mpData, 1); return *this; } template<> Vector& Vector:: BlasAxpy(const int alpha, const Vector& rV) { assert(mDim == rV.mDim); for(int i=0; i Vector& Vector:: BlasGemv(const float alpha, const Matrix& rM, MatrixTrasposeType trans, const Vector& rV, const float beta) { assert((trans == NO_TRANS && rM.Cols() == rV.mDim && rM.Rows() == mDim) || (trans == TRANS && rM.Rows() == rV.mDim && rM.Cols() == mDim)); cblas_sgemv(CblasRowMajor, static_cast(trans), rM.Rows(), rM.Cols(), alpha, rM.pData(), rM.Stride(), rV.pData(), 1, beta, mpData, 1); return *this; } template<> Vector& Vector:: BlasGemv(const double alpha, const Matrix& rM, MatrixTrasposeType trans, const Vector& rV, const double beta) { assert((trans == NO_TRANS && rM.Cols() == rV.mDim && rM.Rows() == mDim) || (trans == TRANS && rM.Rows() == rV.mDim && rM.Cols() == mDim)); cblas_dgemv(CblasRowMajor, static_cast(trans), rM.Rows(), rM.Cols(), alpha, rM.pData(), rM.Stride(), rV.pData(), 1, beta, mpData, 1); return *this; } #else #error Routines in this section are not implemented yet without BLAS #endif } // namespace TNet #endif // TNet_Vector_tcc