various work on the Sparse module:

* added some glue to Eigen/Core (SparseBit, ei_eval, Matrix)
* add two new sparse matrix types:
   HashMatrix: based on std::map (for random writes)
   LinkedVectorMatrix: array of linked vectors
   (for outer coherent writes, e.g. to transpose a matrix)
* add a SparseSetter class to easily set/update any kind of matrices, e.g.:
   { SparseSetter<MatrixType,RandomAccessPattern> wrapper(mymatrix);
     for (...) wrapper->coeffRef(rand(),rand()) = rand(); }
* automatic shallow copy for RValue
* and a lot of mess !
plus:
* remove the remaining ArrayBit related stuff
* don't use alloca in product for very large memory allocation
This commit is contained in:
Gael Guennebaud
2008-06-26 23:22:26 +00:00
parent c5bd1703cb
commit e5d301dc96
17 changed files with 1226 additions and 169 deletions

View File

@@ -12,15 +12,23 @@ using namespace Eigen;
USING_PART_OF_NAMESPACE_EIGEN
#ifndef REPEAT
#define REPEAT 40000000
#define REPEAT 10
#endif
#define REPEATPRODUCT 1
#define SIZE 10
#define DENSITY 0.2
// #define NODENSEMATRIX
typedef MatrixXf DenseMatrix;
// typedef Matrix<float,SIZE,SIZE> DenseMatrix;
typedef SparseMatrix<float> EigenSparseMatrix;
typedef gmm::csc_matrix<float> GmmSparse;
typedef gmm::col_matrix< gmm::wsvector<float> > GmmDynSparse;
void fillMatrix(float density, int rows, int cols, MatrixXf* pDenseMatrix, EigenSparseMatrix* pSparseMatrix, GmmSparse* pGmmMatrix=0)
void fillMatrix(float density, int rows, int cols, DenseMatrix* pDenseMatrix, EigenSparseMatrix* pSparseMatrix, GmmSparse* pGmmMatrix=0)
{
GmmDynSparse gmmT(rows, cols);
if (pSparseMatrix)
@@ -49,52 +57,155 @@ void fillMatrix(float density, int rows, int cols, MatrixXf* pDenseMatrix, Eigen
int main(int argc, char *argv[])
{
int rows = 4000;
int cols = 4000;
float density = 0.1;
int rows = SIZE;
int cols = SIZE;
float density = DENSITY;
// dense matrices
#ifndef NODENSEMATRIX
DenseMatrix m1(rows,cols), m2(rows,cols), m3(rows,cols), m4(rows,cols);
#endif
// sparse matrices
EigenSparseMatrix sm1(rows,cols), sm2(rows,cols), sm3(rows,cols), sm4(rows,cols);
HashMatrix<float> hm4(rows,cols);
// GMM++ matrices
GmmDynSparse gmmT4(rows,cols);
GmmSparse gmmM1(rows,cols), gmmM2(rows,cols), gmmM3(rows,cols), gmmM4(rows,cols);
#ifndef NODENSEMATRIX
fillMatrix(density, rows, cols, &m1, &sm1, &gmmM1);
fillMatrix(density, rows, cols, &m2, &sm2, &gmmM2);
fillMatrix(density, rows, cols, &m3, &sm3, &gmmM3);
#else
fillMatrix(density, rows, cols, 0, &sm1, &gmmM1);
fillMatrix(density, rows, cols, 0, &sm2, &gmmM2);
fillMatrix(density, rows, cols, 0, &sm3, &gmmM3);
#endif
BenchTimer timer;
//--------------------------------------------------------------------------------
// COEFF WISE OPERATORS
//--------------------------------------------------------------------------------
#if 1
std::cout << "\n\n\"m4 = m1 + m2 + 2 * m3\":\n\n";
timer.reset();
timer.start();
for (int k=0; k<10; ++k)
asm("#begin");
for (int k=0; k<REPEAT; ++k)
m4 = m1 + m2 + 2 * m3;
asm("#end");
timer.stop();
std::cout << "Eigen dense = " << timer.value() << endl;
timer.reset();
timer.start();
for (int k=0; k<10; ++k)
for (int k=0; k<REPEAT; ++k)
sm4 = sm1 + sm2 + 2 * sm3;
timer.stop();
std::cout << "Eigen sparse = " << timer.value() << endl;
timer.reset();
timer.start();
for (int k=0; k<10; ++k)
for (int k=0; k<REPEAT; ++k)
hm4 = sm1 + sm2 + 2 * sm3;
timer.stop();
std::cout << "Eigen hash = " << timer.value() << endl;
LinkedVectorMatrix<float> lm4(rows, cols);
timer.reset();
timer.start();
for (int k=0; k<REPEAT; ++k)
lm4 = sm1 + sm2 + 2 * sm3;
timer.stop();
std::cout << "Eigen linked vector = " << timer.value() << endl;
timer.reset();
timer.start();
for (int k=0; k<REPEAT; ++k)
{
gmm::add(gmmM1, gmmM2, gmmT4);
gmm::add(gmm::scaled(gmmM3,2), gmmT4);
}
timer.stop();
std::cout << "GMM++ sparse = " << timer.value() << endl;
#endif
//--------------------------------------------------------------------------------
// PRODUCT
//--------------------------------------------------------------------------------
#if 0
std::cout << "\n\nProduct:\n\n";
#ifndef NODENSEMATRIX
timer.reset();
timer.start();
asm("#begin");
for (int k=0; k<REPEATPRODUCT; ++k)
m1 = m1 * m2;
asm("#end");
timer.stop();
std::cout << "Eigen dense = " << timer.value() << endl;
#endif
timer.reset();
timer.start();
for (int k=0; k<REPEATPRODUCT; ++k)
sm4 = sm1 * sm2;
timer.stop();
std::cout << "Eigen sparse = " << timer.value() << endl;
// timer.reset();
// timer.start();
// for (int k=0; k<REPEATPRODUCT; ++k)
// hm4 = sm1 * sm2;
// timer.stop();
// std::cout << "Eigen hash = " << timer.value() << endl;
timer.reset();
timer.start();
for (int k=0; k<REPEATPRODUCT; ++k)
{
gmm::csr_matrix<float> R(rows,cols);
gmm::copy(gmmM1, R);
//gmm::mult(gmmM1, gmmM2, gmmT4);
}
timer.stop();
std::cout << "GMM++ sparse = " << timer.value() << endl;
#endif
//--------------------------------------------------------------------------------
// VARIOUS
//--------------------------------------------------------------------------------
#if 1
// sm3 = sm1 + m2;
// cout << m4.transpose() << "\n\n" << sm4 << endl;
cout << m4.transpose() << "\n\n";
// sm4 = sm1+sm2;
cout << sm4 << "\n\n";
cout << lm4 << endl;
LinkedVectorMatrix<float,RowMajorBit> lm5(rows, cols);
lm5 = lm4;
lm5 = sm4;
cout << endl << lm5 << endl;
sm3 = sm4.transpose();
cout << endl << lm5 << endl;
cout << endl << "SM1 before random editing: " << endl << sm1 << endl;
{
SparseSetter<EigenSparseMatrix,RandomAccessPattern> w1(sm1);
w1->coeffRef(4,2) = ei_random<float>();
w1->coeffRef(2,6) = ei_random<float>();
w1->coeffRef(0,4) = ei_random<float>();
w1->coeffRef(9,3) = ei_random<float>();
}
cout << endl << "SM1 after random editing: " << endl << sm1 << endl;
#endif
return 0;
}