The SequentialNode contains a set of passes that transform Relax programs from one AST to another semantically equivalent one. More...

#include <transform.h>

Inheritance diagram for tvm::transform::SequentialNode:

Collaboration diagram for tvm::transform::SequentialNode:

Public Member Functions
PassInfo	Info () const override
	Get the pass information/meta data. More...

void	ResolveDependency (const IRModule &mod)
	Resolve the pass dependency. It globs all required passes by a given pass and executes them. More...

IRModule	operator() (IRModule mod, const PassContext &pass_ctx) const final
	Perform optimizations on a series of passes. The aforementioned typical pass manager jobs could be done by it. This function could be overloaded to focus on different metrics, i.e. performance, memory footprint, etc. More...

	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("transform.Sequential", SequentialNode, PassNode)

Public Member Functions inherited from tvm::transform::PassNode
virtual	~PassNode ()

IRModule	operator() (IRModule mod) const
	Transform mod using the default PassContext in the current scope. More...

	TVM_FFI_DECLARE_OBJECT_INFO ("transform.Pass", PassNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
PassInfo	pass_info

tvm::ffi::Array< Pass >	passes
	A list of passes that used to compose a sequential pass. More...

Detailed Description

The SequentialNode contains a set of passes that transform Relax programs from one AST to another semantically equivalent one.

One example of this level of pass is that the pass manager needs to correctly perform a host of optimizations with a given optimization level and disabled passes.

Member Function Documentation

◆ Info()

PassInfo tvm::transform::SequentialNode::Info ( ) const

inlineoverridevirtual

Get the pass information/meta data.

Implements tvm::transform::PassNode.

◆ operator()()

IRModule tvm::transform::SequentialNode::operator()	(	IRModule	mod,
		const PassContext &	pass_ctx
	)		const

finalvirtual

Perform optimizations on a series of passes. The aforementioned typical pass manager jobs could be done by it. This function could be overloaded to focus on different metrics, i.e. performance, memory footprint, etc.

Parameters

mod	The module that these passes are applied on.
pass_ctx	The context that these passes execute on.

Returns: Return the updated module.

Implements tvm::transform::PassNode.

◆ RegisterReflection()

static void tvm::transform::SequentialNode::RegisterReflection ( )

inlinestatic

◆ ResolveDependency()

void tvm::transform::SequentialNode::ResolveDependency ( const IRModule & mod )

Resolve the pass dependency. It globs all required passes by a given pass and executes them.

Parameters

mod	The module that an optimization pass runs on.

TODO(zhiics) Build a dependency graph among the passes using provided metadata, i.e. required_passes. Likely, we can have a data structure, i.e. PassInfo, to store the relevant information including the parent passes.

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::transform::SequentialNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"transform.Sequential"	,
		SequentialNode	,
		PassNode
	)

Member Data Documentation

◆ pass_info

PassInfo tvm::transform::SequentialNode::pass_info

◆ passes

tvm::ffi::Array<Pass> tvm::transform::SequentialNode::passes

A list of passes that used to compose a sequential pass.

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

include/tvm/ir/transform.h

Public Member Functions

Static Public Member Functions

Public Attributes