tvm::meta_schedule::PySpaceGeneratorNode Class Reference

The design space generator with customized methods on the python-side. More...

#include <space_generator.h>

Inheritance diagram for tvm::meta_schedule::PySpaceGeneratorNode:

Collaboration diagram for tvm::meta_schedule::PySpaceGeneratorNode:

Public Types
using	FInitializeWithTuneContext = SpaceGenerator::FInitializeWithTuneContext
using	FGenerateDesignSpace = SpaceGenerator::FGenerateDesignSpace
using	FClone = SpaceGenerator::FClone

Public Member Functions
void	InitializeWithTuneContext (const TuneContext &context) final
	Initialize the design space generator with tuning context. More...
Array< tir::Schedule >	GenerateDesignSpace (const IRModule &mod) final
	Generate design spaces given a module. More...
SpaceGenerator	Clone () const final
	Clone the space generator. More...
	TVM_DECLARE_FINAL_OBJECT_INFO (PySpaceGeneratorNode, SpaceGeneratorNode)
Public Member Functions inherited from tvm::meta_schedule::SpaceGeneratorNode
virtual	~SpaceGeneratorNode ()=default
	Default destructor. More...
	TVM_DECLARE_BASE_OBJECT_INFO (SpaceGeneratorNode, Object)

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

Public Attributes
FInitializeWithTuneContext	f_initialize_with_tune_context
	The packed function to the InitializeWithTuneContext function. More...
FGenerateDesignSpace	f_generate_design_space
	The packed function to the GenerateDesignSpace function. More...
FClone	f_clone
	The packed function to the Clone function. More...
Public Attributes inherited from tvm::meta_schedule::SpaceGeneratorNode
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.PySpaceGenerator"
Static Public Attributes inherited from tvm::meta_schedule::SpaceGeneratorNode
static constexpr const char *	_type_key = "meta_schedule.SpaceGenerator"

Detailed Description

The design space generator with customized methods on the python-side.

Member Typedef Documentation

FClone

using tvm::meta_schedule::PySpaceGeneratorNode::FClone = SpaceGenerator::FClone

FGenerateDesignSpace

using tvm::meta_schedule::PySpaceGeneratorNode::FGenerateDesignSpace = SpaceGenerator::FGenerateDesignSpace

FInitializeWithTuneContext

using tvm::meta_schedule::PySpaceGeneratorNode::FInitializeWithTuneContext = SpaceGenerator::FInitializeWithTuneContext

Member Function Documentation

Clone()

SpaceGenerator tvm::meta_schedule::PySpaceGeneratorNode::Clone ( ) const

finalvirtual

Clone the space generator.

Returns

The cloned space generator.

Implements tvm::meta_schedule::SpaceGeneratorNode.

GenerateDesignSpace()

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

finalvirtual

Generate design spaces given a module.

Parameters

mod	The module used for design space generation.

Returns

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

Implements tvm::meta_schedule::SpaceGeneratorNode.

InitializeWithTuneContext()

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

Reimplemented from tvm::meta_schedule::SpaceGeneratorNode.

RegisterReflection()

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

inlinestatic

TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::meta_schedule::PySpaceGeneratorNode::TVM_DECLARE_FINAL_OBJECT_INFO	(	PySpaceGeneratorNode	,
		SpaceGeneratorNode
	)

Member Data Documentation

_type_key

constexpr const char* tvm::meta_schedule::PySpaceGeneratorNode::_type_key = "meta_schedule.PySpaceGenerator"

staticconstexpr

f_clone

FClone tvm::meta_schedule::PySpaceGeneratorNode::f_clone

The packed function to the Clone function.

f_generate_design_space

FGenerateDesignSpace tvm::meta_schedule::PySpaceGeneratorNode::f_generate_design_space

The packed function to the GenerateDesignSpace function.

f_initialize_with_tune_context

FInitializeWithTuneContext tvm::meta_schedule::PySpaceGeneratorNode::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/space_generator.h