24 #ifndef TVM_IR_ENV_FUNC_H_
25 #define TVM_IR_ENV_FUNC_H_
27 #include <tvm/ffi/function.h>
28 #include <tvm/ffi/reflection/registry.h>
55 refl::ObjectDef<EnvFuncNode>()
71 explicit EnvFunc(ObjectPtr<Object> n) : ObjectRef(n) {}
75 explicit EnvFunc(ffi::UnsafeInit tag) : ObjectRef(tag) {}
83 template <
typename... Args>
87 return n->
func(std::forward<Args>(args)...);
103 template <
typename FType>
115 template <
typename R,
typename... Args>
132 ObjectRef::operator=(other);
144 ICHECK(n !=
nullptr);
145 if constexpr (std::is_same_v<R, void>) {
146 n->
func(std::forward<Args>(args)...);
148 ffi::Any res = n->
func(std::forward<Args>(args)...);
149 if constexpr (std::is_same_v<R, ffi::Any>) {
152 return std::move(res).cast<R>();
A serializable function backed by TVM's global environment.
Definition: env_func.h:43
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind
Definition: env_func.h:60
ffi::Function func
The internal packed function.
Definition: env_func.h:48
EnvFuncNode()
constructor
Definition: env_func.h:50
static void RegisterReflection()
Definition: env_func.h:52
ffi::String name
Unique name of the global function.
Definition: env_func.h:46
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("ir.EnvFunc", EnvFuncNode, Object)
Managed reference to EnvFuncNode.
Definition: env_func.h:68
EnvFunc(ObjectPtr< Object > n)
Definition: env_func.h:71
EnvFunc(ffi::UnsafeInit tag)
constructor with UnsafeInit
Definition: env_func.h:75
static EnvFunc Get(const ffi::String &name)
Get a global function based on the name.
const EnvFuncNode * operator->() const
Definition: env_func.h:77
EnvFunc()
Definition: env_func.h:70
ffi::Any operator()(Args &&... args) const
Invoke the function.
Definition: env_func.h:84
A typed version of EnvFunc. It is backed by a GlobalFuncNode internally.
Definition: env_func.h:116
TypedEnvFunc(ObjectPtr< Object > n)
Definition: env_func.h:121
TypedEnvFunc()
Definition: env_func.h:120
TSelf & operator=(const EnvFunc &other)
Assign global function to a TypedEnvFunc.
Definition: env_func.h:131
const EnvFuncNode * operator->() const
Definition: env_func.h:136
R operator()(Args... args) const
Invoke the function.
Definition: env_func.h:142
TypedEnvFunc(ffi::UnsafeInit tag)
constructor with UnsafeInit
Definition: env_func.h:125
Please refer to TypedEnvFunc<R(Args..)>.
Definition: env_func.h:104
Definition: repr_printer.h:91
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37
Definitions and helper macros for IR/AST nodes.