Brick Library 0.1
Performance-portable stencil datalayout & codegen
Loading...
Searching...
No Matches
Public Member Functions | Public Attributes | Static Public Attributes | Protected Member Functions | List of all members
st.codegen.base.CodeGen Class Reference

Public Member Functions

def __init__ (self, backend, layout, *dag_msize=5, K=2, klimit=20, scatter_thres=1.5, min_fixed=5, unroll=False, dimsplit=False, stride=None)
 
def config (self, backend=None, layout=None)
 
def split (self, rst, nxt, dim, left_dim, f)
 
def gencode (self, List[Grid] grid_out, outfile=sys.stdout)
 

Public Attributes

 LAYOUTREL
 
 TILE_DIM
 
 backend
 
 layout
 
 FOLD
 
 TILE
 
 STRIDE
 
 DIRECT
 
 K
 
 KLIMIT
 
 THRES
 
 MSIZE
 
 MIN_FIXED
 
 UNROLL
 
 DIM_SPLIT
 
 ALIGNED
 
 grids
 
 TILE_SIZE
 
 groups
 

Static Public Attributes

bool FMA = True
 

Protected Member Functions

def _calculate (self)
 

Constructor & Destructor Documentation

◆ __init__()

def st.codegen.base.CodeGen.__init__ (   self,
  backend,
  layout,
dag_msize = 5,
  K = 2,
  klimit = 20,
  scatter_thres = 1.5,
  min_fixed = 5,
  unroll = False,
  dimsplit = False,
  stride = None 
)

Member Function Documentation

◆ _calculate()

def st.codegen.base.CodeGen._calculate (   self)
protected
Here is the call graph for this function:

◆ config()

def st.codegen.base.CodeGen.config (   self,
  backend = None,
  layout = None 
)

◆ gencode()

def st.codegen.base.CodeGen.gencode (   self,
List[Grid grid_out,
  outfile = sys.stdout 
)
Here is the call graph for this function:

◆ split()

def st.codegen.base.CodeGen.split (   self,
  rst,
  nxt,
  dim,
  left_dim,
  f 
)

Member Data Documentation

◆ ALIGNED

st.codegen.base.CodeGen.ALIGNED

◆ backend

st.codegen.base.CodeGen.backend

◆ DIM_SPLIT

st.codegen.base.CodeGen.DIM_SPLIT

◆ DIRECT

st.codegen.base.CodeGen.DIRECT

◆ FMA

bool st.codegen.base.CodeGen.FMA = True
static

◆ FOLD

st.codegen.base.CodeGen.FOLD

◆ grids

st.codegen.base.CodeGen.grids

◆ groups

st.codegen.base.CodeGen.groups

◆ K

st.codegen.base.CodeGen.K

◆ KLIMIT

st.codegen.base.CodeGen.KLIMIT

◆ layout

st.codegen.base.CodeGen.layout

◆ LAYOUTREL

st.codegen.base.CodeGen.LAYOUTREL

◆ MIN_FIXED

st.codegen.base.CodeGen.MIN_FIXED

◆ MSIZE

st.codegen.base.CodeGen.MSIZE

◆ STRIDE

st.codegen.base.CodeGen.STRIDE

◆ THRES

st.codegen.base.CodeGen.THRES

◆ TILE

st.codegen.base.CodeGen.TILE

◆ TILE_DIM

st.codegen.base.CodeGen.TILE_DIM

◆ TILE_SIZE

st.codegen.base.CodeGen.TILE_SIZE

◆ UNROLL

st.codegen.base.CodeGen.UNROLL

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