target_kind.h File Reference

Target kind registry. More...

#include <tvm/node/attr_registry_map.h>
#include <tvm/node/node.h>
#include <memory>
#include <unordered_map>
#include <utility>
#include <vector>

Include dependency graph for target_kind.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	tvm::TargetKindNode
	Target kind, specifies the kind of the target. More...
class	tvm::TargetKind
	Managed reference class to TargetKindNode. More...
class	tvm::TargetKindAttrMap< ValueType >
	Map<TargetKind, ValueType> used to store meta-information about TargetKind. More...
class	tvm::TargetKindRegEntry
	Helper structure to register TargetKind. More...
struct	tvm::detail::is_specialized< Type, Container >
struct	tvm::detail::is_specialized< Container< Args... >, Container >
struct	tvm::detail::ValueTypeInfoMaker< ValueType, IsArray, IsMap >

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::detail
	tvm::attr
	Generic attribute names that can be attached to any function.

Macros
#define	TVM_TARGET_KIND_REGISTER_VAR_DEF    static DMLC_ATTRIBUTE_UNUSED ::tvm::TargetKindRegEntry& __make_##TargetKind
#define	TVM_REGISTER_TARGET_KIND(TargetKindName, DeviceType)
	Register a new target kind, or set attribute of the corresponding target kind. More...

Typedefs
using	tvm::TargetFeatures = Map< String, ObjectRef >
	Map containing parsed features of a specific Target. More...
using	tvm::TargetJSON = Map< String, ObjectRef >
	TargetParser to apply on instantiation of a given TargetKind. More...
using	tvm::FTVMTargetParser = runtime::TypedPackedFunc< TargetJSON(TargetJSON)>

Variables
constexpr const char *	tvm::attr::kIsExternalCodegen = "is_external_codegen"
	A TargetKind attribute of type Bool. If true, then the target kind name also corresponds to an external codegen 'compiler' name. That name may be used: More...
constexpr const char *	tvm::attr::kRelayToTIR = "RelayToTIR"
	A TargetKind attribute of type FTVMRelayToTIR. If set, then the target kind name also corresponds to an external codegen 'compiler' name, and the bound value is a Pass to apply before the TVM lowering. More...

Detailed Description

Target kind registry.

Macro Definition Documentation

TVM_REGISTER_TARGET_KIND

#define TVM_REGISTER_TARGET_KIND	(		TargetKindName,
			DeviceType
	)

Value:

  TVM_STR_CONCAT(TVM_TARGET_KIND_REGISTER_VAR_DEF, __COUNTER__) = \
      ::tvm::TargetKindRegEntry::RegisterOrGet(TargetKindName)    \
          .set_name()                                             \
          .set_default_device_type(DeviceType)                    \
          .add_attr_option<Array<String>>("keys")                 \
          .add_attr_option<String>("tag")                         \
          .add_attr_option<String>("device")                      \
          .add_attr_option<String>("model")                       \
          .add_attr_option<Array<String>>("libs")                 \
          .add_attr_option<Target>("host")                        \
          .add_attr_option<runtime::Int>("from_device")           \
          .add_attr_option<runtime::Int>("target_device_type")

Register a new target kind, or set attribute of the corresponding target kind.

Parameters

TargetKindName	The name of target kind
DeviceType	The DLDeviceType of the target kind

TVM_REGISTER_TARGET_KIND("llvm")
.set_attr<TPreCodegenPass>("TPreCodegenPass", a-pre-codegen-pass)
.add_attr_option<Bool>("system_lib")
.add_attr_option<String>("mtriple")
.add_attr_option<String>("mattr");

TVM_TARGET_KIND_REGISTER_VAR_DEF

#define TVM_TARGET_KIND_REGISTER_VAR_DEF    static DMLC_ATTRIBUTE_UNUSED ::tvm::TargetKindRegEntry& __make_##TargetKind