tvm
|
The python side customizable class for measure candidate generation. More...
#include <search_strategy.h>
Public Member Functions | |
void | InitializeWithTuneContext (const TuneContext &context) final |
Initialize the search strategy with tuning context. More... | |
void | PreTuning (int max_trials, int num_trials_per_iter, const Array< tir::Schedule > &design_spaces, const Optional< Database > &database, const Optional< CostModel > &cost_model) final |
Pre-tuning for the search strategy. More... | |
void | PostTuning () final |
Post-tuning for the search strategy. More... | |
Optional< Array< MeasureCandidate > > | GenerateMeasureCandidates () final |
Generate measure candidates from design spaces for measurement. More... | |
void | NotifyRunnerResults (const Array< MeasureCandidate > &measure_candidates, const Array< RunnerResult > &results) |
Update the search strategy with measurement results. More... | |
SearchStrategy | Clone () const final |
Clone the search strategy. More... | |
TVM_DECLARE_FINAL_OBJECT_INFO (PySearchStrategyNode, SearchStrategyNode) | |
![]() | |
virtual | ~SearchStrategyNode ()=default |
Virtual destructor. More... | |
TVM_DECLARE_BASE_OBJECT_INFO (SearchStrategyNode, Object) | |
Static Public Member Functions | |
static void | RegisterReflection () |
Public Attributes | |
FInitializeWithTuneContext | f_initialize_with_tune_context |
The packed function to the InitializeWithTuneContext method. More... | |
FPreTuning | f_pre_tuning |
The packed function to the PreTuning method. More... | |
FPostTuning | f_post_tuning |
The packed function to the PostTuning method. More... | |
FGenerateMeasureCandidates | f_generate_measure_candidates |
The packed function to the GenerateMeasureCandidates method. More... | |
FNotifyRunnerResults | f_notify_runner_results |
The packed function to the NotifyRunnerResults method. More... | |
FClone | f_clone |
The packed function to the Clone method. More... | |
Static Public Attributes | |
static constexpr const char * | _type_key = "meta_schedule.PySearchStrategy" |
![]() | |
static constexpr const char * | _type_key = "meta_schedule.SearchStrategy" |
The python side customizable class for measure candidate generation.
using tvm::meta_schedule::PySearchStrategyNode::FGenerateMeasureCandidates = SearchStrategy::FGenerateMeasureCandidates |
using tvm::meta_schedule::PySearchStrategyNode::FInitializeWithTuneContext = SearchStrategy::FInitializeWithTuneContext |
using tvm::meta_schedule::PySearchStrategyNode::FNotifyRunnerResults = SearchStrategy::FNotifyRunnerResults |
|
finalvirtual |
Clone the search strategy.
Implements tvm::meta_schedule::SearchStrategyNode.
|
finalvirtual |
Generate measure candidates from design spaces for measurement.
Implements tvm::meta_schedule::SearchStrategyNode.
|
finalvirtual |
Initialize the search strategy with tuning context.
context | The tuning context for initialization. |
Implements tvm::meta_schedule::SearchStrategyNode.
|
virtual |
Update the search strategy with measurement results.
measure_candidates | The candidates to be measured. |
results | The measurement results from the runner. |
Implements tvm::meta_schedule::SearchStrategyNode.
|
finalvirtual |
Post-tuning for the search strategy.
Implements tvm::meta_schedule::SearchStrategyNode.
|
finalvirtual |
Pre-tuning for the search strategy.
max_trials | The maximum number of trials. |
num_trials_per_iter | The number of trials per iteration. |
design_spaces | The design spaces used during tuning process. |
database | The database used during tuning process. |
cost_model | The cost model used during tuning process. |
Implements tvm::meta_schedule::SearchStrategyNode.
|
inlinestatic |
tvm::meta_schedule::PySearchStrategyNode::TVM_DECLARE_FINAL_OBJECT_INFO | ( | PySearchStrategyNode | , |
SearchStrategyNode | |||
) |
|
staticconstexpr |
FClone tvm::meta_schedule::PySearchStrategyNode::f_clone |
The packed function to the Clone
method.
FGenerateMeasureCandidates tvm::meta_schedule::PySearchStrategyNode::f_generate_measure_candidates |
The packed function to the GenerateMeasureCandidates
method.
FInitializeWithTuneContext tvm::meta_schedule::PySearchStrategyNode::f_initialize_with_tune_context |
The packed function to the InitializeWithTuneContext
method.
FNotifyRunnerResults tvm::meta_schedule::PySearchStrategyNode::f_notify_runner_results |
The packed function to the NotifyRunnerResults
method.
FPostTuning tvm::meta_schedule::PySearchStrategyNode::f_post_tuning |
The packed function to the PostTuning
method.
FPreTuning tvm::meta_schedule::PySearchStrategyNode::f_pre_tuning |
The packed function to the PreTuning
method.