Skip to content

Commit d16d672

Browse files
committed
Fix docs.
1 parent 7cf9c4a commit d16d672

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

include/tvm/meta_schedule/search_strategy.h

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,11 +19,10 @@
1919
#ifndef TVM_META_SCHEDULE_SEARCH_STRATEGY_H_
2020
#define TVM_META_SCHEDULE_SEARCH_STRATEGY_H_
2121

22+
#include <tvm/meta_schedule/arg_info.h>
23+
#include <tvm/meta_schedule/runner.h>
2224
#include <tvm/tir/schedule/schedule.h>
2325

24-
#include "./arg_info.h"
25-
#include "./runner.h"
26-
2726
namespace tvm {
2827
namespace meta_schedule {
2928

@@ -224,6 +223,11 @@ class SearchStrategy : public runtime::ObjectRef {
224223
PySearchStrategyNode::FGenerateMeasureCandidates f_generate_measure_candidates, //
225224
PySearchStrategyNode::FNotifyRunnerResults f_notify_runner_results);
226225

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+
*/
227231
TVM_DLL static SearchStrategy ReplayTrace(int num_trials_per_iter, int num_trials_total);
228232

229233
TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS(SearchStrategy, ObjectRef, SearchStrategyNode);

0 commit comments

Comments
 (0)