49 #ifndef TVM_IR_TYPE_H_
50 #define TVM_IR_TYPE_H_
52 #include <tvm/ffi/container/array.h>
53 #include <tvm/ffi/reflection/registry.h>
86 refl::ObjectDef<TypeNode>().def_ro(
"span", &
TypeNode::span, refl::DefaultValue(
Span()),
87 refl::AttachFieldFlag::SEqHashIgnore());
101 class Type :
public ObjectRef {
169 refl::ObjectDef<PointerTypeNode>()
174 static constexpr
const char*
_type_key =
"ir.PointerType";
207 refl::ObjectDef<TupleTypeNode>()
212 static constexpr
const char*
_type_key =
"ir.TupleType";
249 return n && n->
fields.size() == 0;
269 refl::ObjectDef<FuncTypeNode>()
308 static constexpr
const char*
_type_key =
"ir.TensorMapType";
Function type.
Definition: type.h:260
static constexpr const char * _type_key
Definition: type.h:275
Type ret_type
The type of return value.
Definition: type.h:265
TVM_DECLARE_FINAL_OBJECT_INFO(FuncTypeNode, TypeNode)
Array< Type > arg_types
type type of arguments
Definition: type.h:263
static void RegisterReflection()
Definition: type.h:267
Managed reference to FuncTypeNode.
Definition: type.h:283
FuncType(Array< Type > arg_types, Type ret_type, Span span=Span())
Constructor.
TVM_DEFINE_OBJECT_REF_METHODS(FuncType, Type, FuncTypeNode)
Low-level raw pointer type.
Definition: type.h:156
Type element_type
The type of the element which the pointer points to.
Definition: type.h:161
TVM_DECLARE_FINAL_OBJECT_INFO(PointerTypeNode, TypeNode)
static constexpr const char * _type_key
Definition: type.h:174
String storage_scope
The storage scope of the pointer.
Definition: type.h:165
static void RegisterReflection()
Definition: type.h:167
TVM_DEFINE_OBJECT_REF_METHODS(PointerType, Type, PointerTypeNode)
PointerType(Type element_type, String storage_scope="")
Constructor.
Primitive data types used in the low-level IR.
Definition: type.h:114
runtime::DataType dtype
The corresponding dtype field.
Definition: type.h:119
static constexpr const char * _type_key
Definition: type.h:126
TVM_DECLARE_FINAL_OBJECT_INFO(PrimTypeNode, TypeNode)
static void RegisterReflection()
Definition: type.h:121
PrimType(runtime::DataType dtype, Span span=Span())
Constructor.
TVM_DEFINE_OBJECT_REF_METHODS(PrimType, Type, PrimTypeNode)
Definition: source_map.h:113
The type of tensor map.
Definition: type.h:301
TVM_DECLARE_FINAL_OBJECT_INFO(TensorMapTypeNode, TypeNode)
static void RegisterReflection()
Definition: type.h:303
static constexpr const char * _type_key
Definition: type.h:308
Managed reference to TensorMapTypeNode.
Definition: type.h:316
TensorMapType(Span span=Span())
TVM_DEFINE_OBJECT_REF_METHODS_WITHOUT_DEFAULT_CONSTRUCTOR(TensorMapType, Type, TensorMapTypeNode)
The type of tuple values.
Definition: type.h:198
TupleTypeNode()
Definition: type.h:203
Array< Type > fields
The type of each field in the tuple.
Definition: type.h:201
static constexpr const char * _type_key
Definition: type.h:212
static void RegisterReflection()
Definition: type.h:205
TVM_DECLARE_FINAL_OBJECT_INFO(TupleTypeNode, TypeNode)
Managed reference to TupleTypeNode.
Definition: type.h:220
TVM_DEFINE_OBJECT_REF_METHODS(TupleType, Type, TupleTypeNode)
TupleType(Array< Type > fields, Span span=Span())
Constructor.
static TupleType Empty()
Create an empty tuple type that constains nothing.
Type is the base type of all types.
Definition: type.h:75
static constexpr const uint32_t _type_child_slots
Definition: type.h:93
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind
Definition: type.h:90
static void RegisterReflection()
Definition: type.h:83
Span span
Span that points to the original source code. Reserved debug information.
Definition: type.h:81
TVM_DECLARE_BASE_OBJECT_INFO(TypeNode, Object)
static constexpr const char * _type_key
Definition: type.h:91
Managed reference to TypeNode.
Definition: type.h:101
TVM_DEFINE_OBJECT_REF_METHODS(Type, ObjectRef, TypeNode)
Runtime primitive data type.
Definition: data_type.h:47
Definition: repr_printer.h:91
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37
bool IsVoidType(const Type &type)
Check whether the tyep represents void.
Definition: type.h:247
Type VoidType()
Definition: type.h:241
Definitions and helper macros for IR/AST nodes.
A managed object in the TVM runtime.
A map from source names to source code.