3
0
mirror of https://github.com/triqs/dft_tools synced 2024-11-01 03:33:50 +01:00
dft_tools/test/speed/array2.cpp
Olivier Parcollet f2c7d449cc First commit : triqs libs version 1.0 alpha1
for earlier commits, see TRIQS0.x repository.
2013-07-17 19:24:07 +02:00

141 lines
3.4 KiB
C++

/*******************************************************************************
*
* TRIQS: a Toolbox for Research in Interacting Quantum Systems
*
* Copyright (C) 2011 by O. Parcollet
*
* TRIQS is free software: you can redistribute it and/or modify it under the
* terms of the GNU General Public License as published by the Free Software
* Foundation, either version 3 of the License, or (at your option) any later
* version.
*
* TRIQS is distributed in the hope that it will be useful, but WITHOUT ANY
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
* details.
*
* You should have received a copy of the GNU General Public License along with
* TRIQS. If not, see <http://www.gnu.org/licenses/>.
*
******************************************************************************/
#include <triqs/arrays/matrix.hpp>
#include <triqs/arrays/array.hpp>
using namespace std;
using namespace triqs::arrays;
typedef triqs::arrays::matrix<double > MM;
//typedef triqs::arrays::array<double,2 > MM;
struct bare_loop_with_access {
void operator()() {
MM A (20,30);//, FORTRAN_LAYOUT);
for (int i =0; i<20; ++i)
for (int j=0; j<30; ++j)
A(i,j) = 10*i+ j;
MM B(A), C(A), D(A);
for (int u =0; u<5000; ++u)
for (int i =0; i<20; ++i)
for (int j=0; j<30; ++j)
A(i,j) = B(i,j) + 3* C(i,j) + D(i,j);
}
};
struct expression_template {
void operator()() {
MM A (20,30);
for (int i =0; i<20; ++i)
for (int j=0; j<30; ++j)
A(i,j) = 10*i+ j;
MM B(A), C(A), D(A);
auto ex = B + 3* C+ D;
for (int u =0; u<5000; ++u)
A = B + 3*C + D;
}
};
struct bare_loop_with_access_big {
void operator()() {
MM A (200,300);//, FORTRAN_LAYOUT);
for (int i =0; i<200; ++i)
for (int j=0; j<300; ++j)
A(i,j) = 10*i+ j;
MM B(A), C(A), D(A);
//auto ex = A+B;
//typename decltype(ex)::domain_type D;
for (int u =0; u<500; ++u)
for (int i =0; i<200; ++i)
for (int j=0; j<300; ++j)
A(i,j) = B(i,j) + 3* C(i,j) + D(i,j);
}
};
struct expression_template_big {
void operator()() {
MM A (200,300);
for (int i =0; i<200; ++i)
for (int j=0; j<300; ++j)
A(i,j) = 10*i+ j;
MM B(A), C(A), D(A);
auto ex = B + 3* C+ D;
for (int u =0; u<500; ++u)
A = B + 3*C + D;
}
};
/*
struct pointer_and_expression_template {
MM A,B,C;
pointer_and_expression_template():A(20,30) {}
double f(size_t i, size_t j) const { return B(i,j) + 3*C(i,j);}
inline void F(double * restrict p, size_t sh,double const & rhs) { p[sh]=rhs;}
inline void F2(double & r,double const & rhs) { r=rhs;}
void operator()() {
for (int i =0; i<20; ++i)
for (int j=0; j<30; ++j)
A(i,j) = 10*i+ j;
B = A; C=A;
double * p = &A(0,0);
double * restrict pB = &B(0,0);
MM const & BB(B);
size_t s1=1, s2 = 20;
boost::tuple<int,int> t;
for (int u =0; u<5000; ++u)
for (int j=0; j<30; ++j) {
for (int i =0; i<20; ++i ) {
p[i*s1 + j*s2] = (B + 3* C)(i,j);
//F(p,i*s1 + j*s2, (B + 3* C)(i,j));
//F2(p[i*s1 + j*s2], (B + 3* C)(i,j));
}
}
}
};
*/
#include "./speed_tester.hpp"
int main() {
speed_tester<bare_loop_with_access> (5000);
//speed_tester<pointer_and_expression_template> (5000);
speed_tester<expression_template> (5000);
speed_tester<bare_loop_with_access_big> (500);
speed_tester<expression_template_big> (500);
return 0;
}