ground/gcs/src/libs/eigen/bench/basicbenchmark.h

   1
   2 #ifndef EIGEN_BENCH_BASICBENCH_H
   3 #define EIGEN_BENCH_BASICBENCH_H
   4
   5 enum {LazyEval, EarlyEval, OmpEval};
   6
   7 template<int Mode, typename MatrixType>
   8 void benchBasic_loop(const MatrixType& I, MatrixType& m, int iterations) __attribute__((noinline));
   9
  10 template<int Mode, typename MatrixType>
  11 void benchBasic_loop(const MatrixType& I, MatrixType& m, int iterations)
  12 {
  13   for(int a = 0; a < iterations; a++)
  14   {
  15     if (Mode==LazyEval)
  16     {
  17       asm("#begin_bench_loop LazyEval");
  18       if (MatrixType::SizeAtCompileTime!=Eigen::Dynamic) asm("#fixedsize");
  19       m = (I + 0.00005 * (m + m.lazy() * m)).eval();
  20     }
  21     else if (Mode==OmpEval)
  22     {
  23       asm("#begin_bench_loop OmpEval");
  24       if (MatrixType::SizeAtCompileTime!=Eigen::Dynamic) asm("#fixedsize");
  25       m = (I + 0.00005 * (m + m.lazy() * m)).evalOMP();
  26     }
  27     else
  28     {
  29       asm("#begin_bench_loop EarlyEval");
  30       if (MatrixType::SizeAtCompileTime!=Eigen::Dynamic) asm("#fixedsize");
  31       m = I + 0.00005 * (m + m * m);
  32     }
  33     asm("#end_bench_loop");
  34   }
  35 }
  36
  37 template<int Mode, typename MatrixType>
  38 double benchBasic(const MatrixType& mat, int size, int tries) __attribute__((noinline));
  39
  40 template<int Mode, typename MatrixType>
  41 double benchBasic(const MatrixType& mat, int iterations, int tries)
  42 {
  43   const int rows = mat.rows();
  44   const int cols = mat.cols();
  45
  46   MatrixType I(rows,cols);
  47   MatrixType m(rows,cols);
  48
  49   initMatrix_identity(I);
  50
  51   Eigen::BenchTimer timer;
  52   for(uint t=0; t<tries; ++t)
  53   {
  54     initMatrix_random(m);
  55     timer.start();
  56     benchBasic_loop<Mode>(I, m, iterations);
  57     timer.stop();
  58     cerr << m;
  59   }
  60   return timer.value();
  61 };
  62
  63 #endif // EIGEN_BENCH_BASICBENCH_H