File tree Expand file tree Collapse file tree 1 file changed +7
-3
lines changed
include/tvm/meta_schedule Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Original file line number Diff line number Diff line change 19
19
#ifndef TVM_META_SCHEDULE_SEARCH_STRATEGY_H_
20
20
#define TVM_META_SCHEDULE_SEARCH_STRATEGY_H_
21
21
22
+ #include < tvm/meta_schedule/arg_info.h>
23
+ #include < tvm/meta_schedule/runner.h>
22
24
#include < tvm/tir/schedule/schedule.h>
23
25
24
- #include " ./arg_info.h"
25
- #include " ./runner.h"
26
-
27
26
namespace tvm {
28
27
namespace meta_schedule {
29
28
@@ -224,6 +223,11 @@ class SearchStrategy : public runtime::ObjectRef {
224
223
PySearchStrategyNode::FGenerateMeasureCandidates f_generate_measure_candidates, //
225
224
PySearchStrategyNode::FNotifyRunnerResults f_notify_runner_results);
226
225
226
+ /* !
227
+ * \brief Constructor of replay trace search strategy.
228
+ * \param num_trials_per_iter The number of trials per iteration, i.e., the batch size.
229
+ * \param num_trials_total The total number of trials for trace replaying.
230
+ */
227
231
TVM_DLL static SearchStrategy ReplayTrace (int num_trials_per_iter, int num_trials_total);
228
232
229
233
TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS (SearchStrategy, ObjectRef, SearchStrategyNode);
You can’t perform that action at this time.
0 commit comments