tvm
|
#include <tvm/ffi/reflection/registry.h>
#include <utility>
Go to the source code of this file.
Classes | |
class | tvm::tir::InstructionKindNode |
Kind of an instruction, e.g. Split, Reorder, etc. Besides the name, every kind of instruction has its own properties, including: 1) A boolean indicating if the instruction is pure, i.e. change nothing in the schedule state 2) A functor that applies the instruction to a TensorIR schedule 3) A functor that converts the instruction to a statement in python syntax 4) A functor that serialize its attributes to JSON 5) A functor that deserialize its attributes from JSON. More... | |
class | tvm::tir::InstructionKind |
Managed reference to InstructionKindNode. More... | |
class | tvm::tir::InstructionNode |
Schedule instructions each corresponds to a schedule primitive. More... | |
class | tvm::tir::Instruction |
Managed reference to InstructionNode. More... | |
class | tvm::tir::InstructionKindRegEntry |
An entry in the registry of InstructionKind. More... | |
Namespaces | |
tvm | |
Performance counters for profiling via the PAPI library. | |
tvm::tir | |
Macros | |
#define | TVM_INST_KIND_REGISTER_VAR_DEF static DMLC_ATTRIBUTE_UNUSED ::tvm::tir::InstructionKindRegEntry& __make_##InstructionKind |
A helper macro to register InstructionKind, only used in TVM_REGISTER_INST_KIND More... | |
#define | TVM_REGISTER_INST_KIND(InstructionKindName) |
Register an InstructionKind. More... | |
Typedefs | |
using | tvm::tir::FInstructionApply = ffi::TypedFunction< Array< Any >(Schedule sch, const Array< Any > &inputs, const Array< Any > &attrs, const Any &decision)> |
Type of the functor that applies the instruction to a TensorIR schedule. More... | |
using | tvm::tir::FInstructionAsPython = ffi::TypedFunction< String(const Array< Any > &inputs, const Array< Any > &attrs, const Any &decision, const Array< String > &outputs)> |
Type of the functor that converts the instruction to a statement in python syntax. More... | |
using | tvm::tir::FInstructionAttrsAsJSON = ffi::TypedFunction< ObjectRef(Array< Any > attrs)> |
Type of the functor that serialize its attributes to JSON. More... | |
using | tvm::tir::FInstructionAttrsFromJSON = ffi::TypedFunction< Array< Any >(ObjectRef json_attrs)> |
Type of the functor that deserialize its attributes from JSON. More... | |
#define TVM_INST_KIND_REGISTER_VAR_DEF static DMLC_ATTRIBUTE_UNUSED ::tvm::tir::InstructionKindRegEntry& __make_##InstructionKind |
A helper macro to register InstructionKind, only used in TVM_REGISTER_INST_KIND
#define TVM_REGISTER_INST_KIND | ( | InstructionKindName | ) |
Register an InstructionKind.
InstructionKindName | The name of the InstructionKind |
Example: