su2hmc.h

Go to the documentation of this file.

#pragma once
//ARM Based machines. BLAS routines should work with other libraries, so we can set a compiler
//flag to sort them out. But the PRNG routines etc. are MKL exclusive
#include <integrate.h>
#ifdef __cplusplus
#include <cstdio>
#include <cstdlib>
#include <ctime>
#else
#include <time.h>
#endif
 
//Definitions:
//###########
#ifdef _DEBUGCG
#define _DEBUG 1
#endif
//Function Declarations:
//#####################
#if (defined __cplusplus)
extern "C"
{
#endif
   void Force_s(double *dSdpi, Complex_f *ut[2], Complex_f *X1, Complex_f *X2, Complex_f gamval[20],\
         unsigned int *iu, const unsigned short gamin[16],const float akappa, const unsigned short mu);
   void Force_t(double *dSdpi, Complex_f *ut[2],Complex_f *X1, Complex_f *X2, Complex_f gamval[20],\
         float *dk[2], unsigned int *iu, const unsigned short gamin[16],float akappa);
   int Force(double *dSdpi, const bool iflag, double res1, Complex *X0, Complex *X1, Complex *Phi,\
         Complex *ut[2], Complex_f *ut_f[2],unsigned int *iu,unsigned int *id,\
         Complex gamval[20],Complex_f gamval_f[20],const unsigned short gamin[16],Complex *sigval,Complex_f *sigval_f, unsigned short *sigin,\
         double *dk[2], float *dk_f[2],const Complex_f jqq, const float akappa,const float beta,const float c_sw,double *ancg);
   int Gauge_force(double *dSdpi, Complex_f *ut[2],unsigned int *iu,unsigned int *id, float beta);
   int Init(const int istart, const int ibound, const int iread, const float beta, const float fmu, const float akappa,\
         const Complex_f ajq, const float c_sw, Complex *u[2], Complex *ut[2], Complex_f *ut_f[2], Complex gamval[20],\
         Complex_f gamval_f[20], unsigned short gamin[16], double *dk[2], float *dk_f[2],\
         unsigned int *iu, unsigned int *id);   
   int Hamilton(double *h,double *s,double res2,double *pp,Complex *X0,Complex *X1,Complex *Phi, Complex *ud[2],Complex_f *ut[2],
         unsigned int *iu,unsigned int *id, Complex gamval[20], Complex_f gamval_f[20],const unsigned short gamin[16], Complex *sigval, Complex_f *sigval_f,
         unsigned short *sigin, double *dk[2],float *dk_f[2],Complex_f jqq,float akappa,float beta,float c_sw, double *ancgh,
         int traj);
   int Congradq(int na,double res,Complex *X1,Complex *r,Complex *ud[2], Complex_f *ut[2],Complex_f *clover_f[nc],
         unsigned int *iu, unsigned int *id, Complex gamval[20], Complex_f gamval_f[20],const unsigned short gamin[16],
         Complex *sigval, Complex_f *sigval_f,unsigned short *sigin, double *dk[2], float *dk_f[2],
         Complex_f jqq,float akappa,float c_sw,int *itercg);
   int Congradp(int na, double res, Complex *Phi, Complex *xi, Complex *ud[2], Complex_f *ut[2], Complex_f *clover_f[nc],
         unsigned int *iu, unsigned int *id, Complex gamval[20], Complex_f gamval_f[20], const unsigned short gamin[16],
         Complex *sigval, Complex_f *sigval_f,unsigned short *sigin, double *dk[2], float *dk_f[2],
         Complex_f jqq,float akappa,float c_sw,int *itercg);
   int Measure(double *pbp, double *endenf, double *denf, Complex *qq, Complex *qbqb, double res, int *itercg,\
         Complex *ut[2], Complex_f *ut_f[2], unsigned int *iu, unsigned int *id,\
         Complex gamval[20], Complex_f gamval_f[20],  const unsigned short gamin[16],\
         Complex *sigval,Complex_f *sigval_f, unsigned short *sigin, double *dk[2],float *dk_f[2],\
         Complex_f jqq, float akappa,  float c_sw,Complex *Phi);
   int Average_Plaquette(double *hg, double *avplaqs, double *avplaqt, Complex_f *ut[2],unsigned int *iu, float beta);
   int SU2plaq(Complex_f *ut[2], Complex_f Sigma[2], unsigned int *iu, int i, int mu, int nu);

   double Polyakov(Complex_f *ut[2]);
   //Inline functions
   int C_gather(Complex_f *x, Complex_f *y, int n, unsigned int *table, unsigned int mu);
   int Z_gather(Complex *x, Complex *y, int n, unsigned int *table, unsigned int mu);
   int Fill_Small_Phi(int na, Complex *smallPhi, Complex *Phi);
   int UpDownPart(const unsigned int na, Complex *X0, Complex *R1);
   int Reunitarise(Complex *ut[2]);
   int ComplexConvert(Complex_f *a, Complex *b, const unsigned int len, const bool dtof, const unsigned short stride);
 
#ifdef DIAGNOSTIC
   int Diagnostics(int istart, Complex *u[2], Complex *ut[2],Complex_f *ut_f[2],\
         unsigned int *iu, unsigned int *id, int *hu, int *hd, double *dk[2], float *dk_f[2],\
         const unsigned short gamin[16], const Complex gamval[20], const Complex_f gamval_f[20],\
         const Complex *sigval, const Complex_f *sigval_f, const unsigned short *sigin,
         Complex_f jqq,float akappa,float beta, float c_sw, double ancg);
#endif
   //CUDA Declarations:
   //#################
#ifdef USE_GPU
   extern cudaStream_t streams[ndirac*ndim*nadj];
   //Calling Functions:
   //=================
   void cuAverage_Plaquette(double *hgs, double *hgt, Complex_f *u11t, Complex_f *u12t, unsigned int *iu,dim3 dimGrid, dim3 dimBlock);
   void cuPolyakov(Complex_f *Sigma[2], Complex_f *ut[2],dim3 dimGrid, dim3 dimBlock);
   void cuGauge_force(Complex_f *ut[2],double *dSdpi,float beta,unsigned int *iu,unsigned int *id,dim3 dimGrid, dim3 dimBlock);
   void cuForce(double *dSdpi, Complex_f *ut[2], Complex_f *X1, Complex_f *X2, \
         Complex_f gamval[20],float *dk[2],unsigned int *iu,const unsigned short gamin[16],\
         float akappa, dim3 dimGrid, dim3 dimBlock);
   void Init_CUDA(Complex *u11t, Complex *u12t,Complex gamval[20], Complex_f gamval_f[20], unsigned short gamin[16], double*dk4m,\
         double *dk4p, unsigned int *iu, unsigned int *id);
   void cuFill_Small_Phi(const unsigned int na, Complex *smallPhi, Complex *Phi,dim3 dimBlock, dim3 dimGrid);
   void cuComplex_convert(Complex_f *a, Complex *b, const unsigned int len,  const bool dtof, dim3 dimBlock, dim3 dimGrid);
   void cuReal_convert(float *a, double *b, const unsigned int len, const bool dtof, dim3 dimBlock, dim3 dimGrid);
   void cuUpDownPart(const unsigned int na, Complex *X0, Complex *R1,dim3 dimBlock, dim3 dimGrid);
   void cuReunitarise(Complex *ut[2],dim3 dimGrid, dim3 dimBlock);
   void blockInit(int x, int y, int z, int t, dim3 *dimBlock, dim3 *dimGrid);
#endif
#if (defined __cplusplus)
}
#endif