#ifndef EIGEN_BENCH_UTIL_H #define EIGEN_BENCH_UTIL_H #include <Eigen/Core> #include "BenchTimer.h" using namespace std; using namespace Eigen; #include <boost/preprocessor/repetition/enum_params.hpp> #include <boost/preprocessor/repetition.hpp> #include <boost/preprocessor/seq.hpp> #include <boost/preprocessor/array.hpp> #include <boost/preprocessor/arithmetic.hpp> #include <boost/preprocessor/comparison.hpp> #include <boost/preprocessor/punctuation.hpp> #include <boost/preprocessor/punctuation/comma.hpp> #include <boost/preprocessor/stringize.hpp> template<typename MatrixType> void initMatrix_random(MatrixType& mat) __attribute__((noinline)); template<typename MatrixType> void initMatrix_random(MatrixType& mat) { mat.setRandom();// = MatrixType::random(mat.rows(), mat.cols()); } template<typename MatrixType> void initMatrix_identity(MatrixType& mat) __attribute__((noinline)); template<typename MatrixType> void initMatrix_identity(MatrixType& mat) { mat.setIdentity(); } #ifndef __INTEL_COMPILER #define DISABLE_SSE_EXCEPTIONS() { \ int aux; \ asm( \ "stmxcsr %[aux] \n\t" \ "orl $32832, %[aux] \n\t" \ "ldmxcsr %[aux] \n\t" \ : : [aux] "m" (aux)); \ } #else #define DISABLE_SSE_EXCEPTIONS() #endif #ifdef BENCH_GMM #include <gmm/gmm.h> template <typename EigenMatrixType, typename GmmMatrixType> void eiToGmm(const EigenMatrixType& src, GmmMatrixType& dst) { dst.resize(src.rows(),src.cols()); for (int j=0; j<src.cols(); ++j) for (int i=0; i<src.rows(); ++i) dst(i,j) = src.coeff(i,j); } #endif #ifdef BENCH_GSL #include <gsl/gsl_matrix.h> #include <gsl/gsl_linalg.h> #include <gsl/gsl_eigen.h> template <typename EigenMatrixType> void eiToGsl(const EigenMatrixType& src, gsl_matrix** dst) { for (int j=0; j<src.cols(); ++j) for (int i=0; i<src.rows(); ++i) gsl_matrix_set(*dst, i, j, src.coeff(i,j)); } #endif #ifdef BENCH_UBLAS #include <boost/numeric/ublas/matrix.hpp> #include <boost/numeric/ublas/vector.hpp> template <typename EigenMatrixType, typename UblasMatrixType> void eiToUblas(const EigenMatrixType& src, UblasMatrixType& dst) { dst.resize(src.rows(),src.cols()); for (int j=0; j<src.cols(); ++j) for (int i=0; i<src.rows(); ++i) dst(i,j) = src.coeff(i,j); } template <typename EigenType, typename UblasType> void eiToUblasVec(const EigenType& src, UblasType& dst) { dst.resize(src.size()); for (int j=0; j<src.size(); ++j) dst[j] = src.coeff(j); } #endif #endif // EIGEN_BENCH_UTIL_H