1 2 #ifndef EIGEN_BENCH_UTIL_H 3 #define EIGEN_BENCH_UTIL_H 4 5 #include <Eigen/Core> 6 #include "BenchTimer.h" 7 8 using namespace std; 9 using 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 21 template<typename MatrixType> void initMatrix_random(MatrixType& mat) __attribute__((noinline)); 22 template<typename MatrixType> void initMatrix_random(MatrixType& mat) 23 { 24 mat.setRandom();// = MatrixType::random(mat.rows(), mat.cols()); 25 } 26 27 template<typename MatrixType> void initMatrix_identity(MatrixType& mat) __attribute__((noinline)); 28 template<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> 48 template <typename EigenMatrixType, typename GmmMatrixType> 49 void 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> 63 template <typename EigenMatrixType> 64 void 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> 75 template <typename EigenMatrixType, typename UblasMatrixType> 76 void 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 } 83 template <typename EigenType, typename UblasType> 84 void 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
