24 #ifndef TVM_IR_EXPR_H_
25 #define TVM_IR_EXPR_H_
37 #include <type_traits>
58 static constexpr
const char*
_type_key =
"BaseExpr";
393 template <
typename TTypeNode>
394 inline const TTypeNode*
type_as()
const;
464 v->Visit(
"span", &
span);
507 v->Visit(
"dtype", &
dtype);
508 v->Visit(
"value", &
value);
509 v->Visit(
"span", &
span);
554 v->Visit(
"dtype", &
dtype);
555 v->Visit(
"value", &
value);
556 v->Visit(
"span", &
span);
601 operator bool()
const {
return (*this)->value != 0; }
610 return Bool(a.operator
bool() || b.operator
bool());
615 return Bool(a.operator
bool() && b.operator
bool());
621 return a.operator bool() == b.operator bool();
653 template <typename Enum, typename = typename std::enable_if<std::is_enum<Enum>::value>::type>
655 static_assert(std::is_same<
int,
typename std::underlying_type<Enum>::type>::value,
656 "declare enum to be enum int to use visitor");
663 data_ = ObjectRef::GetDataPtr<Object>(other);
670 ICHECK(
data_ !=
nullptr) <<
" Trying to reference a null Integer";
671 return (*this)->value;
675 if (
data_ ==
nullptr)
return Bool(
false);
676 return Bool((*this)->value == other);
679 template <typename Enum, typename = typename std::enable_if<std::is_enum<Enum>::value>::type>
681 return *
this ==
static_cast<int>(other);
683 template <typename Enum, typename = typename std::enable_if<std::is_enum<Enum>::value>::type>
685 return *
this !=
static_cast<int>(other);
704 v->Visit(
"min", &
min);
705 v->Visit(
"extent", &
extent);
706 v->Visit(
"span", &
span);
752 <<
"not populated the checked_type "
753 <<
"field for " << GetRef<RelayExpr>(
this);
757 template <
typename TTypeNode>
759 static_assert(std::is_base_of<TypeNode, TTypeNode>::value,
760 "TType must be a special case of type");
762 <<
"Type inference for this Expr has not completed. Try to call infer_type pass.";
764 ICHECK(node !=
nullptr) <<
"Expected type to be " << TTypeNode::_type_key <<
", but get "
781 int64_t value = val.operator int64_t();
802 return Integer(val.operator
int());
815 int v = val.operator int();
816 ICHECK(v == 0 || v == 1) <<
"ValueError: boolean value can only be 0 or 1, but get " << v;
817 return Bool(
static_cast<bool>(v));
@ kTVMArgInt
Definition: c_runtime_api.h:175
@ kTVMNullptr
Definition: c_runtime_api.h:178
Visitor class to get the attributes of an AST/IR node. The content is going to be called for each fie...
Definition: reflection.h:52
Base type of all the expressions.
Definition: expr.h:50
static constexpr const char * _type_key
Definition: expr.h:58
static constexpr const bool _type_has_method_shash_reduce
Definition: expr.h:60
static constexpr const uint32_t _type_child_slots
Definition: expr.h:61
TVM_DECLARE_BASE_OBJECT_INFO(BaseExprNode, Object)
static constexpr const bool _type_has_method_sequal_reduce
Definition: expr.h:59
Span span
Span that points to the original source code. Reserved debug information.
Definition: expr.h:56
Managed reference to BaseExprNode.
Definition: expr.h:69
TVM_DEFINE_OBJECT_REF_METHODS(BaseExpr, ObjectRef, BaseExprNode)
Boolean constant.
Definition: expr.h:597
Bool operator!() const
Definition: expr.h:600
Bool(bool value, Span span=Span())
Definition: expr.h:599
TVM_DEFINE_NOTNULLABLE_OBJECT_REF_METHODS(Bool, IntImm, IntImmNode)
Constant floating point literals in the program.
Definition: expr.h:548
bool SEqualReduce(const FloatImmNode *other, SEqualReducer equal) const
Definition: expr.h:559
TVM_DECLARE_FINAL_OBJECT_INFO(FloatImmNode, PrimExprNode)
void VisitAttrs(AttrVisitor *v)
Definition: expr.h:553
double value
The constant value content.
Definition: expr.h:551
static constexpr const char * _type_key
Definition: expr.h:568
void SHashReduce(SHashReducer hash_reduce) const
Definition: expr.h:563
Managed reference class to FloatImmNode.
Definition: expr.h:577
TVM_DEFINE_OBJECT_REF_COW_METHOD(FloatImmNode)
TVM_DEFINE_OBJECT_REF_METHODS(FloatImm, PrimExpr, FloatImmNode)
FloatImm(DataType dtype, double value, Span span=Span())
Constructor.
Global variable that lives in the top-level module.
Definition: expr.h:456
TVM_DECLARE_FINAL_OBJECT_INFO(GlobalVarNode, RelayExprNode)
void SHashReduce(SHashReducer hash_reduce) const
Definition: expr.h:474
void VisitAttrs(AttrVisitor *v)
Definition: expr.h:461
static constexpr const char * _type_key
Definition: expr.h:479
String name_hint
The name of the variable, this only acts as a hint.
Definition: expr.h:459
bool SEqualReduce(const GlobalVarNode *other, SEqualReducer equal) const
Definition: expr.h:469
Managed reference to GlobalVarNode.
Definition: expr.h:487
GlobalVar(String name_hint, Type type={}, Span span={})
TVM_DEFINE_OBJECT_REF_METHODS(GlobalVar, RelayExpr, GlobalVarNode)
TVM_DEFINE_OBJECT_REF_COW_METHOD(GlobalVarNode)
Constant integer literals in the program.
Definition: expr.h:501
TVM_DECLARE_FINAL_OBJECT_INFO(IntImmNode, PrimExprNode)
void VisitAttrs(AttrVisitor *v)
Definition: expr.h:506
static constexpr const char * _type_key
Definition: expr.h:521
int64_t value
the Internal value.
Definition: expr.h:504
void SHashReduce(SHashReducer hash_reduce) const
Definition: expr.h:516
bool SEqualReduce(const IntImmNode *other, SEqualReducer equal) const
Definition: expr.h:512
Managed reference class to IntImmNode.
Definition: expr.h:530
TVM_DEFINE_OBJECT_REF_COW_METHOD(IntImmNode)
TVM_DEFINE_OBJECT_REF_METHODS(IntImm, PrimExpr, IntImmNode)
IntImm(DataType dtype, int64_t value, Span span=Span())
Constructor.
Container of constant int that adds more constructors.
Definition: expr.h:632
Bool operator!=(Enum other) const
Definition: expr.h:684
Integer(ObjectPtr< Object > node)
constructor from node.
Definition: expr.h:638
Integer()
Definition: expr.h:634
Bool operator!=(int other) const
Definition: expr.h:678
Integer(Enum value)
Constructor from enum.
Definition: expr.h:654
Bool operator==(int other) const
Definition: expr.h:674
Integer(IntImm other)
Construct integer from int imm.
Definition: expr.h:647
int64_t IntValue() const
convert to int64_t
Definition: expr.h:669
Bool operator==(Enum other) const
Definition: expr.h:680
Integer & operator=(const IntImm &other)
Assign an expression to integer.
Definition: expr.h:662
Integer(int value, Span span=Span())
Construct integer from int value.
Definition: expr.h:642
Base node of all primitive expressions.
Definition: expr.h:86
static constexpr const uint32_t _type_child_slots
Definition: expr.h:107
TVM_OBJECT_ENABLE_SCRIPT_PRINTER()
DataType dtype
The runtime data type of the primitive expression.
Definition: expr.h:102
static constexpr const char * _type_key
Definition: expr.h:106
TVM_DECLARE_BASE_OBJECT_INFO(PrimExprNode, BaseExprNode)
Reference to PrimExprNode.
Definition: expr.h:115
DataType dtype() const
Definition: expr.h:129
TVM_DEFINE_OBJECT_REF_METHODS(PrimExpr, BaseExpr, PrimExprNode)
PrimExpr(float value)
construct from float.
PrimExpr(int32_t value)
construct from integer.
range over one dimension
Definition: expr.h:690
PrimExpr min
beginning of the node
Definition: expr.h:693
void VisitAttrs(AttrVisitor *v)
Definition: expr.h:703
RangeNode(PrimExpr min, PrimExpr extent, Span span=Span())
Definition: expr.h:700
bool SEqualReduce(const RangeNode *other, SEqualReducer equal) const
Definition: expr.h:709
static constexpr const char * _type_key
Definition: expr.h:718
void SHashReduce(SHashReducer hash_reduce) const
Definition: expr.h:713
TVM_DECLARE_FINAL_OBJECT_INFO(RangeNode, Object)
RangeNode()
constructor
Definition: expr.h:699
static constexpr const bool _type_has_method_shash_reduce
Definition: expr.h:720
PrimExpr extent
the extend of range
Definition: expr.h:695
static constexpr const bool _type_has_method_sequal_reduce
Definition: expr.h:719
Span span
the location of this range in the source
Definition: expr.h:697
Range container
Definition: expr.h:725
static Range FromMinExtent(PrimExpr min, PrimExpr extent, Span span=Span())
construct a new range with min and extent The corresponding constructor is removed,...
Range(PrimExpr begin, PrimExpr end, Span span=Span())
constructor by begin and end
TVM_DEFINE_OBJECT_REF_METHODS(Range, ObjectRef, RangeNode)
Base node of all non-primitive expressions.
Definition: expr.h:362
TVM_DECLARE_BASE_OBJECT_INFO(RelayExprNode, BaseExprNode)
const TTypeNode * type_as() const
Check if the inferred(checked) type of the Expr is backed by a TTypeNode and return it.
Definition: expr.h:758
VirtualDevice virtual_device() const
const Type & checked_type() const
Definition: expr.h:750
ObjectRef virtual_device_
The virtual device (VirtualDevice) for this node (the result of device planning). For first-order exp...
Definition: expr.h:418
Optional< ObjectRef > struct_info_
Stores the result of structure information of the expression that encapsulate both static shape and r...
Definition: expr.h:377
Type checked_type_
Stores the result of type inference(type checking).
Definition: expr.h:370
static constexpr const char * _type_key
Definition: expr.h:433
static constexpr const uint32_t _type_child_slots
Definition: expr.h:434
Managed reference to RelayExprNode.
Definition: expr.h:442
TVM_DEFINE_OBJECT_REF_METHODS(RelayExpr, BaseExpr, RelayExprNode)
A Reducer class to reduce the structural equality result of two objects.
Definition: structural_equal.h:126
A Reducer class to reduce the structural hash value.
Definition: structural_hash.h:110
void FreeVarHashImpl(const runtime::Object *var) const
Implementation for hash for a free var.
Definition: structural_hash.h:192
Definition: source_map.h:120
Managed reference to TypeNode.
Definition: type.h:93
Managed reference class to VirtualDeviceNode.
Definition: virtual_device.h:271
Runtime primitive data type.
Definition: data_type.h:43
static DataType Float(int bits, int lanes=1)
Construct an float type.
Definition: data_type.h:234
static DataType Int(int bits, int lanes=1)
Construct an int type.
Definition: data_type.h:217
A custom smart pointer for Object.
Definition: object.h:362
Base class of all object reference.
Definition: object.h:519
bool defined() const
Definition: object.h:552
const Object * get() const
Definition: object.h:554
ObjectPtr< Object > data_
Internal pointer that backs the reference.
Definition: object.h:605
const ObjectType * as() const
Try to downcast the internal Object to a raw pointer of a corresponding type.
Definition: object.h:906
base class of all object containers.
Definition: object.h:171
Optional container that to represent to a Nullable variant of T.
Definition: optional.h:51
Reference to string objects.
Definition: string.h:98
Internal base class to handle conversion to POD values.
Definition: packed_func.h:552
TObjectRef AsObjectRef() const
Definition: packed_func.h:2016
int type_code() const
Definition: packed_func.h:621
IR/AST nodes for the unified type system in TVM.
tvm::Type Type
Definition: type.h:47
tvm::Span Span
Definition: base.h:65
Var var(std::string name_hint, DataType t=DataType::Int(32))
Construct a new Var expression.
runtime implementation for LibTorch/TorchScript.
Definition: analyzer.h:36
PrimExpr operator!=(PrimExpr a, PrimExpr b)
not_equal
PrimExpr max(PrimExpr a, PrimExpr b, Span span=Span())
take maximum of two values
PrimExpr operator/(PrimExpr a, PrimExpr b)
division operator
PrimExpr equal(PrimExpr a, PrimExpr b, Span span=Span())
equal
PrimExpr operator>>(PrimExpr a, PrimExpr b)
right shift operator
PrimExpr operator<(PrimExpr a, PrimExpr b)
less
PrimExpr operator|(PrimExpr a, PrimExpr b)
take bitwise or of two values
PrimExpr operator==(PrimExpr a, PrimExpr b)
equal
PrimExpr operator~(PrimExpr a)
take bitwise negation of two values
PrimExpr operator>=(PrimExpr a, PrimExpr b)
greater_equal
PrimExpr operator<=(PrimExpr a, PrimExpr b)
less_equal
PrimExpr operator*(PrimExpr a, PrimExpr b)
multiplication operator
PrimExpr operator&(PrimExpr a, PrimExpr b)
take bitwise and of two values
PrimExpr min(PrimExpr a, PrimExpr b, Span span=Span())
take minimum of two values
PrimExpr operator!(PrimExpr a)
not
PrimExpr operator^(PrimExpr a, PrimExpr b)
take bitwise xor of two values
PrimExpr operator-(PrimExpr a, PrimExpr b)
subtraction operator
PrimExpr operator||(PrimExpr a, PrimExpr b)
or
PrimExpr operator>(PrimExpr a, PrimExpr b)
greater
PrimExpr operator+(PrimExpr a, PrimExpr b)
add operator
PrimExpr operator<<(PrimExpr a, PrimExpr b)
left shift operator
PrimExpr operator&&(PrimExpr a, PrimExpr b)
and
Definitions and helper macros for IR/AST nodes.
A managed object in the TVM runtime.
A map from source names to source code.
Runtime String container types.
ObjectRef equal functor.
Definition: object.h:665
ObjectRef hash functor.
Definition: object.h:655
static PrimExpr From(const TVMPODValue_ &val)
Definition: expr.h:776
static tvm::Bool From(const TVMPODValue_ &val)
Definition: expr.h:810
static tvm::Integer From(const TVMPODValue_ &val)
Definition: expr.h:797
Type trait to specify special value conversion rules from TVMArgValue and TVMRetValue.
Definition: packed_func.h:1104