tvm::meta_schedule::SpaceGenerator Class Reference

Managed reference to SpaceGeneratorNode. More...

#include <space_generator.h>

Inheritance diagram for tvm::meta_schedule::SpaceGenerator:

Collaboration diagram for tvm::meta_schedule::SpaceGenerator:

Public Types
using	FInitializeWithTuneContext = ffi::TypedFunction< void(const TuneContext &)>
	The function type of InitializeWithTuneContext method. More...
using	FGenerateDesignSpace = ffi::TypedFunction< ffi::Array< tir::Schedule >(const IRModule &)>
	The function type of GenerateDesignSpace method. More...
using	FClone = ffi::TypedFunction< SpaceGenerator()>
	The function type of Clone method. More...

Public Member Functions
	SpaceGenerator (ObjectPtr< SpaceGeneratorNode > data)
	Constructor from ObjectPtr<SpaceGeneratorNode>. More...
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (SpaceGenerator, ObjectRef, SpaceGeneratorNode)

Static Public Member Functions
static SpaceGenerator	PySpaceGenerator (ffi::Optional< ffi::Array< ScheduleRule >> sch_rules, ffi::Optional< ffi::Array< Postproc >> postprocs, ffi::Optional< ffi::Map< Mutator, FloatImm >> mutator_probs, FInitializeWithTuneContext f_initialize_with_tune_context, FGenerateDesignSpace f_generate_design_space, FClone f_clone)
	Create a design space generator with customized methods on the python-side. More...
static SpaceGenerator	ScheduleFn (ffi::Function schedule_fn, ffi::Optional< ffi::Array< ScheduleRule >> sch_rules, ffi::Optional< ffi::Array< Postproc >> postprocs, ffi::Optional< ffi::Map< Mutator, FloatImm >> mutator_probs)
	Create a design space generator with customized schedule function. More...
static SpaceGenerator	SpaceGeneratorUnion (ffi::Array< SpaceGenerator, void > space_generators, ffi::Optional< ffi::Array< ScheduleRule >> sch_rules, ffi::Optional< ffi::Array< Postproc >> postprocs, ffi::Optional< ffi::Map< Mutator, FloatImm >> mutator_probs)
	Create a design space generator that is union of multiple design space generators. More...
static SpaceGenerator	PostOrderApply (ffi::Function f_block_filter, ffi::Optional< ffi::Array< ScheduleRule >> sch_rules, ffi::Optional< ffi::Array< Postproc >> postprocs, ffi::Optional< ffi::Map< Mutator, FloatImm >> mutator_probs)
	Create a design space generator that generates design spaces by applying schedule rules to blocks in post-DFS order. More...

Protected Member Functions
	SpaceGenerator ()=default

Detailed Description

Managed reference to SpaceGeneratorNode.

See also

SpaceGeneratorNode

Member Typedef Documentation

FClone

using tvm::meta_schedule::SpaceGenerator::FClone = ffi::TypedFunction<SpaceGenerator()>

The function type of Clone method.

Returns

The cloned space generator.

FGenerateDesignSpace

using tvm::meta_schedule::SpaceGenerator::FGenerateDesignSpace = ffi::TypedFunction<ffi::Array<tir::Schedule>(const IRModule&)>

The function type of GenerateDesignSpace method.

Parameters

mod	The module used for design space generation.

Returns

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

FInitializeWithTuneContext

using tvm::meta_schedule::SpaceGenerator::FInitializeWithTuneContext = ffi::TypedFunction<void(const TuneContext&)>

The function type of InitializeWithTuneContext method.

Parameters

context

The tuning context for initialization.

Constructor & Destructor Documentation

SpaceGenerator() [1/2]

tvm::meta_schedule::SpaceGenerator::SpaceGenerator ( ObjectPtr< SpaceGeneratorNode >  data )

inlineexplicit

Constructor from ObjectPtr<SpaceGeneratorNode>.

Parameters

data	The object pointer.

SpaceGenerator() [2/2]

tvm::meta_schedule::SpaceGenerator::SpaceGenerator ( )

protecteddefault

Member Function Documentation

PostOrderApply()

static SpaceGenerator tvm::meta_schedule::SpaceGenerator::PostOrderApply ( ffi::Function  f_block_filter, ffi::Optional< ffi::Array< ScheduleRule >>  sch_rules, ffi::Optional< ffi::Array< Postproc >>  postprocs, ffi::Optional< ffi::Map< Mutator, FloatImm >>  mutator_probs  )

static

Create a design space generator that generates design spaces by applying schedule rules to blocks in post-DFS order.

Parameters

f_block_filter	The filter function to filter blocks to be applied with schedule rules.
sch_rules	The schedule rules.
postprocs	The postprocessors.
mutator_probs	The probability of using certain mutator.

Returns

The design space generator created.

PySpaceGenerator()

static SpaceGenerator tvm::meta_schedule::SpaceGenerator::PySpaceGenerator ( ffi::Optional< ffi::Array< ScheduleRule >>  sch_rules, ffi::Optional< ffi::Array< Postproc >>  postprocs, ffi::Optional< ffi::Map< Mutator, FloatImm >>  mutator_probs, FInitializeWithTuneContext  f_initialize_with_tune_context, FGenerateDesignSpace  f_generate_design_space, FClone  f_clone  )

static

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

Parameters

sch_rules	The schedule rules.
postprocs	The postprocessors.
mutator_probs	The probability of using certain mutator.
f_initialize_with_tune_context	The packed function of InitializeWithTuneContext.
f_generate_design_space	The packed function of GenerateDesignSpace.
f_clone	The packed function of Clone.

Returns

The design space generator created.

ScheduleFn()

static SpaceGenerator tvm::meta_schedule::SpaceGenerator::ScheduleFn ( ffi::Function  schedule_fn, ffi::Optional< ffi::Array< ScheduleRule >>  sch_rules, ffi::Optional< ffi::Array< Postproc >>  postprocs, ffi::Optional< ffi::Map< Mutator, FloatImm >>  mutator_probs  )

static

Create a design space generator with customized schedule function.

Parameters

schedule_fn	The schedule function, which can have the following signatures: 1) void(Schedule) 2) Schedule(Schedule) 3) ffi::Array<Schedule>(Schedule)
sch_rules	The schedule rules.
postprocs	The postprocessors.
mutator_probs	The probability of using certain mutator.

SpaceGeneratorUnion()

static SpaceGenerator tvm::meta_schedule::SpaceGenerator::SpaceGeneratorUnion ( ffi::Array< SpaceGenerator, void >  space_generators, ffi::Optional< ffi::Array< ScheduleRule >>  sch_rules, ffi::Optional< ffi::Array< Postproc >>  postprocs, ffi::Optional< ffi::Map< Mutator, FloatImm >>  mutator_probs  )

static

Create a design space generator that is union of multiple design space generators.

Parameters

space_generators	An array of design space generators to be unioned.
sch_rules	The schedule rules.
postprocs	The postprocessors.
mutator_probs	The probability of using certain mutator.

Returns

The design space generator created.

TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE()

tvm::meta_schedule::SpaceGenerator::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE	(	SpaceGenerator	,
		ObjectRef	,
		SpaceGeneratorNode
	)

---

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

• include/tvm/meta_schedule/space_generator.h