tvm
|
Static analyzer for a ComputeDAG. More...
#include <compute_dag.h>
Public Types | |
template<class T > | |
using | OperationMap = std::unordered_map< te::Operation, T, ObjectPtrHash, ObjectPtrEqual > |
Public Types inherited from tvm::runtime::Object | |
typedef void(* | FDeleter) (Object *self) |
Object deleter. More... | |
using | RefCounterType = std::atomic< int32_t > |
Public Member Functions | |
TVM_DECLARE_FINAL_OBJECT_INFO (AccessAnalyzerNode, 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 | |
OperationMap< OperationMap< std::vector< std::vector< PrimExpr > > > > | read_from |
Map an operation to all operations it reads from. For each operation pair, use a two-dimensional array for multiple multi-dimensional accesses The inner vector represents the indices of multi-dimensional access. More... | |
OperationMap< OperationMap< std::vector< std::vector< PrimExpr > > > > | read_by |
Map an operation to all operations it is read by. For each operation pair, use a two-dimensional array for multiple multi-dimensional accesses The inner vector represents the indices of multi-dimensional access. More... | |
OperationMap< OperationMap< int > > | num_common_outer_iterators |
Store the number of common outer iterators for operation pairs that have read-write relations. More... | |
OperationMap< bool > | is_simple_access |
Store whether the operation is an op with only simple access. (e.g., injective, broadcast and elementwise ops without reduction) More... | |
OperationMap< bool > | is_strictly_inlineable |
Store whether the operation is strictly inlineable (e.g., injective, broadcast and elementwise without reduction, branch or expensive operations) More... | |
OperationMap< bool > | needs_multi_level_tiling |
Store whether the operation needs multi-level tiling (e.g., computation-intensive ops with data reuse opportunity like matmul, conv2d) More... | |
OperationMap< bool > | is_output |
Store whether the operation is an output operation. More... | |
Array< te::Operation > | ops_topo_order |
Store the topological order of operations. More... | |
Static Public Attributes | |
static constexpr const char * | _type_key = "auto_scheduler.AccessAnalyzer" |
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... | |
Static analyzer for a ComputeDAG.
using tvm::auto_scheduler::AccessAnalyzerNode::OperationMap = std::unordered_map<te::Operation, T, ObjectPtrHash, ObjectPtrEqual> |
tvm::auto_scheduler::AccessAnalyzerNode::TVM_DECLARE_FINAL_OBJECT_INFO | ( | AccessAnalyzerNode | , |
Object | |||
) |
|
staticconstexpr |
OperationMap<bool> tvm::auto_scheduler::AccessAnalyzerNode::is_output |
Store whether the operation is an output operation.
OperationMap<bool> tvm::auto_scheduler::AccessAnalyzerNode::is_simple_access |
Store whether the operation is an op with only simple access. (e.g., injective, broadcast and elementwise ops without reduction)
OperationMap<bool> tvm::auto_scheduler::AccessAnalyzerNode::is_strictly_inlineable |
Store whether the operation is strictly inlineable (e.g., injective, broadcast and elementwise without reduction, branch or expensive operations)
OperationMap<bool> tvm::auto_scheduler::AccessAnalyzerNode::needs_multi_level_tiling |
Store whether the operation needs multi-level tiling (e.g., computation-intensive ops with data reuse opportunity like matmul, conv2d)
OperationMap<OperationMap<int> > tvm::auto_scheduler::AccessAnalyzerNode::num_common_outer_iterators |
Store the number of common outer iterators for operation pairs that have read-write relations.
Array<te::Operation> tvm::auto_scheduler::AccessAnalyzerNode::ops_topo_order |
Store the topological order of operations.
OperationMap<OperationMap<std::vector<std::vector<PrimExpr> > > > tvm::auto_scheduler::AccessAnalyzerNode::read_by |
Map an operation to all operations it is read by. For each operation pair, use a two-dimensional array for multiple multi-dimensional accesses The inner vector represents the indices of multi-dimensional access.
OperationMap<OperationMap<std::vector<std::vector<PrimExpr> > > > tvm::auto_scheduler::AccessAnalyzerNode::read_from |
Map an operation to all operations it reads from. For each operation pair, use a two-dimensional array for multiple multi-dimensional accesses The inner vector represents the indices of multi-dimensional access.