target_kind.h File Reference

Target kind registry. More...

#include <tvm/ffi/function.h>
#include <tvm/ffi/reflection/registry.h>
#include <tvm/ir/attr_registry_map.h>
#include <tvm/ir/config_schema.h>
#include <tvm/runtime/base.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...

Namespaces
namespace	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.

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

Typedefs
using	tvm::FTargetCanonicalizer = ffi::TypedFunction< ffi::Map< ffi::String, ffi::Any >(ffi::Map< ffi::String, ffi::Any >)>
	Target canonicalizer applied on instantiation of a given TargetKind.

Detailed Description

Target kind registry.

Macro Definition Documentation

TVM_REGISTER_TARGET_KIND

#define TVM_REGISTER_TARGET_KIND	(		TargetKindName,
			DeviceType
	)

Value:

  TVM_FFI_STR_CONCAT(TVM_TARGET_KIND_REGISTER_VAR_DEF, __COUNTER__) = \
      ::tvm::TargetKindRegEntry::RegisterOrGet(TargetKindName)        \
          .set_name()                                                 \
          .set_default_device_type(DeviceType)                        \
          .add_attr_option<ffi::String>("kind")                       \
          .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    [[maybe_unused]] static ::tvm::TargetKindRegEntry& __make_##TargetKind