diff options
author | Narayan Kamath <narayan@google.com> | 2012-11-02 10:59:05 +0000 |
---|---|---|
committer | Xiaotao Duan <xiaotao@google.com> | 2012-11-07 14:17:48 -0800 |
commit | c981c48f5bc9aefeffc0bcb0cc3934c2fae179dd (patch) | |
tree | 54d1c7d66098154c1d7c5bd414394ef4cf255810 /bench/benchEigenSolver.cpp | |
parent | 63f67d748682b46d58be31235a0a2d64d81b998c (diff) | |
download | eigen-c981c48f5bc9aefeffc0bcb0cc3934c2fae179dd.tar.gz |
Initial import of eigen 3.1.1android-cts-4.4_r4android-cts-4.2_r2android-4.4_r1.2.0.1android-4.4_r1.2android-4.4_r1.1.0.1android-4.4_r1.1android-4.4_r1.0.1android-4.4_r1android-4.4_r0.9android-4.4_r0.8android-4.3_r2.3android-4.3_r2.2android-4.3_r2.1android-4.3_r2android-4.3_r1.1android-4.3_r1android-4.3_r0.9.1android-4.3_r0.9android-4.2.2_r1.2android-4.2.2_r1.1android-4.2.2_r1kitkat-releasekitkat-cts-releasejb-mr2.0-releasejb-mr2-releasejb-mr1.1-releasejb-mr1.1-dev
Added a README.android and a MODULE_LICENSE_MPL2 file.
Added empty Android.mk and CleanSpec.mk to optimize Android build.
Non MPL2 license code is disabled in ./Eigen/src/Core/util/NonMPL2.h.
Trying to include such files will lead to an error.
Change-Id: I0e148b7c3e83999bcc4dfaa5809d33bfac2aac32
Diffstat (limited to 'bench/benchEigenSolver.cpp')
-rw-r--r-- | bench/benchEigenSolver.cpp | 212 |
1 files changed, 212 insertions, 0 deletions
diff --git a/bench/benchEigenSolver.cpp b/bench/benchEigenSolver.cpp new file mode 100644 index 000000000..dd78c7e01 --- /dev/null +++ b/bench/benchEigenSolver.cpp @@ -0,0 +1,212 @@ + +// g++ -DNDEBUG -O3 -I.. benchEigenSolver.cpp -o benchEigenSolver && ./benchEigenSolver +// options: +// -DBENCH_GMM +// -DBENCH_GSL -lgsl /usr/lib/libcblas.so.3 +// -DEIGEN_DONT_VECTORIZE +// -msse2 +// -DREPEAT=100 +// -DTRIES=10 +// -DSCALAR=double + +#include <iostream> + +#include <Eigen/Core> +#include <Eigen/QR> +#include <bench/BenchUtil.h> +using namespace Eigen; + +#ifndef REPEAT +#define REPEAT 1000 +#endif + +#ifndef TRIES +#define TRIES 4 +#endif + +#ifndef SCALAR +#define SCALAR float +#endif + +typedef SCALAR Scalar; + +template <typename MatrixType> +__attribute__ ((noinline)) void benchEigenSolver(const MatrixType& m) +{ + int rows = m.rows(); + int cols = m.cols(); + + int stdRepeats = std::max(1,int((REPEAT*1000)/(rows*rows*sqrt(rows)))); + int saRepeats = stdRepeats * 4; + + typedef typename MatrixType::Scalar Scalar; + typedef Matrix<Scalar, MatrixType::RowsAtCompileTime, MatrixType::RowsAtCompileTime> SquareMatrixType; + + MatrixType a = MatrixType::Random(rows,cols); + SquareMatrixType covMat = a * a.adjoint(); + + BenchTimer timerSa, timerStd; + + Scalar acc = 0; + int r = internal::random<int>(0,covMat.rows()-1); + int c = internal::random<int>(0,covMat.cols()-1); + { + SelfAdjointEigenSolver<SquareMatrixType> ei(covMat); + for (int t=0; t<TRIES; ++t) + { + timerSa.start(); + for (int k=0; k<saRepeats; ++k) + { + ei.compute(covMat); + acc += ei.eigenvectors().coeff(r,c); + } + timerSa.stop(); + } + } + + { + EigenSolver<SquareMatrixType> ei(covMat); + for (int t=0; t<TRIES; ++t) + { + timerStd.start(); + for (int k=0; k<stdRepeats; ++k) + { + ei.compute(covMat); + acc += ei.eigenvectors().coeff(r,c); + } + timerStd.stop(); + } + } + + if (MatrixType::RowsAtCompileTime==Dynamic) + std::cout << "dyn "; + else + std::cout << "fixed "; + std::cout << covMat.rows() << " \t" + << timerSa.value() * REPEAT / saRepeats << "s \t" + << timerStd.value() * REPEAT / stdRepeats << "s"; + + #ifdef BENCH_GMM + if (MatrixType::RowsAtCompileTime==Dynamic) + { + timerSa.reset(); + timerStd.reset(); + + gmm::dense_matrix<Scalar> gmmCovMat(covMat.rows(),covMat.cols()); + gmm::dense_matrix<Scalar> eigvect(covMat.rows(),covMat.cols()); + std::vector<Scalar> eigval(covMat.rows()); + eiToGmm(covMat, gmmCovMat); + for (int t=0; t<TRIES; ++t) + { + timerSa.start(); + for (int k=0; k<saRepeats; ++k) + { + gmm::symmetric_qr_algorithm(gmmCovMat, eigval, eigvect); + acc += eigvect(r,c); + } + timerSa.stop(); + } + // the non-selfadjoint solver does not compute the eigen vectors +// for (int t=0; t<TRIES; ++t) +// { +// timerStd.start(); +// for (int k=0; k<stdRepeats; ++k) +// { +// gmm::implicit_qr_algorithm(gmmCovMat, eigval, eigvect); +// acc += eigvect(r,c); +// } +// timerStd.stop(); +// } + + std::cout << " | \t" + << timerSa.value() * REPEAT / saRepeats << "s" + << /*timerStd.value() * REPEAT / stdRepeats << "s"*/ " na "; + } + #endif + + #ifdef BENCH_GSL + if (MatrixType::RowsAtCompileTime==Dynamic) + { + timerSa.reset(); + timerStd.reset(); + + gsl_matrix* gslCovMat = gsl_matrix_alloc(covMat.rows(),covMat.cols()); + gsl_matrix* gslCopy = gsl_matrix_alloc(covMat.rows(),covMat.cols()); + gsl_matrix* eigvect = gsl_matrix_alloc(covMat.rows(),covMat.cols()); + gsl_vector* eigval = gsl_vector_alloc(covMat.rows()); + gsl_eigen_symmv_workspace* eisymm = gsl_eigen_symmv_alloc(covMat.rows()); + + gsl_matrix_complex* eigvectz = gsl_matrix_complex_alloc(covMat.rows(),covMat.cols()); + gsl_vector_complex* eigvalz = gsl_vector_complex_alloc(covMat.rows()); + gsl_eigen_nonsymmv_workspace* einonsymm = gsl_eigen_nonsymmv_alloc(covMat.rows()); + + eiToGsl(covMat, &gslCovMat); + for (int t=0; t<TRIES; ++t) + { + timerSa.start(); + for (int k=0; k<saRepeats; ++k) + { + gsl_matrix_memcpy(gslCopy,gslCovMat); + gsl_eigen_symmv(gslCopy, eigval, eigvect, eisymm); + acc += gsl_matrix_get(eigvect,r,c); + } + timerSa.stop(); + } + for (int t=0; t<TRIES; ++t) + { + timerStd.start(); + for (int k=0; k<stdRepeats; ++k) + { + gsl_matrix_memcpy(gslCopy,gslCovMat); + gsl_eigen_nonsymmv(gslCopy, eigvalz, eigvectz, einonsymm); + acc += GSL_REAL(gsl_matrix_complex_get(eigvectz,r,c)); + } + timerStd.stop(); + } + + std::cout << " | \t" + << timerSa.value() * REPEAT / saRepeats << "s \t" + << timerStd.value() * REPEAT / stdRepeats << "s"; + + gsl_matrix_free(gslCovMat); + gsl_vector_free(gslCopy); + gsl_matrix_free(eigvect); + gsl_vector_free(eigval); + gsl_matrix_complex_free(eigvectz); + gsl_vector_complex_free(eigvalz); + gsl_eigen_symmv_free(eisymm); + gsl_eigen_nonsymmv_free(einonsymm); + } + #endif + + std::cout << "\n"; + + // make sure the compiler does not optimize too much + if (acc==123) + std::cout << acc; +} + +int main(int argc, char* argv[]) +{ + const int dynsizes[] = {4,6,8,12,16,24,32,64,128,256,512,0}; + std::cout << "size selfadjoint generic"; + #ifdef BENCH_GMM + std::cout << " GMM++ "; + #endif + #ifdef BENCH_GSL + std::cout << " GSL (double + ATLAS) "; + #endif + std::cout << "\n"; + for (uint i=0; dynsizes[i]>0; ++i) + benchEigenSolver(Matrix<Scalar,Dynamic,Dynamic>(dynsizes[i],dynsizes[i])); + + benchEigenSolver(Matrix<Scalar,2,2>()); + benchEigenSolver(Matrix<Scalar,3,3>()); + benchEigenSolver(Matrix<Scalar,4,4>()); + benchEigenSolver(Matrix<Scalar,6,6>()); + benchEigenSolver(Matrix<Scalar,8,8>()); + benchEigenSolver(Matrix<Scalar,12,12>()); + benchEigenSolver(Matrix<Scalar,16,16>()); + return 0; +} + |