tvm::meta_schedule::PyMutatorNode Class Reference

The mutator with customized methods on the python-side. More...

#include <mutator.h>

Inheritance diagram for tvm::meta_schedule::PyMutatorNode:

Collaboration diagram for tvm::meta_schedule::PyMutatorNode:

Public Types
using	FInitializeWithTuneContext = Mutator::FInitializeWithTuneContext
using	FApply = Mutator::FApply
using	FClone = Mutator::FClone
using	FAsString = Mutator::FAsString

Public Member Functions
void	InitializeWithTuneContext (const TuneContext &context) final
	Initialize the design space generator with tuning context. More...
Optional< tir::Trace >	Apply (const tir::Trace &trace, support::LinearCongruentialEngine::TRandState *rand_state) final
	Apply the mutator function to the given trace. More...
Mutator	Clone () const final
	Clone the mutator. More...
	TVM_DECLARE_FINAL_OBJECT_INFO (PyMutatorNode, MutatorNode)
Public Member Functions inherited from tvm::meta_schedule::MutatorNode
virtual	~MutatorNode ()=default
	Virtual destructor. More...
	TVM_DECLARE_BASE_OBJECT_INFO (MutatorNode, Object)

Static Public Member Functions
static void	RegisterReflection ()
Static Public Member Functions inherited from tvm::meta_schedule::MutatorNode
static void	RegisterReflection ()

Public Attributes
FInitializeWithTuneContext	f_initialize_with_tune_context
	The packed function to the InitializeWithTuneContext function. More...
FApply	f_apply
	The packed function to the Apply function. More...
FClone	f_clone
	The packed function to the Clone function. More...
FAsString	f_as_string
	The packed function to the AsString function. More...

Static Public Attributes
static constexpr const char *	_type_key = "meta_schedule.PyMutator"
Static Public Attributes inherited from tvm::meta_schedule::MutatorNode
static constexpr const char *	_type_key = "meta_schedule.Mutator"

Detailed Description

The mutator with customized methods on the python-side.

Member Typedef Documentation

FApply

using tvm::meta_schedule::PyMutatorNode::FApply = Mutator::FApply

FAsString

using tvm::meta_schedule::PyMutatorNode::FAsString = Mutator::FAsString

FClone

using tvm::meta_schedule::PyMutatorNode::FClone = Mutator::FClone

FInitializeWithTuneContext

using tvm::meta_schedule::PyMutatorNode::FInitializeWithTuneContext = Mutator::FInitializeWithTuneContext

Member Function Documentation

Apply()

Optional<tir::Trace> tvm::meta_schedule::PyMutatorNode::Apply ( const tir::Trace &  trace, support::LinearCongruentialEngine::TRandState *  rand_state  )

finalvirtual

Apply the mutator function to the given trace.

Parameters

trace	The given trace for mutation.
rand_state	The random state for mutation.

Returns

None if mutator failed, otherwise return the mutated trace.

Implements tvm::meta_schedule::MutatorNode.

Clone()

Mutator tvm::meta_schedule::PyMutatorNode::Clone ( ) const

finalvirtual

Clone the mutator.

Returns

The cloned mutator.

Implements tvm::meta_schedule::MutatorNode.

InitializeWithTuneContext()

void tvm::meta_schedule::PyMutatorNode::InitializeWithTuneContext ( const TuneContext &  context )

finalvirtual

Initialize the design space generator with tuning context.

Parameters

context

The tuning context for initialization.

Note

This method is supposed to be called only once before every other method.

Implements tvm::meta_schedule::MutatorNode.

RegisterReflection()

static void tvm::meta_schedule::PyMutatorNode::RegisterReflection ( )

inlinestatic

TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::meta_schedule::PyMutatorNode::TVM_DECLARE_FINAL_OBJECT_INFO	(	PyMutatorNode	,
		MutatorNode
	)

Member Data Documentation

_type_key

constexpr const char* tvm::meta_schedule::PyMutatorNode::_type_key = "meta_schedule.PyMutator"

staticconstexpr

f_apply

FApply tvm::meta_schedule::PyMutatorNode::f_apply

The packed function to the Apply function.

f_as_string

FAsString tvm::meta_schedule::PyMutatorNode::f_as_string

The packed function to the AsString function.

f_clone

FClone tvm::meta_schedule::PyMutatorNode::f_clone

The packed function to the Clone function.

f_initialize_with_tune_context

FInitializeWithTuneContext tvm::meta_schedule::PyMutatorNode::f_initialize_with_tune_context

The packed function to the InitializeWithTuneContext function.

---

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

• include/tvm/meta_schedule/mutator.h