Deeploy.TilingExtension.CodeTransformationPasses.TilingVariableReplacement.TilingVariableReplacementUpdate

class Deeploy.TilingExtension.CodeTransformationPasses.TilingVariableReplacement.TilingVariableReplacementUpdate(targetMemLevel: str, tileIdxVar: str = 'TILING_I')

Bases: CodeTransformationPass, IntrospectiveCodeTransformationMixIn, TilingHoistingMixIn

Methods

__init__(targetMemLevel: str, tileIdxVar: str = 'TILING_I')

__init__(targetMemLevel[, tileIdxVar])

apply(ctxt, executionBlock, name[, verbose])

Apply the CodeTransformation to an ExecutionBlock

dereferenceVars(template, varNames)

extractDynamicReferences(ctxt[, ...])

indexVars(template, varNames, index)

Attributes

parseTreeDict

prefix

apply(ctxt: NetworkContext, executionBlock: ExecutionBlock, name: str, verbose: CodeGenVerbosity = CodeGenVerbosity(tilingProfiling=None, untiledProfiling=None)) → Tuple[NetworkContext, ExecutionBlock]

Apply the CodeTransformation to an ExecutionBlock

Parameters:

  • ctxt (NetworkContext) – Current NetworkContext

  • executionBlock (ExecutionBlock) – ExecutionBlock whose code you’d like to transform

  • name (str) – Graph node name of the operator being targetted

  • verbose (CodeGenVerbosity) – Control the verbosity of code generation

Returns:

Updated NetworkContext and ExecutionBlock

Return type:

Tuple[NetworkContext, ExecutionBlock]