st.codegen.backend.avx512.BackendAVX512 Class Reference

Inheritance diagram for st.codegen.backend.avx512.BackendAVX512:

Collaboration diagram for st.codegen.backend.avx512.BackendAVX512:

Public Member Functions
def	__init__ (self)
def	setLayout (self, layout)
def	prequel (self, toplevel)
def	declare_buf (self, Buffer buf, CodeBlock block)
def	declare_gridref (self, Grid grid, CodeBlock block)
def	genVectorLoop (self, CodeBlock group)
def	genStoreLoop (self, CodeBlock group)
def	gen_lhs (self, Buffer buf, List[int] offset, rel=None, dimrels=None)
def	gen_rhs (self, Expr comp, List[int] shift, List[int] offset, rel=None, dimrels=None)
def	declare_reg (self, name, CodeBlock block)
def	declare_vec (self, name, CodeBlock block)
def	store_vecbuf (self, vecbuf_name, reg_name, CodeBlock block)
def	merge (self, rego, regl, regr, dim, shift, CodeBlock block)
def	read_aligned (self, Grid grid, offset, str name, CodeBlock block, rel=None)
Public Member Functions inherited from st.codegen.backend.base.Backend
def	__init__ (self)
def	setCodeGen (self, codegen)
def	setLayout (self, layout)
def	prequel (self, toplevel)
def	checkConfig (self)
def	declare_buf (self, Buffer buf, CodeBlock block)
def	declare_gridref (self, Grid grid, CodeBlock block)
def	stride (self, dim)
def	declare_vecbuf (self, Grid grid, vec_shift, CodeBlock block)
def	genVectorLoop (self, CodeBlock group)
def	genStoreLoop (self, CodeBlock group)
def	store (self, Buffer buf, CodeBlock group)
def	storeTile (self, Buffer buf, CodeBlock group)
def	genStoreLoc (self, Grid grid, shift, offset, rel, dimrels)
def	genStoreTileLoop (self, CodeBlock group, dims)
def	gen_lhs (self, Buffer buf, List[int] offset, rel=None, dimrels=None)
def	gen_rhs (self, Expr comp, List[int] shift, List[int] offset, rel=None, dimrels=None)
def	declare_reg (self, name, CodeBlock block)
def	declare_vec (self, name, CodeBlock block)
def	store_vecbuf (self, vecbuf_name, reg_name, CodeBlock block)
def	merge (self, rego, regl, regr, dim, shift, CodeBlock block)
def	read_aligned (self, Grid grid, offset, str name, CodeBlock block, rel=None)

Public Attributes
	VECLEN
Public Attributes inherited from st.codegen.backend.base.Backend
	printer
	codegen
	prec
	layout
	STRIDE
	VECLEN
	ALIGNED

Additional Inherited Members
Static Public Member Functions inherited from st.codegen.backend.base.Backend
def	gridref_name (grid)
def	vectmp_name (idx)
def	vecbuf_name (grid, vec_shift)
def	vecreg_name (grid, vec_shift)
def	index_name (idx)
def	rel_name (idx=None)
Static Public Attributes inherited from st.codegen.backend.base.Backend
str	preffix = "_cg"

Constructor & Destructor Documentation

__init__()

def st.codegen.backend.avx512.BackendAVX512.__init__

(

self

)

Reimplemented from st.codegen.backend.base.Backend.

Reimplemented in st.codegen.backend.asimd.BackendASIMD, st.codegen.backend.avx2.BackendAVX2, st.codegen.backend.flex.BackendFlex, st.codegen.backend.sse.BackendSSE, and st.codegen.backend.asimd.BackendSVE.

Here is the call graph for this function:

Member Function Documentation

declare_buf()

def st.codegen.backend.avx512.BackendAVX512.declare_buf	(		self,
		Buffer	buf,
		CodeBlock	block
	)

Reimplemented from st.codegen.backend.base.Backend.

declare_gridref()

def st.codegen.backend.avx512.BackendAVX512.declare_gridref	(		self,
		Grid	grid,
		CodeBlock	block
	)

