The abstract class for design space generation. More...

#include <space_generator.h>

Inheritance diagram for tvm::meta_schedule::SpaceGeneratorNode:

Collaboration diagram for tvm::meta_schedule::SpaceGeneratorNode:

Public Member Functions
virtual	~SpaceGeneratorNode ()=default
	Default destructor. More...

virtual void	InitializeWithTuneContext (const TuneContext &context)
	Initialize the design space generator with tuning context. More...

virtual Array< tir::Schedule >	GenerateDesignSpace (const IRModule &mod)=0
	Generate design spaces given a module. More...

virtual SpaceGenerator	Clone () const =0
	Clone the space generator. More...

	TVM_DECLARE_BASE_OBJECT_INFO (SpaceGeneratorNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
Optional< Array< ScheduleRule > >	sch_rules
	The schedule rules. More...

Optional< Array< Postproc > >	postprocs
	The postprocessors. More...

Optional< Map< Mutator, FloatImm > >	mutator_probs
	The probability of using certain mutator. More...

Static Public Attributes
static constexpr const char *	_type_key = "meta_schedule.SpaceGenerator"

Detailed Description

The abstract class for design space generation.

Note: The relationship between SpaceGenerator and other classes are as follows: +-----------------------------------------------------------—+ +–+--------------------------------------------------------—+ | +–+---------------— Tune Context --------------------------—+ | | | +------------------—+ | | | | | | Generate | | | | | Space Generator +-----------—+ | | | | | | | | | | | +------------------—+ v | | | | Design Space | | | | +------------------—+ | | | | | Generate | | Pretuning | | | | | +--------—+ Search Strategy |<----------—+ | | | | | | | | +–+ | | +------------------—+ +–+ +-—+-------------------------------------------------------—+ | | +-—+-------------— Managed By Task Scheduler ------------------—+ | | +--------—+ | | | Send to | | Send to | | v +----------—+| Builder +-------—+ | | Measure Candidate | Builder | | Runner | | | | | +--------—+ | | | | +---------—+---------—+ | | | | | | +--------—+ | | | +-—>| Task Scheduler | | | | | | | | | Runner |<--—+ | | +----------------------—+ | | | | ^ +--—+--—+ | | | | | | +-— Runner Future <----—+ | +------------------------------------------------------------------—+

Constructor & Destructor Documentation

◆ ~SpaceGeneratorNode()

virtual tvm::meta_schedule::SpaceGeneratorNode::~SpaceGeneratorNode ( )

virtualdefault

Default destructor.

Member Function Documentation

◆ Clone()

virtual SpaceGenerator tvm::meta_schedule::SpaceGeneratorNode::Clone ( ) const

pure virtual

Clone the space generator.

Returns: The cloned space generator.

Implemented in tvm::meta_schedule::PySpaceGeneratorNode.

◆ GenerateDesignSpace()

virtual Array<tir::Schedule> tvm::meta_schedule::SpaceGeneratorNode::GenerateDesignSpace ( const IRModule & mod )

pure virtual

Generate design spaces given a module.

Parameters

mod	The module used for design space generation.

Returns: The generated design spaces, i.e., schedules.

Implemented in tvm::meta_schedule::PySpaceGeneratorNode.

◆ InitializeWithTuneContext()

virtual void tvm::meta_schedule::SpaceGeneratorNode::InitializeWithTuneContext ( const TuneContext & context )

virtual

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.

Reimplemented in tvm::meta_schedule::PySpaceGeneratorNode.

◆ RegisterReflection()

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

inlinestatic

◆ TVM_DECLARE_BASE_OBJECT_INFO()

tvm::meta_schedule::SpaceGeneratorNode::TVM_DECLARE_BASE_OBJECT_INFO	(	SpaceGeneratorNode	,
		Object
	)

Member Data Documentation

◆ _type_key

constexpr const char* tvm::meta_schedule::SpaceGeneratorNode::_type_key = "meta_schedule.SpaceGenerator"

staticconstexpr

◆ mutator_probs

Optional<Map<Mutator, FloatImm> > tvm::meta_schedule::SpaceGeneratorNode::mutator_probs

The probability of using certain mutator.

◆ postprocs

Optional<Array<Postproc> > tvm::meta_schedule::SpaceGeneratorNode::postprocs

The postprocessors.

◆ sch_rules

Optional<Array<ScheduleRule> > tvm::meta_schedule::SpaceGeneratorNode::sch_rules

The schedule rules.

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

include/tvm/meta_schedule/space_generator.h

Public Member Functions

Static Public Member Functions

Public Attributes

Static Public Attributes