mirror/eigen
2
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-01-30 17:40:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

extend sparse product benchmark with ublas

Browse Source
This commit is contained in:
Gael Guennebaud 2010-02-09 15:55:36 +01:00
parent 285bc336d5
commit 905050b239
2 changed files with 98 additions and 63 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
bench/BenchSparseUtil.h
View File

@ -123,4 +123,18 @@ void eiToCSparse(const EigenSparseMatrix& src, cs* &dst)
#include <boost/numeric/ublas/matrix_sparse.hpp> #include <boost/numeric/ublas/matrix_sparse.hpp>
#include <boost/numeric/ublas/vector_of_vector.hpp> #include <boost/numeric/ublas/vector_of_vector.hpp>
// using namespace boost;
// using namespace boost::numeric;
// using namespace boost::numeric::ublas;
typedef boost::numeric::ublas::compressed_matrix<Scalar,boost::numeric::ublas::column_major> UblasMatrix;
void eiToUblas(const EigenSparseMatrix& src, UblasMatrix& dst)
{
for (int j=0; j<src.cols(); ++j)
for (EigenSparseMatrix::InnerIterator it(src.derived(), j); it; ++it)
dst(it.index(),j) = it.value();
}
#endif #endif

147
bench/sparse_product.cpp
View File

@ -3,12 +3,15 @@
//g++ -O3 -g0 -DNDEBUG sparse_product.cpp -I.. -I/home/gael/Coding/LinearAlgebra/mtl4/ -DDENSITY=0.05 -DSIZE=2000 && ./a.out //g++ -O3 -g0 -DNDEBUG sparse_product.cpp -I.. -I/home/gael/Coding/LinearAlgebra/mtl4/ -DDENSITY=0.05 -DSIZE=2000 && ./a.out
// -DNOGMM -DNOMTL -DCSPARSE // -DNOGMM -DNOMTL -DCSPARSE
// -I /home/gael/Coding/LinearAlgebra/CSparse/Include/ /home/gael/Coding/LinearAlgebra/CSparse/Lib/libcsparse.a // -I /home/gael/Coding/LinearAlgebra/CSparse/Include/ /home/gael/Coding/LinearAlgebra/CSparse/Lib/libcsparse.a
#include <typeinfo>
#ifndef SIZE #ifndef SIZE
#define SIZE 10000 #define SIZE 1000000
#endif #endif
#ifndef NNZPERCOL #ifndef NNZPERCOL
#define NNZPERCOL 32 #define NNZPERCOL 6
#endif #endif
#ifndef REPEAT #ifndef REPEAT
@ -20,7 +23,7 @@
#include "BenchSparseUtil.h" #include "BenchSparseUtil.h"
#ifndef NBTRIES #ifndef NBTRIES
#define NBTRIES 4 #define NBTRIES 1
#endif #endif
#define BENCH(X) \ #define BENCH(X) \
@ -60,6 +63,7 @@
cs* cs_sorted_multiply(const cs* a, const cs* b) cs* cs_sorted_multiply(const cs* a, const cs* b)
{ {
// return cs_multiply(a,b); // return cs_multiply(a,b);
cs* A = cs_transpose(a, 1); cs* A = cs_transpose(a, 1);
cs* B = cs_transpose(b, 1); cs* B = cs_transpose(b, 1);
cs* D = cs_multiply(B,A); /* D = B'*A' */ cs* D = cs_multiply(B,A); /* D = B'*A' */
@ -155,26 +159,26 @@ int main(int argc, char *argv[])
// BENCH(sm3 = sm1.transpose() * sm2; ) // BENCH(sm3 = sm1.transpose() * sm2; )
// std::cout << " a' * b:\t" << timer.value() << endl; // std::cout << " a' * b:\t" << timer.value() << endl;
// // //
// BENCH(sm3 = sm1.transpose() * sm2.transpose(); ) // BENCH(sm3 = sm1.transpose() * sm2.transpose(); )
// std::cout << " a' * b':\t" << timer.value() << endl; // std::cout << " a' * b':\t" << timer.value() << endl;
// // //
// BENCH(sm3 = sm1 * sm2.transpose(); ) // BENCH(sm3 = sm1 * sm2.transpose(); )
// std::cout << " a * b' :\t" << timer.value() << endl; // std::cout << " a * b' :\t" << timer.value() << endl;
// std::cout << "\n\n"; // std::cout << "\n";
//
BENCH( sm3.setprod(sm1, sm2); ) // BENCH( sm3._experimentalNewProduct(sm1, sm2); )
std::cout << " a * b:\t" << timer.value() << endl; // std::cout << " a * b:\t" << timer.value() << endl;
//
// BENCH(sm3.setprod(sm1.transpose(),sm2); ) // BENCH(sm3._experimentalNewProduct(sm1.transpose(),sm2); )
// std::cout << " a' * b:\t" << timer.value() << endl; // std::cout << " a' * b:\t" << timer.value() << endl;
// // //
// BENCH(sm3.setprod(sm1.transpose(),sm2.transpose()); ) // BENCH(sm3._experimentalNewProduct(sm1.transpose(),sm2.transpose()); )
// std::cout << " a' * b':\t" << timer.value() << endl; // std::cout << " a' * b':\t" << timer.value() << endl;
// // //
// BENCH(sm3.setprod(sm1, sm2.transpose());) // BENCH(sm3._experimentalNewProduct(sm1, sm2.transpose());)
// std::cout << " a * b' :\t" << timer.value() << endl; // std::cout << " a * b' :\t" << timer.value() << endl;
} }
@ -247,6 +251,23 @@ int main(int argc, char *argv[])
} }
#endif #endif
#ifndef NOUBLAS
{
std::cout << "ublas\t" << nnzPerCol << "%\n";
UblasMatrix m1(rows,cols), m2(rows,cols), m3(rows,cols);
eiToUblas(sm1, m1);
eiToUblas(sm2, m2);
BENCH(boost::numeric::ublas::prod(m1, m2, m3););
// timer.reset();
// timer.start();
// for (int k=0; k<REPEAT; ++k)
// gmm::mult(m1, m2, gmmT3);
// timer.stop();
std::cout << " a * b:\t" << timer.value() << endl;
}
#endif
// GMM++ // GMM++
#ifndef NOGMM #ifndef NOGMM
{ {
@ -263,34 +284,34 @@ int main(int argc, char *argv[])
timer.stop(); timer.stop();
std::cout << " a * b:\t" << timer.value() << endl; std::cout << " a * b:\t" << timer.value() << endl;
timer.reset(); // timer.reset();
timer.start(); // timer.start();
for (int k=0; k<REPEAT; ++k) // for (int k=0; k<REPEAT; ++k)
gmm::mult(gmm::transposed(m1), m2, gmmT3); // gmm::mult(gmm::transposed(m1), m2, gmmT3);
timer.stop(); // timer.stop();
std::cout << " a' * b:\t" << timer.value() << endl; // std::cout << " a' * b:\t" << timer.value() << endl;
//
if (rows<500) // if (rows<500)
{ // {
timer.reset(); // timer.reset();
timer.start(); // timer.start();
for (int k=0; k<REPEAT; ++k) // for (int k=0; k<REPEAT; ++k)
gmm::mult(gmm::transposed(m1), gmm::transposed(m2), gmmT3); // gmm::mult(gmm::transposed(m1), gmm::transposed(m2), gmmT3);
timer.stop(); // timer.stop();
std::cout << " a' * b':\t" << timer.value() << endl; // std::cout << " a' * b':\t" << timer.value() << endl;
//
timer.reset(); // timer.reset();
timer.start(); // timer.start();
for (int k=0; k<REPEAT; ++k) // for (int k=0; k<REPEAT; ++k)
gmm::mult(m1, gmm::transposed(m2), gmmT3); // gmm::mult(m1, gmm::transposed(m2), gmmT3);
timer.stop(); // timer.stop();
std::cout << " a * b':\t" << timer.value() << endl; // std::cout << " a * b':\t" << timer.value() << endl;
} // }
else // else
{ // {
std::cout << " a' * b':\t" << "forever" << endl; // std::cout << " a' * b':\t" << "forever" << endl;
std::cout << " a * b':\t" << "forever" << endl; // std::cout << " a * b':\t" << "forever" << endl;
} // }
} }
#endif #endif
@ -309,26 +330,26 @@ int main(int argc, char *argv[])
timer.stop(); timer.stop();
std::cout << " a * b:\t" << timer.value() << endl; std::cout << " a * b:\t" << timer.value() << endl;
timer.reset(); // timer.reset();
timer.start(); // timer.start();
for (int k=0; k<REPEAT; ++k) // for (int k=0; k<REPEAT; ++k)
m3 = trans(m1) * m2; // m3 = trans(m1) * m2;
timer.stop(); // timer.stop();
std::cout << " a' * b:\t" << timer.value() << endl; // std::cout << " a' * b:\t" << timer.value() << endl;
//
timer.reset(); // timer.reset();
timer.start(); // timer.start();
for (int k=0; k<REPEAT; ++k) // for (int k=0; k<REPEAT; ++k)
m3 = trans(m1) * trans(m2); // m3 = trans(m1) * trans(m2);
timer.stop(); // timer.stop();
std::cout << " a' * b':\t" << timer.value() << endl; // std::cout << " a' * b':\t" << timer.value() << endl;
//
timer.reset(); // timer.reset();
timer.start(); // timer.start();
for (int k=0; k<REPEAT; ++k) // for (int k=0; k<REPEAT; ++k)
m3 = m1 * trans(m2); // m3 = m1 * trans(m2);
timer.stop(); // timer.stop();
std::cout << " a * b' :\t" << timer.value() << endl; // std::cout << " a * b' :\t" << timer.value() << endl;
} }
#endif #endif