transform.h File Reference

Utilities for IR transformations. More...

#include <tvm/ffi/container/array.h>
#include <tvm/ffi/function.h>
#include <tvm/ffi/reflection/creator.h>
#include <tvm/ffi/reflection/registry.h>
#include <tvm/ffi/string.h>
#include <tvm/ir/module.h>
#include <tvm/ir/with_context.h>
#include <string>
#include <type_traits>
#include <utility>

Include dependency graph for transform.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	tvm::transform::PassContext
	PassContext that is used to configure the pass behavior. More...
class	tvm::transform::PassInfoObj
	Meta data that will be used to help optimization and analysis. More...
class	tvm::transform::PassInfo
	Managed reference class for PassInfoObj. More...
class	tvm::transform::PassObj
	PassObj is the base type of differnt types of optimization passes. It is designed as a pure class and implemented by different pass subclasses at different granularity of Relax nodes. More...
class	tvm::transform::Pass
class	tvm::transform::SequentialObj
	The SequentialObj contains a set of passes that transform Relax programs from one AST to another semantically equivalent one. More...
class	tvm::transform::Sequential
class	tvm::transform::PassInstrumentObj
	PassInstrumentObj forms an instrument implementation. It provides API for users to register callbacks at different instrumentation points. More...
class	tvm::transform::PassInstrument
	Managed reference class for PassInstrumentObj. More...
class	tvm::transform::PassContextObj
	PassContextObj contains the information that a pass can rely on, such as analysis results. More...

Namespaces
namespace	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
namespace	tvm::transform

Functions
template<typename TConfig >
TConfig	tvm::transform::PassConfigWithDefaults ()
	Create a pass-config object with all default values, using the reflection defaults.
TVM_TIRX_DLL Pass	tvm::transform::CreateModulePass (std::function< IRModule(IRModule, PassContext)> pass_func, int opt_level, ffi::String name)
TVM_TIRX_DLL Pass	tvm::transform::PrintIR (ffi::String header="")
	A special trace pass that prints the header and IR to LOG(INFO).
TVM_TIRX_DLL ffi::Error	tvm::transform::EnrichPassErrorWithContext (const ffi::Error &err, const IRModule &mod, ffi::String pass_name, ffi::Optional< GlobalVar > func=ffi::Optional< GlobalVar >(std::nullopt))
	Enrich a pass-time error with a TVMScript-rendered, underlined source location derived from the error's embedded VisitErrorContext.

Detailed Description

Utilities for IR transformations.

Pass represents an IRModule-to-IRModule transformation. PassContext provides scoped configuration and instrumentation. Sequential applies a list of passes in order.