deg_tm_dslash_cuda_gen Namespace Reference

Functions
def	complexify (a)
	complex numbers ######################################################################## More...
def	complexToStr (c)
def	gplus (g1, g2)
def	gminus (g1, g2)
def	projectorToStr (p)
def	indent (code)
	code generation ######################################################################## More...
def	block (code)
def	sign (x)
def	nthFloat4 (n)
def	nthFloat2 (n)
def	in_re (s, c)
def	in_im (s, c)
def	g_re (d, m, n)
def	g_im (d, m, n)
def	out_re (s, c)
def	out_im (s, c)
def	h1_re (h, c)
def	h1_im (h, c)
def	h2_re (h, c)
def	h2_im (h, c)
def	a_re (b, s, c)
def	a_im (b, s, c)
def	acc_re (s, c)
def	acc_im (s, c)
def	tmp_re (s, c)
def	tmp_im (s, c)
def	spinor (name, s, c, z)
def	def_input_spinor ()
def	def_gauge ()
def	def_output_spinor ()
def	prolog ()
def	gen (dir, pack_only=False)
def	input_spinor (s, c, z)
def	twisted_xpay ()
def	epilog ()
def	pack_face (facenum)
def	generate_pack ()
def	generate_dslash ()
def	generate_dslash_kernels (arch)

Variables
def	id
	projector matrices ######################################################################## More...
def	gamma1
def	gamma2
def	gamma3
def	gamma4
def	igamma5
list	projectors
bool	dslash = False
bool	dagger = False
bool	twist = False
int	sharedFloats = 0
bool	sharedDslash = False
bool	pack = False
int	arch = 200
	f = open('dslash_core/wilson_pack_twisted_face_core.h', 'w')

Function Documentation

a_im()

def deg_tm_dslash_cuda_gen.a_im	(		b,
			s,
			c
	)

Definition at line 128 of file deg_tm_dslash_cuda_gen.py.

Referenced by dot().

Here is the caller graph for this function:

a_re()

def deg_tm_dslash_cuda_gen.a_re	(		b,
			s,
			c
	)

Definition at line 127 of file deg_tm_dslash_cuda_gen.py.

Referenced by dot().

Here is the caller graph for this function:

acc_im()

def deg_tm_dslash_cuda_gen.acc_im	(		s,
			c
	)

Definition at line 131 of file deg_tm_dslash_cuda_gen.py.

Referenced by def_input_spinor(), epilog(), and twisted_xpay().

Here is the caller graph for this function:

acc_re()

def deg_tm_dslash_cuda_gen.acc_re	(		s,
			c
	)

Definition at line 130 of file deg_tm_dslash_cuda_gen.py.

Referenced by def_input_spinor(), epilog(), and twisted_xpay().

Here is the caller graph for this function:

block()

def deg_tm_dslash_cuda_gen.block

(

code

)

Definition at line 101 of file deg_tm_dslash_cuda_gen.py.

References indent().

Referenced by quda::TunableVectorY.advanceBlockDim(), quda::TunableVectorYZ.advanceBlockDim(), quda.axpy(), coordsFromIndex(), epilog(), quda.GaugeFixHit_AtomicAdd(), quda.GaugeFixHit_NoAtomicAdd(), quda.GaugeFixHit_NoAtomicAdd_LessSM(), gen(), generate_pack(), quda.IndexBlock(), quda.operator+=(), quda.operator-=(), and pack_face().

Here is the call graph for this function:

Here is the caller graph for this function:

complexify()

def deg_tm_dslash_cuda_gen.complexify

(

a

)

complex numbers ########################################################################

Definition at line 5 of file deg_tm_dslash_cuda_gen.py.

complexToStr()

def deg_tm_dslash_cuda_gen.complexToStr

(

c

)

Definition at line 8 of file deg_tm_dslash_cuda_gen.py.

References int.

Referenced by projectorToStr().

Here is the caller graph for this function:

def_gauge()

def deg_tm_dslash_cuda_gen.def_gauge

(

)

Definition at line 181 of file deg_tm_dslash_cuda_gen.py.

