An integration context that allows application of historically best records from a database. More...
#include <apply_history_best.h>
Inheritance diagram for tvm::meta_schedule::ApplyHistoryBestNode:
Collaboration diagram for tvm::meta_schedule::ApplyHistoryBestNode:
Public Types | |
| using | FTEFilterFunc = runtime::TypedPackedFunc< Optional< tir::PrimFunc >(const Array< te::Tensor, void > &)> |
| A callback function that filters TE compute. More... | |
| using | FTakeTuningRecord = runtime::TypedPackedFunc< void(const TuningRecord &)> |
| A callback function that takes a tuning record and does something with it. More... | |
 Public Types inherited from tvm::runtime::Object | |
| typedef void(* | FDeleter) (Object *self) |
| Object deleter. More... | |
| using | RefCounterType = std::atomic< int32_t > |
Public Member Functions | |
| void | VisitAttrs (tvm::AttrVisitor *v) |
| Optional< IRModule > | Query (runtime::String task_name, IRModule mod, Target target, Optional< Array< IRModule >> dispatched, FTakeTuningRecord f_take_tuning_record) |
| Query the best entry from the database. More... | |
| TVM_DECLARE_FINAL_OBJECT_INFO (ApplyHistoryBestNode, runtime::Object) | |
| Public Member Functions inherited from tvm::runtime::Object | |
| uint32_t | type_index () const |
| std::string | GetTypeKey () const |
| size_t | GetTypeKeyHash () const |
| template<typename TargetType > | |
| bool | IsInstance () const |
| bool | unique () const |
| Object () | |
| Object (const Object &other) | |
| Object (Object &&other) | |
| Object & | operator= (const Object &other) |
| Object & | operator= (Object &&other) |
Public Attributes | |
| Database | database {nullptr} |
| The database to be queried from. More... | |
| FTEFilterFunc | te_filter_func {nullptr} |
| The filtering function for TE computation. More... | |
| PackedFunc | logging_func |
| The logging function to be used. More... | |
Static Public Attributes | |
| static constexpr const char * | _type_key = "meta_schedule.ApplyHistoryBest" |
| Static Public Attributes inherited from tvm::runtime::Object | |
| static constexpr const char * | _type_key = "runtime.Object" |
| static constexpr bool | _type_final = false |
| static constexpr uint32_t | _type_child_slots = 0 |
| static constexpr bool | _type_child_slots_can_overflow = true |
| static constexpr bool | _type_has_method_visit_attrs = true |
| static constexpr bool | _type_has_method_sequal_reduce = false |
| static constexpr bool | _type_has_method_shash_reduce = false |
| static constexpr uint32_t | _type_index = TypeIndex::kDynamic |
Additional Inherited Members | |
| Static Public Member Functions inherited from tvm::runtime::Object | |
| static std::string | TypeIndex2Key (uint32_t tindex) |
| Get the type key of the corresponding index from runtime. More... | |
| static size_t | TypeIndex2KeyHash (uint32_t tindex) |
| Get the type key hash of the corresponding index from runtime. More... | |
| static uint32_t | TypeKey2Index (const std::string &key) |
| Get the type index of the corresponding key from runtime. More... | |
| static uint32_t | _GetOrAllocRuntimeTypeIndex () |
| static uint32_t | RuntimeTypeIndex () |
| Protected Member Functions inherited from tvm::runtime::Object | |
| void | IncRef () |
| developer function, increases reference counter. More... | |
| void | DecRef () |
| developer function, decrease reference counter. More... | |
| Static Protected Member Functions inherited from tvm::runtime::Object | |
| static uint32_t | GetOrAllocRuntimeTypeIndex (const std::string &key, uint32_t static_tindex, uint32_t parent_tindex, uint32_t type_child_slots, bool type_child_slots_can_overflow) |
| Get the type index using type key. More... | |
| Protected Attributes inherited from tvm::runtime::Object | |
| uint32_t | type_index_ {0} |
| Type index(tag) that indicates the type of the object. More... | |
| RefCounterType | ref_counter_ {0} |
| The internal reference counter. More... | |
| FDeleter | deleter_ = nullptr |
| deleter of this object to enable customized allocation. If the deleter is nullptr, no deletion will be performed. The creator of the object must always set the deleter field properly. More... | |
Detailed Description
An integration context that allows application of historically best records from a database.
Member Typedef Documentation
◆ FTakeTuningRecord
| using tvm::meta_schedule::ApplyHistoryBestNode::FTakeTuningRecord = runtime::TypedPackedFunc<void(const TuningRecord&)> |
A callback function that takes a tuning record and does something with it.
◆ FTEFilterFunc
| using tvm::meta_schedule::ApplyHistoryBestNode::FTEFilterFunc = runtime::TypedPackedFunc<Optional<tir::PrimFunc>(const Array<te::Tensor, void>&)> |
A callback function that filters TE compute.
Member Function Documentation
◆ Query()
| Optional<IRModule> tvm::meta_schedule::ApplyHistoryBestNode::Query | ( | runtime::String | task_name, |
| IRModule | mod, | ||
| Target | target, | ||
| Optional< Array< IRModule >> | dispatched, | ||
| FTakeTuningRecord | f_take_tuning_record | ||
| ) |
Query the best entry from the database.
- Parameters
-
task_name The name of the task to be queried mod The module to be queried target The target to be queried dispatched The IRs after dispatch f_take_tuning_record A callback function that takes a tuning record and does something with it
◆ TVM_DECLARE_FINAL_OBJECT_INFO()
| tvm::meta_schedule::ApplyHistoryBestNode::TVM_DECLARE_FINAL_OBJECT_INFO | ( | ApplyHistoryBestNode | , |
| runtime::Object | |||
| ) |
◆ VisitAttrs()
|
inline |
Member Data Documentation
◆ _type_key
|
static |
◆ database
| Database tvm::meta_schedule::ApplyHistoryBestNode::database {nullptr} |
The database to be queried from.
◆ logging_func
| PackedFunc tvm::meta_schedule::ApplyHistoryBestNode::logging_func |
The logging function to be used.
◆ te_filter_func
| FTEFilterFunc tvm::meta_schedule::ApplyHistoryBestNode::te_filter_func {nullptr} |
The filtering function for TE computation.
The documentation for this class was generated from the following file:
- include/tvm/meta_schedule/apply_history_best.h
