1 2#ifndef EIGEN_BENCH_UTIL_H 3#define EIGEN_BENCH_UTIL_H 4 5#include <Eigen/Core> 6#include "BenchTimer.h" 7 8using namespace std; 9using namespace Eigen; 10 11#include <boost/preprocessor/repetition/enum_params.hpp> 12#include <boost/preprocessor/repetition.hpp> 13#include <boost/preprocessor/seq.hpp> 14#include <boost/preprocessor/array.hpp> 15#include <boost/preprocessor/arithmetic.hpp> 16#include <boost/preprocessor/comparison.hpp> 17#include <boost/preprocessor/punctuation.hpp> 18#include <boost/preprocessor/punctuation/comma.hpp> 19#include <boost/preprocessor/stringize.hpp> 20 21template<typename MatrixType> void initMatrix_random(MatrixType& mat) __attribute__((noinline)); 22template<typename MatrixType> void initMatrix_random(MatrixType& mat) 23{ 24 mat.setRandom();// = MatrixType::random(mat.rows(), mat.cols()); 25} 26 27template<typename MatrixType> void initMatrix_identity(MatrixType& mat) __attribute__((noinline)); 28template<typename MatrixType> void initMatrix_identity(MatrixType& mat) 29{ 30 mat.setIdentity(); 31} 32 33#ifndef __INTEL_COMPILER 34#define DISABLE_SSE_EXCEPTIONS() { \ 35 int aux; \ 36 asm( \ 37 "stmxcsr %[aux] \n\t" \ 38 "orl $32832, %[aux] \n\t" \ 39 "ldmxcsr %[aux] \n\t" \ 40 : : [aux] "m" (aux)); \ 41} 42#else 43#define DISABLE_SSE_EXCEPTIONS() 44#endif 45 46#ifdef BENCH_GMM 47#include <gmm/gmm.h> 48template <typename EigenMatrixType, typename GmmMatrixType> 49void eiToGmm(const EigenMatrixType& src, GmmMatrixType& dst) 50{ 51 dst.resize(src.rows(),src.cols()); 52 for (int j=0; j<src.cols(); ++j) 53 for (int i=0; i<src.rows(); ++i) 54 dst(i,j) = src.coeff(i,j); 55} 56#endif 57 58 59#ifdef BENCH_GSL 60#include <gsl/gsl_matrix.h> 61#include <gsl/gsl_linalg.h> 62#include <gsl/gsl_eigen.h> 63template <typename EigenMatrixType> 64void eiToGsl(const EigenMatrixType& src, gsl_matrix** dst) 65{ 66 for (int j=0; j<src.cols(); ++j) 67 for (int i=0; i<src.rows(); ++i) 68 gsl_matrix_set(*dst, i, j, src.coeff(i,j)); 69} 70#endif 71 72#ifdef BENCH_UBLAS 73#include <boost/numeric/ublas/matrix.hpp> 74#include <boost/numeric/ublas/vector.hpp> 75template <typename EigenMatrixType, typename UblasMatrixType> 76void eiToUblas(const EigenMatrixType& src, UblasMatrixType& dst) 77{ 78 dst.resize(src.rows(),src.cols()); 79 for (int j=0; j<src.cols(); ++j) 80 for (int i=0; i<src.rows(); ++i) 81 dst(i,j) = src.coeff(i,j); 82} 83template <typename EigenType, typename UblasType> 84void eiToUblasVec(const EigenType& src, UblasType& dst) 85{ 86 dst.resize(src.size()); 87 for (int j=0; j<src.size(); ++j) 88 dst[j] = src.coeff(j); 89} 90#endif 91 92#endif // EIGEN_BENCH_UTIL_H 93