stmt_functor.h File Reference

Functors for tirx stmts utility functions to call common functors. More...

#include <tvm/ir/expr_functor.h>
#include <tvm/ir/object_functor.h>
#include <tvm/ir/prim/expr.h>
#include <tvm/tirx/function.h>
#include <tvm/tirx/stmt.h>
#include <unordered_map>
#include <utility>

Include dependency graph for stmt_functor.h:

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

Go to the source code of this file.

Classes
class	tvm::tirx::StmtFunctor< R(const Stmt &, Args...)>
class	tvm::tirx::StmtExprVisitor
	Native visitor for TIRx statements and their expression operands. More...
class	tvm::tirx::StmtExprMutator
	Mutator that recursively mutates stmts and exprs on them. More...

Namespaces
namespace	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
namespace	tvm::tirx
	Backend-defined execution hierarchy nodes shared by layouts and tile calls.

Functions
TVM_TIRX_DLL Stmt	tvm::tirx::SubstituteWithDataTypeLegalization (Stmt stmt, std::function< ffi::Optional< PrimExpr >(const Var &)> vmap)
	Substitute the var specified by vmap and legalize data types after substitution.
TVM_TIRX_DLL PrimExpr	tvm::tirx::SubstituteWithDataTypeLegalization (PrimExpr expr, std::function< ffi::Optional< PrimExpr >(const Var &)> vmap)
	Substitute the var specified by vmap and legalize data types after substitution.
template<typename Node , typename = std::enable_if_t<std::is_base_of_v<StmtNode, Node>>>
bool	tvm::tirx::ContainsNode (const Stmt &stmt)
	Check if the statement contains the specified node type.

Detailed Description

Functors for tirx stmts utility functions to call common functors.