References g_im(), g_re(), nthFloat2(), and nthFloat4().

Referenced by prolog().

Here is the call graph for this function:

Here is the caller graph for this function:

def_input_spinor()

def deg_tm_dslash_cuda_gen.def_input_spinor

(

)

Definition at line 140 of file deg_tm_dslash_cuda_gen.py.

References acc_im(), acc_re(), in_im(), in_re(), nthFloat2(), and nthFloat4().

Referenced by generate_pack(), and prolog().

Here is the call graph for this function:

Here is the caller graph for this function:

def_output_spinor()

def deg_tm_dslash_cuda_gen.def_output_spinor

(

)

Definition at line 213 of file deg_tm_dslash_cuda_gen.py.

References out_im(), and out_re().

Referenced by prolog().

Here is the call graph for this function:

Here is the caller graph for this function:

epilog()

def deg_tm_dslash_cuda_gen.epilog

(

)

Definition at line 786 of file deg_tm_dslash_cuda_gen.py.

References acc_im(), acc_re(), block(), g_im(), g_re(), in_im(), in_re(), out_im(), out_re(), and twisted_xpay().

Referenced by generate_dslash().

Here is the call graph for this function:

Here is the caller graph for this function:

g_im()

def deg_tm_dslash_cuda_gen.g_im	(		d,
			m,
			n
	)

Definition at line 120 of file deg_tm_dslash_cuda_gen.py.

Referenced by def_gauge(), epilog(), and gen().

Here is the caller graph for this function:

g_re()

def deg_tm_dslash_cuda_gen.g_re	(		d,
			m,
			n
	)

Definition at line 119 of file deg_tm_dslash_cuda_gen.py.

Referenced by def_gauge(), epilog(), and gen().

Here is the caller graph for this function:

gen()

def deg_tm_dslash_cuda_gen.gen	(		dir,
			pack_only = False
	)

Definition at line 375 of file deg_tm_dslash_cuda_gen.py.

References block(), g_im(), g_re(), h1_im(), h1_re(), h2_im(), h2_re(), in_im(), in_re(), indent(), out_im(), out_re(), projectorToStr(), and sign().

Here is the call graph for this function:

generate_dslash()

def deg_tm_dslash_cuda_gen.generate_dslash

(

)

Definition at line 908 of file deg_tm_dslash_cuda_gen.py.

References epilog(), and prolog().

Referenced by generate_dslash_kernels().

Here is the call graph for this function:

Here is the caller graph for this function:

generate_dslash_kernels()

def deg_tm_dslash_cuda_gen.generate_dslash_kernels

(

arch

)

Definition at line 912 of file deg_tm_dslash_cuda_gen.py.

References generate_dslash().

Here is the call graph for this function:

generate_pack()

def deg_tm_dslash_cuda_gen.generate_pack

(

)

Definition at line 881 of file deg_tm_dslash_cuda_gen.py.

References block(), def_input_spinor(), in_im(), in_re(), and pack_face().

Here is the call graph for this function:

gminus()

def deg_tm_dslash_cuda_gen.gminus	(		g1,
			g2
	)

Definition at line 77 of file deg_tm_dslash_cuda_gen.py.

gplus()

def deg_tm_dslash_cuda_gen.gplus	(		g1,
			g2
	)

Definition at line 74 of file deg_tm_dslash_cuda_gen.py.

h1_im()

def deg_tm_dslash_cuda_gen.h1_im	(		h,
			c
	)

Definition at line 124 of file deg_tm_dslash_cuda_gen.py.

Referenced by gen().

Here is the caller graph for this function:

h1_re()

def deg_tm_dslash_cuda_gen.h1_re	(		h,
			c
	)

Definition at line 123 of file deg_tm_dslash_cuda_gen.py.

Referenced by gen().

Here is the caller graph for this function:

h2_im()

def deg_tm_dslash_cuda_gen.h2_im	(		h,
			c
	)

Definition at line 126 of file deg_tm_dslash_cuda_gen.py.

Referenced by gen().

