tvm::s_tir::meta_schedule::TuneContextObj Class Reference

The auto tuning context. More...

#include <tune_context.h>

Inheritance diagram for tvm::s_tir::meta_schedule::TuneContextObj:

Collaboration diagram for tvm::s_tir::meta_schedule::TuneContextObj:

Public Types
using	TRandState = LinearCongruentialEngine::TRandState

Public Member Functions
void	Initialize ()
	Initialize members that needs initialization with tune context.
TuneContext	Clone () const
	Clone the tune context.
	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("s_tir.meta_schedule.TuneContext", TuneContextObj, ffi::Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
ffi::Optional< IRModule >	mod
	The workload to be tuned.
ffi::Optional< Target >	target
	The target to be tuned for.
ffi::Optional< SpaceGenerator >	space_generator
	The design space generator.
ffi::Optional< SearchStrategy >	search_strategy
	The search strategy.
ffi::Optional< ffi::String >	task_name
	The name of the tuning task.
int	num_threads
	The number of threads to be used.
TRandState	rand_state
	The random state.
ffi::Function	logger
	The tuning task's logging function. t.

Static Public Attributes
static constexpr const bool	_type_mutable = true

Detailed Description

The auto tuning context.

Member Typedef Documentation

TRandState

using tvm::s_tir::meta_schedule::TuneContextObj::TRandState = LinearCongruentialEngine::TRandState

Member Function Documentation

Clone()

TuneContext tvm::s_tir::meta_schedule::TuneContextObj::Clone

(

)

const

Clone the tune context.

Returns

The cloned tune context.

Initialize()

void tvm::s_tir::meta_schedule::TuneContextObj::Initialize

(

)

Initialize members that needs initialization with tune context.

RegisterReflection()

static void tvm::s_tir::meta_schedule::TuneContextObj::RegisterReflection ( )

inlinestatic

TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::s_tir::meta_schedule::TuneContextObj::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"s_tir.meta_schedule.TuneContext"	,
		TuneContextObj	,
		ffi::Object
	)

Member Data Documentation

_type_mutable

constexpr const bool tvm::s_tir::meta_schedule::TuneContextObj::_type_mutable = true

staticconstexpr

logger

ffi::Function tvm::s_tir::meta_schedule::TuneContextObj::logger

The tuning task's logging function. t.

mod

ffi::Optional<IRModule> tvm::s_tir::meta_schedule::TuneContextObj::mod

The workload to be tuned.

num_threads

int tvm::s_tir::meta_schedule::TuneContextObj::num_threads

The number of threads to be used.

rand_state

TRandState tvm::s_tir::meta_schedule::TuneContextObj::rand_state

The random state.

search_strategy

ffi::Optional<SearchStrategy> tvm::s_tir::meta_schedule::TuneContextObj::search_strategy

The search strategy.

space_generator

ffi::Optional<SpaceGenerator> tvm::s_tir::meta_schedule::TuneContextObj::space_generator

The design space generator.

target

ffi::Optional<Target> tvm::s_tir::meta_schedule::TuneContextObj::target

The target to be tuned for.

task_name

ffi::Optional<ffi::String> tvm::s_tir::meta_schedule::TuneContextObj::task_name

The name of the tuning task.

---

The documentation for this class was generated from the following file:

• include/tvm/s_tir/meta_schedule/tune_context.h