dslash_quda.h

Go to the documentation of this file.

#ifndef _DSLASH_QUDA_H
#define _DSLASH_QUDA_H

#include <quda_internal.h>
#include <tune_quda.h>
#include <face_quda.h>
#include <gauge_field.h>

namespace quda {

  void setDslashTuning(QudaTune tune, QudaVerbosity verbose);

  void setKernelPackT(bool pack);

  bool getKernelPackT();

#ifdef DSLASH_PROFILING
  void printDslashProfile();
#endif

  void setFace(const FaceBuffer &face);

  bool getDslashLaunch();

  void createDslashEvents();
  void destroyDslashEvents();

  void initLatticeConstants(const LatticeField &lat);
  void initGaugeConstants(const cudaGaugeField &gauge);
  void initSpinorConstants(const cudaColorSpinorField &spinor);
  void initDslashConstants();
  void initCloverConstants (const cudaCloverField &clover);
  void initStaggeredConstants(const cudaGaugeField &fatgauge, const cudaGaugeField &longgauge);
  void initTwistedMassConstants(const int flv_stride);

  // plain Wilson Dslash  
  void wilsonDslashCuda(cudaColorSpinorField *out, const cudaGaugeField &gauge, const cudaColorSpinorField *in,
                        const int oddBit, const int daggerBit, const cudaColorSpinorField *x,
                        const double &k, const int *commDim);

  // clover Dslash
  void cloverDslashCuda(cudaColorSpinorField *out, const cudaGaugeField &gauge, 
                        const FullClover cloverInv, const cudaColorSpinorField *in, 
                        const int oddBit, const int daggerBit, const cudaColorSpinorField *x,
                        const double &k, const int *commDim);

  // clover Dslash
  void asymCloverDslashCuda(cudaColorSpinorField *out, const cudaGaugeField &gauge, 
                            const FullClover cloverInv, const cudaColorSpinorField *in, 
                            const int oddBit, const int daggerBit, const cudaColorSpinorField *x,
                            const double &k, const int *commDim);

  // solo clover term
  void cloverCuda(cudaColorSpinorField *out, const cudaGaugeField &gauge, const FullClover clover, 
                  const cudaColorSpinorField *in, const int oddBit);

  // domain wall Dslash  
  void domainWallDslashCuda(cudaColorSpinorField *out, const cudaGaugeField &gauge, const cudaColorSpinorField *in, 
                            const int parity, const int dagger, const cudaColorSpinorField *x, 
                            const double &m_f, const double &k, const int *commDim);

  // staggered Dslash    
  void staggeredDslashCuda(cudaColorSpinorField *out, const cudaGaugeField &fatGauge, const cudaGaugeField &longGauge,
                           const cudaColorSpinorField *in, const int parity, const int dagger, 
                           const cudaColorSpinorField *x, const double &k, 
                           const int *commDim);

  // twisted mass Dslash  
  void twistedMassDslashCuda(cudaColorSpinorField *out, const cudaGaugeField &gauge, const   cudaColorSpinorField *in, 
                             const int parity, const int dagger, const cudaColorSpinorField *x, 
                             const double &kappa, const double &mu, const double &epsilon, const int *commDim);

  // solo twist term
  void twistGamma5Cuda(cudaColorSpinorField *out, const cudaColorSpinorField *in, const int dagger,     
                       const double &kappa, const double &mu, const double &epsilon, 
                       const QudaTwistGamma5Type);

  // face packing routines
  void packFace(void *ghost_buf, cudaColorSpinorField &in, const int dagger, const int parity, const cudaStream_t &stream);

}

#endif // _DSLASH_QUDA_H