Here is the caller graph for this function:

h2_re()

def deg_tm_dslash_cuda_gen.h2_re	(		h,
			c
	)

Definition at line 125 of file deg_tm_dslash_cuda_gen.py.

Referenced by gen().

Here is the caller graph for this function:

in_im()

def deg_tm_dslash_cuda_gen.in_im	(		s,
			c
	)

Definition at line 118 of file deg_tm_dslash_cuda_gen.py.

Referenced by def_input_spinor(), epilog(), gen(), generate_pack(), input_spinor(), and prolog().

Here is the caller graph for this function:

in_re()

def deg_tm_dslash_cuda_gen.in_re	(		s,
			c
	)

Definition at line 117 of file deg_tm_dslash_cuda_gen.py.

Referenced by def_input_spinor(), epilog(), gen(), generate_pack(), input_spinor(), and prolog().

Here is the caller graph for this function:

indent()

def deg_tm_dslash_cuda_gen.indent

(

code

)

code generation ########################################################################

Definition at line 97 of file deg_tm_dslash_cuda_gen.py.

Referenced by block(), gen(), pack_face(), and prolog().

Here is the caller graph for this function:

input_spinor()

def deg_tm_dslash_cuda_gen.input_spinor	(		s,
			c,
			z
	)

Definition at line 726 of file deg_tm_dslash_cuda_gen.py.

References in_im(), in_re(), out_im(), and out_re().

Here is the call graph for this function:

nthFloat2()

def deg_tm_dslash_cuda_gen.nthFloat2

(

n

)

Definition at line 113 of file deg_tm_dslash_cuda_gen.py.

Referenced by def_gauge(), and def_input_spinor().

Here is the caller graph for this function:

nthFloat4()

def deg_tm_dslash_cuda_gen.nthFloat4

(

n

)

Definition at line 110 of file deg_tm_dslash_cuda_gen.py.

Referenced by def_gauge(), and def_input_spinor().

Here is the caller graph for this function:

out_im()

def deg_tm_dslash_cuda_gen.out_im	(		s,
			c
	)

Definition at line 122 of file deg_tm_dslash_cuda_gen.py.

Referenced by def_output_spinor(), epilog(), gen(), input_spinor(), prolog(), and twisted_xpay().

Here is the caller graph for this function:

out_re()

def deg_tm_dslash_cuda_gen.out_re	(		s,
			c
	)

Definition at line 121 of file deg_tm_dslash_cuda_gen.py.

Referenced by def_output_spinor(), epilog(), gen(), input_spinor(), prolog(), and twisted_xpay().

Here is the caller graph for this function:

pack_face()

def deg_tm_dslash_cuda_gen.pack_face

(

facenum

)

Definition at line 867 of file deg_tm_dslash_cuda_gen.py.

References block(), and indent().

Referenced by generate_pack().

Here is the call graph for this function:

Here is the caller graph for this function:

projectorToStr()

def deg_tm_dslash_cuda_gen.projectorToStr

(

p

)

Definition at line 80 of file deg_tm_dslash_cuda_gen.py.

References complexToStr().

Referenced by gen().

Here is the call graph for this function:

Here is the caller graph for this function:

prolog()

def deg_tm_dslash_cuda_gen.prolog

(

)

Definition at line 232 of file deg_tm_dslash_cuda_gen.py.

References def_gauge(), def_input_spinor(), def_output_spinor(), in_im(), in_re(), indent(), out_im(), and out_re().

Referenced by generate_dslash().

Here is the call graph for this function:

Here is the caller graph for this function:

sign()

def deg_tm_dslash_cuda_gen.sign

(

x

)

Definition at line 104 of file deg_tm_dslash_cuda_gen.py.

Referenced by __fast_pow(), accumulateComplexProduct(), accumulateConjugateProduct(), applyGaugeFieldScaling_long(), quda.computeCoarseLocal(), quda.computeYreverse(), gen(), quda::clover::BQCDOrder< Float, length >.load(), quda::gauge::FloatNOrder< Float, length, N, reconLenParam, stag_phase, huge_alloc >.load(), quda::gauge.milcStaggeredPhase(), quda::ColorSpinor< Float, Nc, 4 >.project(), quda::ColorSpinor< Float, Nc, 2 >.reconstruct(), and smallSVD().

