executor.h File Reference

Object representation of Executor configuration and registry. More...

#include <dmlc/registry.h>
#include <tvm/ir/attrs.h>
#include <tvm/ir/expr.h>
#include <tvm/ir/type.h>
#include <tvm/ir/type_relation.h>
#include <tvm/node/attr_registry_map.h>
#include <tvm/runtime/registry.h>
#include <string>
#include <unordered_map>
#include <utility>
#include <vector>

Include dependency graph for executor.h:

Go to the source code of this file.

Classes
class	tvm::AttrRegistry< typename, typename >
class	tvm::relay::ExecutorNode
	Executor information. More...
class	tvm::relay::Executor
	Managed reference class to ExecutorNode. More...
class	tvm::relay::ExecutorRegEntry
	Helper structure to register Executors. More...

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::relay
	Relay: a high level functional IR for TVM.

Macros
#define	TVM_EXECUTOR_REGISTER_VAR_DEF   static DMLC_ATTRIBUTE_UNUSED ::tvm::relay::ExecutorRegEntry& __make_##Executor
#define	TVM_REGISTER_EXECUTOR(ExecutorName)
	Register a new executor, or set attribute of the corresponding executor. More...

Detailed Description

Object representation of Executor configuration and registry.

Macro Definition Documentation

TVM_EXECUTOR_REGISTER_VAR_DEF

#define TVM_EXECUTOR_REGISTER_VAR_DEF   static DMLC_ATTRIBUTE_UNUSED ::tvm::relay::ExecutorRegEntry& __make_##Executor

TVM_REGISTER_EXECUTOR

#define TVM_REGISTER_EXECUTOR

(

ExecutorName

)

Value:

TVM_STR_CONCAT(TVM_EXECUTOR_REGISTER_VAR_DEF, __COUNTER__) = \
      ::tvm::relay::ExecutorRegEntry::RegisterOrGet(ExecutorName)

Register a new executor, or set attribute of the corresponding executor.

Parameters

ExecutorName

The name of registry

TVM_REGISTER_EXECUTOR("aot")
.add_attr_option<String>("my_option");
.add_attr_option<String>("my_option_default", String("default"));