tvm::meta_schedule::PyScheduleRuleNode Class Reference

The schedule rule with customized methods on the python-side. More...

#include <schedule_rule.h>

Inheritance diagram for tvm::meta_schedule::PyScheduleRuleNode:

Collaboration diagram for tvm::meta_schedule::PyScheduleRuleNode:

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

Public Member Functions
void	InitializeWithTuneContext (const TuneContext &context) final
	Initialize the design space generator with tuning context. More...
Array< tir::Schedule >	Apply (const tir::Schedule &sch, const tir::BlockRV &block) final
	Apply a schedule rule to the specific block in the given schedule. More...
ScheduleRule	Clone () const final
	Deep clone the schedule rule. More...
	TVM_DECLARE_FINAL_OBJECT_INFO (PyScheduleRuleNode, ScheduleRuleNode)
Public Member Functions inherited from tvm::meta_schedule::ScheduleRuleNode
virtual	~ScheduleRuleNode ()=default
	Virtual destructor. More...
	TVM_DECLARE_BASE_OBJECT_INFO (ScheduleRuleNode, Object)

Static Public Member Functions
static void	RegisterReflection ()
Static Public Member Functions inherited from tvm::meta_schedule::ScheduleRuleNode
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...
FAsString	f_as_string
	The packed function to the AsString function. More...
FClone	f_clone
	The packed function to the Clone function. More...

Static Public Attributes
static constexpr const char *	_type_key = "meta_schedule.PyScheduleRule"
Static Public Attributes inherited from tvm::meta_schedule::ScheduleRuleNode
static constexpr const char *	_type_key = "meta_schedule.ScheduleRule"

Detailed Description

The schedule rule with customized methods on the python-side.

Member Typedef Documentation

FApply

using tvm::meta_schedule::PyScheduleRuleNode::FApply = ScheduleRule::FApply

FAsString

using tvm::meta_schedule::PyScheduleRuleNode::FAsString = ScheduleRule::FAsString

FClone

using tvm::meta_schedule::PyScheduleRuleNode::FClone = ScheduleRule::FClone

FInitializeWithTuneContext

using tvm::meta_schedule::PyScheduleRuleNode::FInitializeWithTuneContext = ScheduleRule::FInitializeWithTuneContext

Member Function Documentation

Apply()

Array<tir::Schedule> tvm::meta_schedule::PyScheduleRuleNode::Apply ( const tir::Schedule &  sch, const tir::BlockRV &  block  )

finalvirtual

Apply a schedule rule to the specific block in the given schedule.

Parameters

sch	The schedule to be modified.
block	The specific block to apply the schedule rule.

Returns

The list of schedules generated by applying the schedule rule.

Implements tvm::meta_schedule::ScheduleRuleNode.

Clone()

ScheduleRule tvm::meta_schedule::PyScheduleRuleNode::Clone ( ) const

finalvirtual

Deep clone the schedule rule.

Returns

The cloned schedule rule.

Implements tvm::meta_schedule::ScheduleRuleNode.

InitializeWithTuneContext()

void tvm::meta_schedule::PyScheduleRuleNode::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::ScheduleRuleNode.

RegisterReflection()

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

inlinestatic

TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::meta_schedule::PyScheduleRuleNode::TVM_DECLARE_FINAL_OBJECT_INFO	(	PyScheduleRuleNode	,
		ScheduleRuleNode
	)

Member Data Documentation

_type_key

constexpr const char* tvm::meta_schedule::PyScheduleRuleNode::_type_key = "meta_schedule.PyScheduleRule"

staticconstexpr

f_apply

FApply tvm::meta_schedule::PyScheduleRuleNode::f_apply

The packed function to the Apply function.

f_as_string

FAsString tvm::meta_schedule::PyScheduleRuleNode::f_as_string

The packed function to the AsString function.

f_clone

FClone tvm::meta_schedule::PyScheduleRuleNode::f_clone

The packed function to the Clone function.

f_initialize_with_tune_context

FInitializeWithTuneContext tvm::meta_schedule::PyScheduleRuleNode::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/schedule_rule.h