Here is the caller graph for this function:

spinor()

def deg_tm_dslash_cuda_gen.spinor	(		name,
			s,
			c,
			z
	)

Definition at line 136 of file deg_tm_dslash_cuda_gen.py.

tmp_im()

def deg_tm_dslash_cuda_gen.tmp_im	(		s,
			c
	)

Definition at line 134 of file deg_tm_dslash_cuda_gen.py.

tmp_re()

def deg_tm_dslash_cuda_gen.tmp_re	(		s,
			c
	)

Definition at line 133 of file deg_tm_dslash_cuda_gen.py.

twisted_xpay()

def deg_tm_dslash_cuda_gen.twisted_xpay

(

)

Definition at line 735 of file deg_tm_dslash_cuda_gen.py.

References acc_im(), acc_re(), out_im(), and out_re().

Referenced by epilog().

Here is the call graph for this function:

Here is the caller graph for this function:

Variable Documentation

arch

int deg_tm_dslash_cuda_gen.arch = 200

Definition at line 967 of file deg_tm_dslash_cuda_gen.py.

dagger

bool deg_tm_dslash_cuda_gen.dagger = False

Definition at line 960 of file deg_tm_dslash_cuda_gen.py.

Referenced by quda.ApplyCoarse(), quda.ApplyTwistClover(), quda.ApplyTwistGamma(), quda.asymCloverDslashCuda(), clover_dslash(), clover_mat(), clover_matpc(), quda::CloverArg< Float, nSpin, nColor, dynamic_clover_ >.CloverArg(), quda.cloverDslashCuda(), quda::cudaColorSpinorField.commsStart(), quda.domainWallDslashCuda(), quda::DiracTwistedCloverPC.Dslash(), quda::DiracTwistedCloverPC.DslashXpay(), quda.exchangeExtendedGhost(), quda::cudaColorSpinorField.exchangeGhost(), quda::cpuColorSpinorField.exchangeGhost(), quda::GammaArg< Float, nColor >.GammaArg(), quda::cudaColorSpinorField.gather(), quda.genericPackGhost(), quda.improvedStaggeredDslashCuda(), MatPC(), mdw_mat(), mdw_matpc(), quda.MDWFDslashCuda(), ndeg_twist_gamma5(), quda.ndegTwistedMassDslashCuda(), ndegTwistGamma5(), quda::cudaColorSpinorField.pack(), quda::cudaColorSpinorField.packExtended(), quda::cudaColorSpinorField.packGhost(), quda::cpuColorSpinorField.packGhost(), quda::cudaColorSpinorField.packGhostExtended(), printQudaInvertParam(), quda::cudaColorSpinorField.scatter(), quda::cudaColorSpinorField.scatterExtended(), quda::cudaColorSpinorField.sendGhost(), quda::cudaColorSpinorField.sendStart(), quda.shiftColorSpinorField(), quda.staggeredDslashCuda(), tmc_dslash(), tmc_mat(), tmc_matpc(), twistCloverGamma5(), quda.twistedCloverDslashCuda(), quda.twistedMassDslashCuda(), twistGamma5(), quda::cudaColorSpinorField.unpackGhostExtended(), and quda.wilsonDslashCuda().

dslash

bool deg_tm_dslash_cuda_gen.dslash = False

Definition at line 959 of file deg_tm_dslash_cuda_gen.py.

Referenced by anonymous_namespace{dslash_policy.cuh}::DslashPolicyTune.apply(), quda.asymCloverDslashCuda(), quda.cloverDslashCuda(), quda.domainWallDslashCuda(), anonymous_namespace{dslash_policy.cuh}::DslashPolicyTune.DslashPolicyTune(), quda.improvedStaggeredDslashCuda(), quda.MDWFDslashCuda(), quda.ndegTwistedMassDslashCuda(), anonymous_namespace{dslash_policy.cuh}::DslashPthreads.operator()(), quda.staggeredDslashCuda(), quda.twistedCloverDslashCuda(), quda.twistedMassDslashCuda(), and quda.wilsonDslashCuda().

