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