void DiracDomainWall::Dslash(ColorSpinorField &out, const ColorSpinorField &in,
const QudaParity parity) const
{
if ( in.Ndim() != 5 || out.Ndim() != 5) errorQuda("Wrong number of dimensions\n");
checkParitySpinor(in, out);
checkSpinorAlias(in, out);
if (checkLocation(out, in) == QUDA_CUDA_FIELD_LOCATION) {
domainWallDslashCuda(&static_cast<cudaColorSpinorField&>(out), *gauge,
&static_cast<const cudaColorSpinorField&>(in),
parity, dagger, 0, mass, 0, commDim, profile);
} else {
errorQuda("Not implemented");
}
long long Ls = in.X(4);
long long bulk = (Ls-2)*(in.Volume()/Ls);
long long wall = 2*in.Volume()/Ls;
flops += 1320LL*(long long)in.Volume() + 96LL*bulk + 120LL*wall;
}
void DiracStaggered::Dslash(ColorSpinorField &out, const ColorSpinorField &in,
const QudaParity parity) const
{
checkParitySpinor(in, out);
if (checkLocation(out, in) == QUDA_CUDA_FIELD_LOCATION) {
staggeredDslashCuda(&static_cast<cudaColorSpinorField&>(out),
*gauge, &static_cast<const cudaColorSpinorField&>(in), parity,
dagger, 0, 0, commDim, profile);
} else {
errorQuda("Not supported");
}
flops += 570ll*in.Volume();
}
void DiracStaggered::checkParitySpinor(const ColorSpinorField &in, const ColorSpinorField &out) const
{
if (in.Ndim() != 5 || out.Ndim() != 5) {
errorQuda("Staggered dslash requires 5-d fermion fields");
}
if (in.Precision() != out.Precision()) {
errorQuda("Input and output spinor precisions don't match in dslash_quda");
}
if (in.Stride() != out.Stride()) {
errorQuda("Input %d and output %d spinor strides don't match in dslash_quda", in.Stride(), out.Stride());
}
if (in.SiteSubset() != QUDA_PARITY_SITE_SUBSET || out.SiteSubset() != QUDA_PARITY_SITE_SUBSET) {
errorQuda("ColorSpinorFields are not single parity, in = %d, out = %d",
in.SiteSubset(), out.SiteSubset());
}
if ((out.Volume()/out.X(4) != 2*gauge->VolumeCB() && out.SiteSubset() == QUDA_FULL_SITE_SUBSET) ||
(out.Volume()/out.X(4) != gauge->VolumeCB() && out.SiteSubset() == QUDA_PARITY_SITE_SUBSET) ) {
errorQuda("Spinor volume %d doesn't match gauge volume %d", out.Volume(), gauge->VolumeCB());
}
}
