19 #ifndef TVM_META_SCHEDULE_SEARCH_STRATEGY_H_
20 #define TVM_META_SCHEDULE_SEARCH_STRATEGY_H_
22 #include <tvm/ffi/container/array.h>
23 #include <tvm/ffi/function.h>
24 #include <tvm/ffi/optional.h>
25 #include <tvm/ffi/reflection/registry.h>
35 namespace meta_schedule {
100 virtual void PreTuning(
int max_trials,
int num_trials_per_iter,
101 const Array<tir::Schedule>& design_spaces,
102 const Optional<Database>& database,
103 const Optional<CostModel>& cost_model) = 0;
124 const Array<RunnerResult>& results) = 0;
132 static constexpr
const char*
_type_key =
"meta_schedule.SearchStrategy";
151 ffi::TypedFunction<void(
int max_trials,
int num_trials_per_iter,
const Array<tir::Schedule>&,
152 const Optional<Database>&,
const Optional<CostModel>&)>;
165 ffi::TypedFunction<void(
const Array<MeasureCandidate>&,
const Array<RunnerResult>&)>;
210 double init_measured_ratio,
211 int init_min_unmeasured,
213 int genetic_num_iters,
214 double genetic_mutate_prob,
215 int genetic_max_fail_count,
254 void PreTuning(
int max_trials,
int num_trials_per_iter,
const Array<tir::Schedule>& design_spaces,
255 const Optional<Database>& database,
const Optional<CostModel>& cost_model)
final;
262 static constexpr const
char*
_type_key = "meta_schedule.PySearchStrategy";
Managed reference to TuneContextNode.
Definition: tune_context.h:98
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37
A managed object in the TVM runtime.