tvm
|
Primitive functions that contains TIR statements. More...
#include <function.h>
Public Member Functions | |
FuncType | func_type_annotation () const |
Return the derived function annotation of this function. More... | |
TVM_OBJECT_ENABLE_SCRIPT_PRINTER () | |
TVM_DECLARE_FINAL_OBJECT_INFO (PrimFuncNode, BaseFuncNode) | |
![]() | |
template<typename TObjectRef > | |
Optional< TObjectRef > | GetAttr (const std::string &attr_key, Optional< TObjectRef > default_value=std::nullopt) const |
Get a function attribute. More... | |
template<typename TObjectRef > | |
Optional< TObjectRef > | GetAttr (const std::string &attr_key, TObjectRef default_value) const |
bool | HasNonzeroAttr (const std::string &attr_key) const |
Check whether the function has an non-zero integer attr. More... | |
LinkageType | GetLinkageType () const |
Get the type of the linkage. More... | |
TVM_DECLARE_BASE_OBJECT_INFO (BaseFuncNode, RelaxExprNode) | |
![]() | |
TVM_DECLARE_BASE_OBJECT_INFO (RelaxExprNode, BaseExprNode) | |
![]() | |
TVM_DECLARE_BASE_OBJECT_INFO (BaseExprNode, Object) | |
Static Public Member Functions | |
static void | RegisterReflection () |
![]() | |
static void | RegisterReflection () |
![]() | |
static void | RegisterReflection () |
![]() | |
static void | RegisterReflection () |
Public Attributes | |
Array< tir::Var > | params |
Function parameters. More... | |
Type | ret_type |
The return type of the function. More... | |
Map< tir::Var, Buffer > | buffer_map |
Maps some parameters to specific Buffer data structures. More... | |
tir::Stmt | body |
The body of the function. More... | |
![]() | |
DictAttrs | attrs |
Additional attributes storing the meta-data. More... | |
![]() | |
Optional< ObjectRef > | struct_info_ = Optional<ObjectRef>() |
Stores the result of structure information of the expression that encapsulate both static shape and runtime information such as shape. More... | |
![]() | |
Span | span |
Span that points to the original source code. Reserved debug information. More... | |
Static Public Attributes | |
static constexpr const char * | _type_key = "tir.PrimFunc" |
![]() | |
static constexpr const char * | _type_key = "ir.BaseFunc" |
static constexpr const uint32_t | _type_child_slots = 2 |
![]() | |
static constexpr const char * | _type_key = "ir.RelaxExpr" |
static constexpr const uint32_t | _type_child_slots = 22 |
![]() | |
static constexpr const char * | _type_key = "ir.BaseExpr" |
static constexpr TVMFFISEqHashKind | _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode |
static constexpr const uint32_t | _type_child_slots = 64 |
Primitive functions that contains TIR statements.
The PrimFunc provides low-level code representation does not automatically manage
FuncType tvm::tir::PrimFuncNode::func_type_annotation | ( | ) | const |
Return the derived function annotation of this function.
|
inlinestatic |
tvm::tir::PrimFuncNode::TVM_DECLARE_FINAL_OBJECT_INFO | ( | PrimFuncNode | , |
BaseFuncNode | |||
) |
tvm::tir::PrimFuncNode::TVM_OBJECT_ENABLE_SCRIPT_PRINTER | ( | ) |
|
staticconstexpr |
tir::Stmt tvm::tir::PrimFuncNode::body |
The body of the function.
Maps some parameters to specific Buffer data structures.
buffer_map provides a way to express data structure's field and shape constraints. The provided information is used in the program analysis and the code generation.
buffer_map is a sugar to express:
While we could have express parameter unpacking and constraint using normal statements, making buffer_map as first class citizen of PrimFunc will make program analysis much easier.
Prior to buffer flattening, which is performed FlattenBuffer for TIR-based schedules, these buffer objects are used directly in the body of the function. After buffer flattening, these buffer objects remain unflattened for use in argument validation, but all usage in the body of the function is done through a flattened alias of the buffer.
Array<tir::Var> tvm::tir::PrimFuncNode::params |
Function parameters.
Type tvm::tir::PrimFuncNode::ret_type |
The return type of the function.