tvm::relax::backend::adreno::transform Namespace Reference

Typedefs
using	Pass = tvm::transform::Pass
using	PassInfo = tvm::transform::PassInfo
using	PassContext = tvm::transform::PassContext
using	Function = tvm::relax::Function
using	DataflowBlock = tvm::relax::DataflowBlock

Functions
Pass	AnnotateCustomMemoryScope (Target target)
	This pass is designed to annotate the memory scope information via VDevice attribute. This pass need operator attrbutes which in general vanish aftre legalization. FuseOps and FuseTIR are modified to pass on the operator specific attributes and also op_pattern details as part of the PrimFunc. This pass is Adreno specific and annotates each BindingVar with appropriate HintInDevice. RealizeVDevice pass followed by handles these hints. Followed by this pass we also invoke SpecializePrimFuncBasedOnCallSite which updates the var_buffer_map based on this new VDevice information. More...
Pass	FoldVDeviceScopeChange ()

Typedef Documentation

DataflowBlock

using tvm::relax::backend::adreno::transform::DataflowBlock = typedef tvm::relax::DataflowBlock

Function

using tvm::relax::backend::adreno::transform::Function = typedef tvm::relax::Function

Pass

using tvm::relax::backend::adreno::transform::Pass = typedef tvm::transform::Pass

PassContext

using tvm::relax::backend::adreno::transform::PassContext = typedef tvm::transform::PassContext

PassInfo

using tvm::relax::backend::adreno::transform::PassInfo = typedef tvm::transform::PassInfo

Function Documentation

AnnotateCustomMemoryScope()

Pass tvm::relax::backend::adreno::transform::AnnotateCustomMemoryScope

(

Target

target

)

This pass is designed to annotate the memory scope information via VDevice attribute. This pass need operator attrbutes which in general vanish aftre legalization. FuseOps and FuseTIR are modified to pass on the operator specific attributes and also op_pattern details as part of the PrimFunc. This pass is Adreno specific and annotates each BindingVar with appropriate HintInDevice. RealizeVDevice pass followed by handles these hints. Followed by this pass we also invoke SpecializePrimFuncBasedOnCallSite which updates the var_buffer_map based on this new VDevice information.

FoldVDeviceScopeChange()

Pass tvm::relax::backend::adreno::transform::FoldVDeviceScopeChange

(

)