xref: /aosp_15_r20/external/eigen/bench/basicbenchmark.cpp (revision bf2c37156dfe67e5dfebd6d394bad8b2ab5804d4)
1*bf2c3715SXin Li 
2*bf2c3715SXin Li #include <iostream>
3*bf2c3715SXin Li #include "BenchUtil.h"
4*bf2c3715SXin Li #include "basicbenchmark.h"
5*bf2c3715SXin Li 
main(int argc,char * argv[])6*bf2c3715SXin Li int main(int argc, char *argv[])
7*bf2c3715SXin Li {
8*bf2c3715SXin Li   DISABLE_SSE_EXCEPTIONS();
9*bf2c3715SXin Li 
10*bf2c3715SXin Li   // this is the list of matrix type and size we want to bench:
11*bf2c3715SXin Li   // ((suffix) (matrix size) (number of iterations))
12*bf2c3715SXin Li   #define MODES ((3d)(3)(4000000)) ((4d)(4)(1000000)) ((Xd)(4)(1000000)) ((Xd)(20)(10000))
13*bf2c3715SXin Li //   #define MODES ((Xd)(20)(10000))
14*bf2c3715SXin Li 
15*bf2c3715SXin Li   #define _GENERATE_HEADER(R,ARG,EL) << BOOST_PP_STRINGIZE(BOOST_PP_SEQ_HEAD(EL)) << "-" \
16*bf2c3715SXin Li     << BOOST_PP_STRINGIZE(BOOST_PP_SEQ_ELEM(1,EL)) << "x" \
17*bf2c3715SXin Li     << BOOST_PP_STRINGIZE(BOOST_PP_SEQ_ELEM(1,EL)) << "   /   "
18*bf2c3715SXin Li 
19*bf2c3715SXin Li   std::cout BOOST_PP_SEQ_FOR_EACH(_GENERATE_HEADER, ~, MODES ) << endl;
20*bf2c3715SXin Li 
21*bf2c3715SXin Li   const int tries = 10;
22*bf2c3715SXin Li 
23*bf2c3715SXin Li   #define _RUN_BENCH(R,ARG,EL) \
24*bf2c3715SXin Li     std::cout << ARG( \
25*bf2c3715SXin Li       BOOST_PP_CAT(Matrix, BOOST_PP_SEQ_HEAD(EL)) (\
26*bf2c3715SXin Li          BOOST_PP_SEQ_ELEM(1,EL),BOOST_PP_SEQ_ELEM(1,EL)), BOOST_PP_SEQ_ELEM(2,EL), tries) \
27*bf2c3715SXin Li     << "   ";
28*bf2c3715SXin Li 
29*bf2c3715SXin Li   BOOST_PP_SEQ_FOR_EACH(_RUN_BENCH, benchBasic<LazyEval>, MODES );
30*bf2c3715SXin Li   std::cout << endl;
31*bf2c3715SXin Li   BOOST_PP_SEQ_FOR_EACH(_RUN_BENCH, benchBasic<EarlyEval>, MODES );
32*bf2c3715SXin Li   std::cout << endl;
33*bf2c3715SXin Li 
34*bf2c3715SXin Li   return 0;
35*bf2c3715SXin Li }
36