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
|
---|