portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-06-13 04:37:05 +01:00
Code Releases Activity

Tweak to initialisation example

Browse Source
This commit is contained in:
Michael Marshall
2019-02-16 17:08:22 +00:00
parent b6803a070a
commit 00e9416e0a
2 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/hadrons/Test_hadrons_distil.cc
View File

@ -522,22 +522,28 @@ void EigenSliceExample2()
using T2 = Eigen::Tensor<TestScalar, 2, Options>;
T3 a(2,3,4);
std::cout << "Initialising:a";
std::cout << "Initialising a:";
for_all( a, [&](TestScalar &c, float f, const std::array<size_t,T3::NumIndices> &Dims ){
c = TestScalar{f,-f};
std::cout << " a(" << Dims[0] << "," << Dims[1] << "," << Dims[2] << ")=" << c;
} );
std::cout << std::endl;
//std::cout << "Validating a:";
float z = 0;
for( int i = 0 ; i < a.dimension(0) ; i++ )
for( int j = 0 ; j < a.dimension(1) ; j++ )
for( int k = 0 ; k < a.dimension(2) ; k++ ) {
TestScalar w{z, -z};
a(i,j,k) = w;
std::cout << " a(" << i << "," << j << "," << k << ")=" << w;
//std::cout << " a(" << i << "," << j << "," << k << ")=" << w;
assert( a(i,j,k) == w );
z++;
}
std::cout << std::endl;
//std::cout << std::endl;
//std::cout << "a initialised to:\n" << a << std::endl;
DumpMemoryOrder( a, "a" );
std::cout << "for_all(a):";
for_all( a, [&](TestScalar c, typename T3::Index n, const std::size_t * pDims ){
std::cout << " (" << pDims[0] << "," << pDims[1] << "," << pDims[2] << ")<" << n << ">=" << c;
for_all( a, [&](TestScalar c, typename T3::Index n, const std::array<size_t,T3::NumIndices> &Dims ){
std::cout << " (" << Dims[0] << "," << Dims[1] << "," << Dims[2] << ")<" << n << ">=" << c;
} );
std::cout << std::endl;
Eigen::array<typename T3::Index, 3> offsets = {0,1,1};