portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-04-09 21:50:45 +01:00
Code Releases Activity

Namespace

Browse Source
This commit is contained in:
paboyle 2018-01-14 22:31:39 +00:00
parent 2c2da60cc2
commit 71ebd61327
1 changed files with 44 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

88
lib/qcd/hmc/HMC_GridModules.h
View File

@ -26,11 +26,11 @@ with this program; if not, write to the Free Software Foundation, Inc.,
See the full license in the file "LICENSE" in the top level distribution See the full license in the file "LICENSE" in the top level distribution
directory directory
*************************************************************************************/ *************************************************************************************/
/* END LEGAL */ /* END LEGAL */
#ifndef HMC_GRID_MODULES #ifndef HMC_GRID_MODULES
#define HMC_GRID_MODULES #define HMC_GRID_MODULES
namespace Grid { NAMESPACE_BEGIN(Grid);
// Resources // Resources
// Modules for grids // Modules for grids
@ -40,8 +40,8 @@ namespace Grid {
class GridModuleParameters: Serializable{ class GridModuleParameters: Serializable{
public: public:
GRID_SERIALIZABLE_CLASS_MEMBERS(GridModuleParameters, GRID_SERIALIZABLE_CLASS_MEMBERS(GridModuleParameters,
std::string, lattice, std::string, lattice,
std::string, mpi); std::string, mpi);
std::vector<int> getLattice() const {return strToVec<int>(lattice);} std::vector<int> getLattice() const {return strToVec<int>(lattice);}
std::vector<int> getMpi() const {return strToVec<int>(mpi);} std::vector<int> getMpi() const {return strToVec<int>(mpi);}
@ -51,7 +51,7 @@ public:
if (getLattice().size() != getMpi().size() ) { if (getLattice().size() != getMpi().size() ) {
std::cout << GridLogError std::cout << GridLogError
<< "Error in GridModuleParameters: lattice and mpi dimensions " << "Error in GridModuleParameters: lattice and mpi dimensions "
"do not match" "do not match"
<< std::endl; << std::endl;
exit(1); exit(1);
} }
@ -70,12 +70,12 @@ public:
write(Writer, name, *this); write(Writer, name, *this);
} }
private: private:
std::string name; std::string name;
}; };
// Lower level class // Lower level class
class GridModule { class GridModule {
public: public:
GridCartesian* get_full() { GridCartesian* get_full() {
std::cout << GridLogDebug << "Getting cartesian in module"<< std::endl; std::cout << GridLogDebug << "Getting cartesian in module"<< std::endl;
return grid_.get(); } return grid_.get(); }
@ -88,7 +88,7 @@ class GridModule {
void show_full_decomposition(){ grid_->show_decomposition(); } void show_full_decomposition(){ grid_->show_decomposition(); }
void show_rb_decomposition(){ rbgrid_->show_decomposition(); } void show_rb_decomposition(){ rbgrid_->show_decomposition(); }
protected: protected:
std::unique_ptr<GridCartesian> grid_; std::unique_ptr<GridCartesian> grid_;
std::unique_ptr<GridRedBlackCartesian> rbgrid_; std::unique_ptr<GridRedBlackCartesian> rbgrid_;
@ -106,9 +106,9 @@ public:
{ {
using namespace QCD; using namespace QCD;
set_full(SpaceTimeGrid::makeFourDimGrid( set_full(SpaceTimeGrid::makeFourDimGrid(
GridDefaultLatt(), GridDefaultLatt(),
GridDefaultSimd(4, vector_type::Nsimd()), GridDefaultSimd(4, vector_type::Nsimd()),
GridDefaultMpi())); GridDefaultMpi()));
set_rb(SpaceTimeGrid::makeFourDimRedBlackGrid(grid_.get())); set_rb(SpaceTimeGrid::makeFourDimRedBlackGrid(grid_.get()));
} }
@ -116,12 +116,12 @@ public:
{ {
using namespace QCD; using namespace QCD;
if (tweak_simd.size() != 4) if (tweak_simd.size() != 4)
{ {
std::cout << GridLogError std::cout << GridLogError
<< "Error in GridFourDimModule: SIMD size different from 4" << "Error in GridFourDimModule: SIMD size different from 4"
<< std::endl; << std::endl;
exit(1); exit(1);
} }
// Checks that the product agrees with the expectation // Checks that the product agrees with the expectation
int simd_sum = 1; int simd_sum = 1;
@ -130,20 +130,20 @@ public:
std::cout << GridLogDebug << "TweakSIMD: " << tweak_simd << " Sum: " << simd_sum << std::endl; std::cout << GridLogDebug << "TweakSIMD: " << tweak_simd << " Sum: " << simd_sum << std::endl;
if (simd_sum == vector_type::Nsimd()) if (simd_sum == vector_type::Nsimd())
{ {
set_full(SpaceTimeGrid::makeFourDimGrid( set_full(SpaceTimeGrid::makeFourDimGrid(
GridDefaultLatt(), GridDefaultLatt(),
tweak_simd, tweak_simd,
GridDefaultMpi())); GridDefaultMpi()));
set_rb(SpaceTimeGrid::makeFourDimRedBlackGrid(grid_.get())); set_rb(SpaceTimeGrid::makeFourDimRedBlackGrid(grid_.get()));
} }
else else
{ {
std::cout << GridLogError std::cout << GridLogError
<< "Error in GridFourDimModule: SIMD lanes must sum to " << "Error in GridFourDimModule: SIMD lanes must sum to "
<< vector_type::Nsimd() << vector_type::Nsimd()
<< std::endl; << std::endl;
} }
} }
GridFourDimModule(const GridModuleParameters Params) GridFourDimModule(const GridModuleParameters Params)
@ -152,26 +152,26 @@ public:
std::vector<int> lattice_v = Params.getLattice(); std::vector<int> lattice_v = Params.getLattice();
std::vector<int> mpi_v = Params.getMpi(); std::vector<int> mpi_v = Params.getMpi();
if (lattice_v.size() == 4) if (lattice_v.size() == 4)
{ {
set_full(SpaceTimeGrid::makeFourDimGrid( set_full(SpaceTimeGrid::makeFourDimGrid(
lattice_v, lattice_v,
GridDefaultSimd(4, vector_type::Nsimd()), GridDefaultSimd(4, vector_type::Nsimd()),
mpi_v)); mpi_v));
set_rb(SpaceTimeGrid::makeFourDimRedBlackGrid(grid_.get())); set_rb(SpaceTimeGrid::makeFourDimRedBlackGrid(grid_.get()));
} }
else else
{ {
std::cout << GridLogError std::cout << GridLogError
<< "Error in GridFourDimModule: lattice dimension different from 4" << "Error in GridFourDimModule: lattice dimension different from 4"
<< std::endl; << std::endl;
exit(1); exit(1);
} }
} }
}; };
typedef GridFourDimModule<vComplex> GridDefaultFourDimModule; typedef GridFourDimModule<vComplex> GridDefaultFourDimModule;
} // namespace Grid NAMESPACE_END(Grid);
#endif // HMC_GRID_MODULES #endif // HMC_GRID_MODULES