28 #ifndef TVM_IR_ATTRS_H_
29 #define TVM_IR_ATTRS_H_
31 #include <dmlc/common.h>
32 #include <tvm/ffi/container/map.h>
33 #include <tvm/ffi/function.h>
34 #include <tvm/ffi/reflection/accessor.h>
35 #include <tvm/ffi/reflection/registry.h>
42 #include <type_traits>
43 #include <unordered_map>
54 template <
typename TObjectRef>
56 static_assert(TObjectRef::_type_is_nullable,
"Can only get NullValue for nullable types");
57 return TObjectRef(ObjectPtr<Object>(
nullptr));
78 namespace rfl = ffi::reflection;
79 rfl::ObjectDef<AttrFieldInfoNode>()
85 static constexpr
const char*
_type_key =
"ir.AttrFieldInfo";
112 template <
typename... Args>
122 bool allow_unknown =
false) = 0;
151 namespace rfl = ffi::reflection;
158 static constexpr
const char*
_type_key =
"ir.DictAttrs";
196 template <
typename TObjectRef>
198 const std::string& attr_key,
199 Optional<TObjectRef> default_value = Optional<TObjectRef>(std::nullopt))
const {
200 if (!defined())
return default_value;
202 auto it = node->
dict.find(attr_key);
203 if (it != node->
dict.end()) {
204 return (*it).second.cast<TObjectRef>();
206 return default_value;
210 template <
typename TObjectRef>
211 Optional<TObjectRef>
GetAttr(
const std::string& attr_key, TObjectRef default_value)
const {
212 return GetAttr<TObjectRef>(attr_key, Optional<TObjectRef>(default_value));
234 return GetAttr<Integer>(attr_key, 0).value_or(0).IntValue() != 0;
267 return WithAttr(std::move(attrs), String(key), std::move(value));
308 template <
typename TFunc>
309 inline TFunc
WithAttr(TFunc input,
const std::string& attr_key, Any attr_value) {
310 using TNode =
typename TFunc::ContainerType;
311 static_assert(TNode::_type_final,
"Can only operate on the leaf nodes");
312 TNode* node = input.CopyOnWrite();
313 node->attrs =
WithAttr(std::move(node->attrs), attr_key, attr_value);
327 template <
typename TFunc>
328 inline TFunc
WithAttrs(TFunc input, Map<String, Any> attrs) {
329 using TNode =
typename TFunc::ContainerType;
330 static_assert(TNode::_type_final,
"Can only operate on the leaf nodes");
331 TNode* node = input.CopyOnWrite();
333 node->attrs =
WithAttrs(std::move(node->attrs), attrs);
364 template <
typename TFunc>
365 inline TFunc
WithoutAttr(TFunc input,
const std::string& attr_key) {
366 using TNode =
typename TFunc::ContainerType;
367 static_assert(TNode::_type_final,
"Can only operate on the leaf nodes");
369 TNode* node = input.CopyOnWrite();
370 node->attrs =
WithoutAttr(std::move(node->attrs), attr_key);
383 template <
typename DerivedType>
387 LOG(FATAL) <<
"`" << DerivedType::_type_key <<
"` uses new reflection mechanism for init";
391 DerivedType*
self()
const {
392 return const_cast<DerivedType*
>(
static_cast<const DerivedType*
>(
this));
401 template <
typename TAttrs>
403 static_assert(std::is_base_of_v<Attrs, TAttrs>,
"Can only take attr nodes");
404 using ContainerType =
typename TAttrs::ContainerType;
406 static auto finit_object = ffi::Function::GetGlobalRequired(
"ffi.MakeObjectFromPackedArgs");
407 AnyView packed_args[1];
408 packed_args[0] = ContainerType::RuntimeTypeIndex();
410 finit_object.CallPacked(ffi::PackedArgs(packed_args, 1), &rv);
411 return rv.cast<TAttrs>();
413 auto n = make_object<ContainerType>();
414 n->InitByPackedArgs(ffi::PackedArgs(
nullptr, 0),
false);
Information about attribute fields in string representations.
Definition: attrs.h:68
static void RegisterReflection()
Definition: attrs.h:77
String description
detailed description of the type
Definition: attrs.h:75
TVM_DECLARE_FINAL_OBJECT_INFO(AttrFieldInfoNode, Object)
String type_info
type docstring information in str.
Definition: attrs.h:73
String name
name of the field
Definition: attrs.h:71
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind
Definition: attrs.h:86
static constexpr const char * _type_key
Definition: attrs.h:85
AttrFieldInfo.
Definition: attrs.h:92
TVM_DEFINE_OBJECT_REF_METHODS(AttrFieldInfo, ObjectRef, AttrFieldInfoNode)
Adapter for AttrsNode with the new reflection API.
Definition: attrs.h:384
void InitByPackedArgs(const ffi::PackedArgs &args, bool allow_unknown) final
Initialize the attributes by arguments.
Definition: attrs.h:386
Managed reference to BaseAttrsNode.
Definition: attrs.h:134
TVM_DEFINE_OBJECT_REF_METHODS(Attrs, ObjectRef, BaseAttrsNode)
Base class of all attribute class.
Definition: attrs.h:103
virtual ~BaseAttrsNode()
virtual destructor
Definition: attrs.h:106
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind
Definition: attrs.h:124
static constexpr const char * _type_key
Definition: attrs.h:126
TVM_DECLARE_BASE_OBJECT_INFO(BaseAttrsNode, Object)
void InitBySeq(Args &&... args)
Initialize the attributes by sequence of arguments.
virtual void InitByPackedArgs(const ffi::PackedArgs &kwargs, bool allow_unknown=false)=0
Initialize the attributes by arguments.
Specialized attribute type that is backed by a map. The DictAttrsNode implements the Attrs behavior,...
Definition: attrs.h:145
TVM_DECLARE_FINAL_OBJECT_INFO(DictAttrsNode, BaseAttrsNode)
static void RegisterReflection()
Definition: attrs.h:150
Map< String, ffi::Any > dict
internal attrs map
Definition: attrs.h:148
void InitByPackedArgs(const ffi::PackedArgs &args, bool allow_unknown) final
Initialize the attributes by arguments.
static constexpr const char * _type_key
Definition: attrs.h:158
Managed reference to DictAttrsNode.
Definition: attrs.h:166
DictAttrs(Map< String, Any > dict={})
Consruct a Attrs backed by DictAttrsNode.
bool HasNonzeroAttr(const std::string &attr_key) const
Check whether the function has an non-zero integer attr.
Definition: attrs.h:233
Optional< TObjectRef > GetAttr(const std::string &attr_key, Optional< TObjectRef > default_value=Optional< TObjectRef >(std::nullopt)) const
Get a function attribute.
Definition: attrs.h:197
TVM_DEFINE_OBJECT_REF_COW_METHOD(DictAttrsNode)
Optional< TObjectRef > GetAttr(const std::string &attr_key, TObjectRef default_value) const
Definition: attrs.h:211
TVM_DEFINE_OBJECT_REF_METHODS_WITHOUT_DEFAULT_CONSTRUCTOR(DictAttrs, Attrs, DictAttrsNode)
Runtime primitive data type.
Definition: data_type.h:47
@ kHandle
Definition: data_type.h:61
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37
DictAttrs WithoutAttr(DictAttrs attrs, const std::string &key)
Copy the DictAttrs, but without a specific attribute.
DataType NullValue< DataType >()
Definition: attrs.h:61
TAttrs AttrsWithDefaultValues()
Create an Attr object with all default values.
Definition: attrs.h:402
DictAttrs WithAttr(DictAttrs attrs, String key, Any value)
Copy the DictAttrs, but overrides a single attribute.
runtime::DataType DataType
Definition: data_type.h:458
DictAttrs WithAttrs(DictAttrs attrs, Map< String, Any > new_attrs)
Copy the DictAttrs, but overrides attributes with the entries from attrs.
TObjectRef NullValue()
Create a NodeRef type that represents null.
Definition: attrs.h:55
Structural equality comparison.