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());
100 class Type :
public ObjectRef {
166 refl::ObjectDef<PointerTypeNode>()
202 refl::ObjectDef<TupleTypeNode>()
242 return n && n->
fields.size() == 0;
262 refl::ObjectDef<FuncTypeNode>()
Function type.
Definition: type.h:253
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("ir.FuncType", FuncTypeNode, TypeNode)
Type ret_type
The type of return value.
Definition: type.h:258
ffi::Array< Type > arg_types
type type of arguments
Definition: type.h:256
static void RegisterReflection()
Definition: type.h:260
Managed reference to FuncTypeNode.
Definition: type.h:274
FuncType(ffi::Array< Type > arg_types, Type ret_type, Span span=Span())
Constructor.
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(FuncType, Type, FuncTypeNode)
Low-level raw pointer type.
Definition: type.h:153
Type element_type
The type of the element which the pointer points to.
Definition: type.h:158
ffi::String storage_scope
The storage scope of the pointer.
Definition: type.h:162
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("ir.PointerType", PointerTypeNode, TypeNode)
static void RegisterReflection()
Definition: type.h:164
PointerType(Type element_type, ffi::String storage_scope="")
Constructor.
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(PointerType, Type, PointerTypeNode)
Primitive data types used in the low-level IR.
Definition: type.h:113
runtime::DataType dtype
The corresponding dtype field.
Definition: type.h:118
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("ir.PrimType", PrimTypeNode, TypeNode)
static void RegisterReflection()
Definition: type.h:120
PrimType(runtime::DataType dtype, Span span=Span())
Constructor.
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(PrimType, Type, PrimTypeNode)
Definition: source_map.h:111
The type of tensor map.
Definition: type.h:292
static void RegisterReflection()
Definition: type.h:294
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("ir.TensorMapType", TensorMapTypeNode, TypeNode)
Managed reference to TensorMapTypeNode.
Definition: type.h:305
TensorMapType(Span span=Span())
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE_WITHOUT_DEFAULT_CONSTRUCTOR(TensorMapType, Type, TensorMapTypeNode)
The type of tuple values.
Definition: type.h:193
TupleTypeNode()
Definition: type.h:198
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("ir.TupleType", TupleTypeNode, TypeNode)
static void RegisterReflection()
Definition: type.h:200
ffi::Array< Type > fields
The type of each field in the tuple.
Definition: type.h:196
Managed reference to TupleTypeNode.
Definition: type.h:213
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(TupleType, Type, TupleTypeNode)
TupleType(ffi::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
TVM_FFI_DECLARE_OBJECT_INFO("ir.Type", TypeNode, Object)
static constexpr const uint32_t _type_child_slots
Definition: type.h:92
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
Managed reference to TypeNode.
Definition: type.h:100
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(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:240
Type VoidType()
Definition: type.h:234
Definitions and helper macros for IR/AST nodes.
A managed object in the TVM runtime.
A map from source names to source code.