mirror of
https://github.com/paboyle/Grid.git
synced 2024-11-10 07:55:35 +00:00
46 lines
1.4 KiB
C
46 lines
1.4 KiB
C
|
#ifndef GRID_LATTICE_REDUCTION_H
|
||
|
|
#define GRID_LATTICE_REDUCTION_H
|
||
|
|
|
||
|
|
namespace Grid {
|
||
|
|
|
||
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
||
|
|
// Reduction operations
|
||
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
||
|
|
template<class vobj>
|
||
|
|
inline RealD norm2(const Lattice<vobj> &arg){
|
||
|
|
|
||
|
|
typedef typename vobj::scalar_type scalar;
|
||
|
|
typedef typename vobj::vector_type vector;
|
||
|
|
decltype(innerProduct(arg._odata[0],arg._odata[0])) vnrm=zero;
|
||
|
|
scalar nrm;
|
||
|
|
//FIXME make this loop parallelisable
|
||
|
|
vnrm=zero;
|
||
|
|
for(int ss=0;ss<arg._grid->oSites(); ss++){
|
||
|
|
vnrm = vnrm + innerProduct(arg._odata[ss],arg._odata[ss]);
|
||
|
|
}
|
||
|
|
vector vvnrm =TensorRemove(vnrm) ;
|
||
|
|
nrm = Reduce(vvnrm);
|
||
|
|
arg._grid->GlobalSum(nrm);
|
||
|
|
return real(nrm);
|
||
|
|
}
|
||
|
|
|
||
|
|
template<class vobj>
|
||
|
|
inline auto innerProduct(const Lattice<vobj> &left,const Lattice<vobj> &right) ->decltype(innerProduct(left._odata[0],right._odata[0]))
|
||
|
|
{
|
||
|
|
typedef typename vobj::scalar_type scalar;
|
||
|
|
decltype(innerProduct(left._odata[0],right._odata[0])) vnrm=zero;
|
||
|
|
|
||
|
|
scalar nrm;
|
||
|
|
//FIXME make this loop parallelisable
|
||
|
|
for(int ss=0;ss<left._grid->oSites(); ss++){
|
||
|
|
vnrm = vnrm + innerProduct(left._odata[ss],right._odata[ss]);
|
||
|
|
}
|
||
|
|
nrm = Reduce(vnrm);
|
||
|
|
right._grid->GlobalSum(nrm);
|
||
|
|
return nrm;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|