xref: /aosp_15_r20/external/eigen/bench/sparse_transpose.cpp (revision bf2c37156dfe67e5dfebd6d394bad8b2ab5804d4)
1*bf2c3715SXin Li 
2*bf2c3715SXin Li //g++ -O3 -g0 -DNDEBUG  sparse_transpose.cpp -I.. -I/home/gael/Coding/LinearAlgebra/mtl4/ -DDENSITY=0.005 -DSIZE=10000 && ./a.out
3*bf2c3715SXin Li // -DNOGMM -DNOMTL
4*bf2c3715SXin Li // -DCSPARSE -I /home/gael/Coding/LinearAlgebra/CSparse/Include/ /home/gael/Coding/LinearAlgebra/CSparse/Lib/libcsparse.a
5*bf2c3715SXin Li 
6*bf2c3715SXin Li #ifndef SIZE
7*bf2c3715SXin Li #define SIZE 10000
8*bf2c3715SXin Li #endif
9*bf2c3715SXin Li 
10*bf2c3715SXin Li #ifndef DENSITY
11*bf2c3715SXin Li #define DENSITY 0.01
12*bf2c3715SXin Li #endif
13*bf2c3715SXin Li 
14*bf2c3715SXin Li #ifndef REPEAT
15*bf2c3715SXin Li #define REPEAT 1
16*bf2c3715SXin Li #endif
17*bf2c3715SXin Li 
18*bf2c3715SXin Li #include "BenchSparseUtil.h"
19*bf2c3715SXin Li 
20*bf2c3715SXin Li #ifndef MINDENSITY
21*bf2c3715SXin Li #define MINDENSITY 0.0004
22*bf2c3715SXin Li #endif
23*bf2c3715SXin Li 
24*bf2c3715SXin Li #ifndef NBTRIES
25*bf2c3715SXin Li #define NBTRIES 10
26*bf2c3715SXin Li #endif
27*bf2c3715SXin Li 
28*bf2c3715SXin Li #define BENCH(X) \
29*bf2c3715SXin Li   timer.reset(); \
30*bf2c3715SXin Li   for (int _j=0; _j<NBTRIES; ++_j) { \
31*bf2c3715SXin Li     timer.start(); \
32*bf2c3715SXin Li     for (int _k=0; _k<REPEAT; ++_k) { \
33*bf2c3715SXin Li         X  \
34*bf2c3715SXin Li   } timer.stop(); }
35*bf2c3715SXin Li 
main(int argc,char * argv[])36*bf2c3715SXin Li int main(int argc, char *argv[])
37*bf2c3715SXin Li {
38*bf2c3715SXin Li   int rows = SIZE;
39*bf2c3715SXin Li   int cols = SIZE;
40*bf2c3715SXin Li   float density = DENSITY;
41*bf2c3715SXin Li 
42*bf2c3715SXin Li   EigenSparseMatrix sm1(rows,cols), sm3(rows,cols);
43*bf2c3715SXin Li 
44*bf2c3715SXin Li   BenchTimer timer;
45*bf2c3715SXin Li   for (float density = DENSITY; density>=MINDENSITY; density*=0.5)
46*bf2c3715SXin Li   {
47*bf2c3715SXin Li     fillMatrix(density, rows, cols, sm1);
48*bf2c3715SXin Li 
49*bf2c3715SXin Li     // dense matrices
50*bf2c3715SXin Li     #ifdef DENSEMATRIX
51*bf2c3715SXin Li     {
52*bf2c3715SXin Li       DenseMatrix m1(rows,cols), m3(rows,cols);
53*bf2c3715SXin Li       eiToDense(sm1, m1);
54*bf2c3715SXin Li       BENCH(for (int k=0; k<REPEAT; ++k) m3 = m1.transpose();)
55*bf2c3715SXin Li       std::cout << "  Eigen dense:\t" << timer.value() << endl;
56*bf2c3715SXin Li     }
57*bf2c3715SXin Li     #endif
58*bf2c3715SXin Li 
59*bf2c3715SXin Li     std::cout << "Non zeros: " << sm1.nonZeros()/float(sm1.rows()*sm1.cols())*100 << "%\n";
60*bf2c3715SXin Li 
61*bf2c3715SXin Li     // eigen sparse matrices
62*bf2c3715SXin Li     {
63*bf2c3715SXin Li       BENCH(for (int k=0; k<REPEAT; ++k) sm3 = sm1.transpose();)
64*bf2c3715SXin Li       std::cout << "  Eigen:\t" << timer.value() << endl;
65*bf2c3715SXin Li     }
66*bf2c3715SXin Li 
67*bf2c3715SXin Li     // CSparse
68*bf2c3715SXin Li     #ifdef CSPARSE
69*bf2c3715SXin Li     {
70*bf2c3715SXin Li       cs *m1, *m3;
71*bf2c3715SXin Li       eiToCSparse(sm1, m1);
72*bf2c3715SXin Li 
73*bf2c3715SXin Li       BENCH(for (int k=0; k<REPEAT; ++k) { m3 = cs_transpose(m1,1); cs_spfree(m3);})
74*bf2c3715SXin Li       std::cout << "  CSparse:\t" << timer.value() << endl;
75*bf2c3715SXin Li     }
76*bf2c3715SXin Li     #endif
77*bf2c3715SXin Li 
78*bf2c3715SXin Li     // GMM++
79*bf2c3715SXin Li     #ifndef NOGMM
80*bf2c3715SXin Li     {
81*bf2c3715SXin Li       GmmDynSparse  gmmT3(rows,cols);
82*bf2c3715SXin Li       GmmSparse m1(rows,cols), m3(rows,cols);
83*bf2c3715SXin Li       eiToGmm(sm1, m1);
84*bf2c3715SXin Li       BENCH(for (int k=0; k<REPEAT; ++k) gmm::copy(gmm::transposed(m1),m3);)
85*bf2c3715SXin Li       std::cout << "  GMM:\t\t" << timer.value() << endl;
86*bf2c3715SXin Li     }
87*bf2c3715SXin Li     #endif
88*bf2c3715SXin Li 
89*bf2c3715SXin Li     // MTL4
90*bf2c3715SXin Li     #ifndef NOMTL
91*bf2c3715SXin Li     {
92*bf2c3715SXin Li       MtlSparse m1(rows,cols), m3(rows,cols);
93*bf2c3715SXin Li       eiToMtl(sm1, m1);
94*bf2c3715SXin Li       BENCH(for (int k=0; k<REPEAT; ++k) m3 = trans(m1);)
95*bf2c3715SXin Li       std::cout << "  MTL4:\t\t" << timer.value() << endl;
96*bf2c3715SXin Li     }
97*bf2c3715SXin Li     #endif
98*bf2c3715SXin Li 
99*bf2c3715SXin Li     std::cout << "\n\n";
100*bf2c3715SXin Li   }
101*bf2c3715SXin Li 
102*bf2c3715SXin Li   return 0;
103*bf2c3715SXin Li }
104*bf2c3715SXin Li 
105