Reimplemented from st.codegen.backend.base.Backend.

Here is the call graph for this function:

declare_reg()

def st.codegen.backend.avx512.BackendAVX512.declare_reg	(		self,
			name,
		CodeBlock	block
	)

Reimplemented from st.codegen.backend.base.Backend.

declare_vec()

def st.codegen.backend.avx512.BackendAVX512.declare_vec	(		self,
			name,
		CodeBlock	block
	)

Reimplemented from st.codegen.backend.base.Backend.

Reimplemented in st.codegen.backend.asimd.BackendASIMD, st.codegen.backend.asimd.BackendSVE, st.codegen.backend.avx2.BackendAVX2, st.codegen.backend.flex.BackendFlex, and st.codegen.backend.sse.BackendSSE.

gen_lhs()

def st.codegen.backend.avx512.BackendAVX512.gen_lhs	(		self,
		Buffer	buf,
		List[int]	offset,
			rel = None,
			dimrels = None
	)

Reimplemented from st.codegen.backend.base.Backend.

gen_rhs()

def st.codegen.backend.avx512.BackendAVX512.gen_rhs	(		self,
		Expr	comp,
		List[int]	shift,
		List[int]	offset,
			rel = None,
			dimrels = None
	)

:param comp: The expression to print
:param shift: The shift of scatter
:param offset: Scattered from
:param rel: Added offset when using loops
:return:

Reimplemented from st.codegen.backend.base.Backend.

genStoreLoop()

def st.codegen.backend.avx512.BackendAVX512.genStoreLoop	(		self,
		CodeBlock	group
	)

Reimplemented from st.codegen.backend.base.Backend.

genVectorLoop()

def st.codegen.backend.avx512.BackendAVX512.genVectorLoop	(		self,
		CodeBlock	group
	)

Reimplemented from st.codegen.backend.base.Backend.

merge()

def st.codegen.backend.avx512.BackendAVX512.merge	(		self,
			rego,
			regl,
			regr,
			dim,
			shift,
		CodeBlock	block
	)

Reimplemented from st.codegen.backend.base.Backend.

Reimplemented in st.codegen.backend.asimd.BackendASIMD, st.codegen.backend.asimd.BackendSVE, st.codegen.backend.avx2.BackendAVX2, st.codegen.backend.flex.BackendFlex, and st.codegen.backend.sse.BackendSSE.

Here is the call graph for this function:

prequel()

def st.codegen.backend.avx512.BackendAVX512.prequel	(		self,
			toplevel
	)

Reimplemented from st.codegen.backend.base.Backend.

read_aligned()

def st.codegen.backend.avx512.BackendAVX512.read_aligned	(		self,
		Grid	grid,
			offset,
		str	name,
		CodeBlock	block,
			rel = None
	)

Reimplemented from st.codegen.backend.base.Backend.

Reimplemented in st.codegen.backend.asimd.BackendASIMD, st.codegen.backend.asimd.BackendSVE, st.codegen.backend.avx2.BackendAVX2, st.codegen.backend.flex.BackendFlex, and st.codegen.backend.sse.BackendSSE.

setLayout()

def st.codegen.backend.avx512.BackendAVX512.setLayout	(		self,
			layout
	)

Reimplemented from st.codegen.backend.base.Backend.

Reimplemented in st.codegen.backend.flex.BackendFlex.

Here is the call graph for this function:

store_vecbuf()

def st.codegen.backend.avx512.BackendAVX512.store_vecbuf	(		self,
			vecbuf_name,
			reg_name,
		CodeBlock	block
	)

Reimplemented from st.codegen.backend.base.Backend.

Reimplemented in st.codegen.backend.asimd.BackendASIMD, st.codegen.backend.asimd.BackendSVE, st.codegen.backend.avx2.BackendAVX2, st.codegen.backend.flex.BackendFlex, and st.codegen.backend.sse.BackendSSE.

Member Data Documentation

VECLEN

st.codegen.backend.avx512.BackendAVX512.VECLEN

---

The documentation for this class was generated from the following file:

• codegen/st/codegen/backend/avx512.py