target_kind.h File Reference

Target kind registry. More...

#include <tvm/ffi/function.h>
#include <tvm/ffi/reflection/registry.h>
#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 >
	ffi::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
	Performance counters for profiling via the PAPI library.
	tvm::detail

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 = ffi::Map< ffi::String, ffi::Any >
	Map containing parsed features of a specific Target. More...
using	tvm::TargetJSON = ffi::Map< ffi::String, ffi::Any >
	TargetParser to apply on instantiation of a given TargetKind. More...
using	tvm::FTVMTargetParser = ffi::TypedFunction< TargetJSON(TargetJSON)>

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<ffi::Array<ffi::String>>("keys")       \
          .add_attr_option<ffi::String>("tag")                    \
          .add_attr_option<ffi::String>("device")                 \
          .add_attr_option<ffi::String>("model")                  \
          .add_attr_option<ffi::Array<ffi::String>>("libs")       \
          .add_attr_option<Target>("host")                        \
          .add_attr_option<int64_t>("from_device")                \
          .add_attr_option<int64_t>("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<ffi::String>("mtriple")
.add_attr_option<ffi::String>("mattr");

TVM_TARGET_KIND_REGISTER_VAR_DEF

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