tvm
Namespaces | Typedefs | Functions
transform.h File Reference

Adreno GPU specific transformation passes. More...

#include <tvm/relax/expr.h>
#include <tvm/relax/transform.h>
Include dependency graph for transform.h:

Go to the source code of this file.

Namespaces

 tvm
 Performance counters for profiling via the PAPI library.
 
 tvm::relax
 
 tvm::relax::backend
 
 tvm::relax::backend::adreno
 
 tvm::relax::backend::adreno::transform
 

Typedefs

using tvm::relax::backend::adreno::transform::Pass = tvm::transform::Pass
 
using tvm::relax::backend::adreno::transform::PassInfo = tvm::transform::PassInfo
 
using tvm::relax::backend::adreno::transform::PassContext = tvm::transform::PassContext
 
using tvm::relax::backend::adreno::transform::Function = tvm::relax::Function
 
using tvm::relax::backend::adreno::transform::DataflowBlock = tvm::relax::DataflowBlock
 

Functions

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. More...
 
Pass tvm::relax::backend::adreno::transform::FoldVDeviceScopeChange ()
 

Detailed Description

Adreno GPU specific transformation passes.