mirror of
				https://github.com/paboyle/Grid.git
				synced 2025-11-03 21:44:33 +00:00 
			
		
		
		
	Christoph needs SchurDiagTwoKappa which is mobius specific.
This commit is contained in:
		@@ -58,6 +58,7 @@ Author: Peter Boyle <pabobyle@ph.ed.ac.uk>
 | 
				
			|||||||
#include <Grid/qcd/action/fermion/DomainWallFermion.h>
 | 
					#include <Grid/qcd/action/fermion/DomainWallFermion.h>
 | 
				
			||||||
#include <Grid/qcd/action/fermion/MobiusFermion.h>
 | 
					#include <Grid/qcd/action/fermion/MobiusFermion.h>
 | 
				
			||||||
#include <Grid/qcd/action/fermion/ZMobiusFermion.h>
 | 
					#include <Grid/qcd/action/fermion/ZMobiusFermion.h>
 | 
				
			||||||
 | 
					#include <Grid/qcd/action/fermion/SchurDiagTwoKappa.h>
 | 
				
			||||||
#include <Grid/qcd/action/fermion/ScaledShamirFermion.h>
 | 
					#include <Grid/qcd/action/fermion/ScaledShamirFermion.h>
 | 
				
			||||||
#include <Grid/qcd/action/fermion/MobiusZolotarevFermion.h>
 | 
					#include <Grid/qcd/action/fermion/MobiusZolotarevFermion.h>
 | 
				
			||||||
#include <Grid/qcd/action/fermion/ShamirZolotarevFermion.h>
 | 
					#include <Grid/qcd/action/fermion/ShamirZolotarevFermion.h>
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										102
									
								
								lib/qcd/action/fermion/SchurDiagTwoKappa.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										102
									
								
								lib/qcd/action/fermion/SchurDiagTwoKappa.h
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,102 @@
 | 
				
			|||||||
 | 
					    /*************************************************************************************
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    Grid physics library, www.github.com/paboyle/Grid 
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    Source file: SchurDiagTwoKappa.h
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    Copyright (C) 2017
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					Author: Christoph Lehner
 | 
				
			||||||
 | 
					Author: Peter Boyle <paboyle@ph.ed.ac.uk>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    This program 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 2 of the License, or
 | 
				
			||||||
 | 
					    (at your option) any later version.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    This program 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 this program; if not, write to the Free Software Foundation, Inc.,
 | 
				
			||||||
 | 
					    51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    See the full license in the file "LICENSE" in the top level distribution directory
 | 
				
			||||||
 | 
					    *************************************************************************************/
 | 
				
			||||||
 | 
					    /*  END LEGAL */
 | 
				
			||||||
 | 
					#ifndef  _SCHUR_DIAG_TWO_KAPPA_H
 | 
				
			||||||
 | 
					#define  _SCHUR_DIAG_TWO_KAPPA_H
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					namespace Grid {
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  // This is specific to (Z)mobius fermions
 | 
				
			||||||
 | 
					  template<class Matrix, class Field>
 | 
				
			||||||
 | 
					    class KappaSimilarityTransform {
 | 
				
			||||||
 | 
					  public:
 | 
				
			||||||
 | 
					    INHERIT_IMPL_TYPES(Matrix);
 | 
				
			||||||
 | 
					    std::vector<Coeff_t> kappa, kappaDag, kappaInv, kappaInvDag;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    KappaSimilarityTransform (Matrix &zmob) {
 | 
				
			||||||
 | 
					      for (int i=0;i<(int)zmob.bs.size();i++) {
 | 
				
			||||||
 | 
						Coeff_t k = 1.0 / ( 2.0 * (zmob.bs[i] *(4 - zmob.M5) + 1.0) );
 | 
				
			||||||
 | 
						kappa.push_back( k );
 | 
				
			||||||
 | 
						kappaDag.push_back( conj(k) );
 | 
				
			||||||
 | 
						kappaInv.push_back( 1.0 / k );
 | 
				
			||||||
 | 
						kappaInvDag.push_back( 1.0 / conj(k) );
 | 
				
			||||||
 | 
					      }
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  template<typename vobj>
 | 
				
			||||||
 | 
					    void sscale(const Lattice<vobj>& in, Lattice<vobj>& out, Coeff_t* s) {
 | 
				
			||||||
 | 
					    GridBase *grid=out._grid;
 | 
				
			||||||
 | 
					    out.checkerboard = in.checkerboard;
 | 
				
			||||||
 | 
					    assert(grid->_simd_layout[0] == 1); // should be fine for ZMobius for now
 | 
				
			||||||
 | 
					    int Ls = grid->_rdimensions[0];
 | 
				
			||||||
 | 
					    parallel_for(int ss=0;ss<grid->oSites();ss++){
 | 
				
			||||||
 | 
					      vobj tmp = s[ss % Ls]*in._odata[ss];
 | 
				
			||||||
 | 
					      vstream(out._odata[ss],tmp);
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  RealD sscale_norm(const Field& in, Field& out, Coeff_t* s) {
 | 
				
			||||||
 | 
					    sscale(in,out,s);
 | 
				
			||||||
 | 
					    return norm2(out);
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  virtual RealD M       (const Field& in, Field& out) { return sscale_norm(in,out,&kappa[0]);   }
 | 
				
			||||||
 | 
					  virtual RealD MDag    (const Field& in, Field& out) { return sscale_norm(in,out,&kappaDag[0]);}
 | 
				
			||||||
 | 
					  virtual RealD MInv    (const Field& in, Field& out) { return sscale_norm(in,out,&kappaInv[0]);}
 | 
				
			||||||
 | 
					  virtual RealD MInvDag (const Field& in, Field& out) { return sscale_norm(in,out,&kappaInvDag[0]);}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  template<class Matrix,class Field>
 | 
				
			||||||
 | 
					    class SchurDiagTwoKappaOperator :  public SchurOperatorBase<Field> {
 | 
				
			||||||
 | 
					  public:
 | 
				
			||||||
 | 
					    KappaSimilarityTransform<Matrix, Field> _S;
 | 
				
			||||||
 | 
					    SchurDiagTwoOperator<Matrix, Field> _Mat;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    SchurDiagTwoKappaOperator (Matrix &Mat): _S(Mat), _Mat(Mat) {};
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    virtual  RealD Mpc      (const Field &in, Field &out) {
 | 
				
			||||||
 | 
					      Field tmp(in._grid);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					      _S.MInv(in,out);
 | 
				
			||||||
 | 
					      _Mat.Mpc(out,tmp);
 | 
				
			||||||
 | 
					      return _S.M(tmp,out);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					    virtual  RealD MpcDag   (const Field &in, Field &out){
 | 
				
			||||||
 | 
					      Field tmp(in._grid);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					      _S.MDag(in,out);
 | 
				
			||||||
 | 
					      _Mat.MpcDag(out,tmp);
 | 
				
			||||||
 | 
					      return _S.MInvDag(tmp,out);
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					#endif
 | 
				
			||||||
		Reference in New Issue
	
	Block a user