junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r985190427
##########
include/tvm/meta_schedule/search_strategy.h:
##########
@@ -257,8 +249,8 @@ class PySearchStrategyNode : public SearchStrategyNode {
}
void InitializeWithTuneContext(const TuneContext& context) final;
- void PreTuning(const Array<tir::Schedule>& design_spaces, const
Optional<Database>& database,
- const Optional<CostModel>& cost_model) final;
+ void PreTuning(int max_trials, int num_trials_per_iter, const
Array<tir::Schedule>& design_spaces,
Review Comment:
It limits the maximum number of trials for a single task. For example,
search ends once # trials reaches it:
https://github.com/apache/tvm/blob/main/src/meta_schedule/search_strategy/evolutionary_search.cc#L676
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