f

deg_tm_dslash_cuda_gen.f = open('dslash_core/wilson_pack_twisted_face_core.h', 'w')

Definition at line 980 of file deg_tm_dslash_cuda_gen.py.

gamma1

def deg_tm_dslash_cuda_gen.gamma1

Initial value:

=  complexify([
    0,  0, 0, 1j,
    0,  0, 1j, 0,
    0, -1j, 0, 0,
    -1j,  0, 0, 0
])

Definition at line 38 of file deg_tm_dslash_cuda_gen.py.

gamma2

def deg_tm_dslash_cuda_gen.gamma2

Initial value:

=  complexify([
    0, 0, 0, 1,
    0, 0, -1,  0,
    0, -1, 0,  0,
    1, 0, 0,  0
])

Definition at line 45 of file deg_tm_dslash_cuda_gen.py.

gamma3

def deg_tm_dslash_cuda_gen.gamma3

Initial value:

=  complexify([
    0, 0, 1j,  0,
    0, 0, 0, -1j,
    -1j, 0, 0,  0,
    0, 1j, 0,  0
])

Definition at line 52 of file deg_tm_dslash_cuda_gen.py.

gamma4

def deg_tm_dslash_cuda_gen.gamma4

Initial value:

=  complexify([
    1, 0, 0, 0,
    0, 1, 0, 0,
    0, 0, -1, 0,
    0, 0, 0, -1
])

Definition at line 59 of file deg_tm_dslash_cuda_gen.py.

id

def deg_tm_dslash_cuda_gen.id

Initial value:

=  complexify([
    1, 0, 0, 0,
    0, 1, 0, 0,
    0, 0, 1, 0,
    0, 0, 0, 1
])

projector matrices ########################################################################

Definition at line 31 of file deg_tm_dslash_cuda_gen.py.

Referenced by quda::cudaColorSpinorField.CopySubset(), do_color_matrix_hisq_force_reference(), do_halfwilson_hisq_force_reference(), and Identity< N, T >.operator()().

igamma5

def deg_tm_dslash_cuda_gen.igamma5

Initial value:

=  complexify([
    0, 0, 1j, 0,
    0, 0, 0, 1j,
    1j, 0, 0, 0,
    0, 1j, 0, 0
])

Definition at line 66 of file deg_tm_dslash_cuda_gen.py.

pack

bool deg_tm_dslash_cuda_gen.pack = False

Definition at line 964 of file deg_tm_dslash_cuda_gen.py.

Referenced by anonymous_namespace{dslash_policy.cuh}.issuePack(), anonymous_namespace{dslash_policy.cuh}::DslashPthreads.operator()(), and set_kernel_pack_t_().

projectors

list deg_tm_dslash_cuda_gen.projectors

Initial value:

=  [
    gminus(id,gamma1), gplus(id,gamma1),
    gminus(id,gamma2), gplus(id,gamma2),
    gminus(id,gamma3), gplus(id,gamma3),
    gminus(id,gamma4), gplus(id,gamma4),
]

Definition at line 88 of file deg_tm_dslash_cuda_gen.py.

sharedDslash

bool deg_tm_dslash_cuda_gen.sharedDslash = False

Definition at line 963 of file deg_tm_dslash_cuda_gen.py.

sharedFloats

int deg_tm_dslash_cuda_gen.sharedFloats = 0

Definition at line 962 of file deg_tm_dslash_cuda_gen.py.

twist

bool deg_tm_dslash_cuda_gen.twist = False

Definition at line 961 of file deg_tm_dslash_cuda_gen.py.

Referenced by quda.ApplyTwistClover(), quda::GammaArg< Float, nColor >.GammaArg(), ndeg_twist_gamma5(), ndegTwistGamma5(), twist_gamma5(), twistCloverGamma5(), and twistGamma5().