Skip to content

Commit 2708b6c

Browse files
authored
[MetaSchedule][Minor] Fix EvaluatorConfig Argument Description (#11766)
Pointed out by @sunggg that the description of `number` and `repeat` for evaluator configuration is not accurate, updated to a version more consistent with `TimeEvaluator`. ![TimeEvaluator](https://user-images.githubusercontent.com/3203174/174385966-74d3dbf6-dcca-43ea-9c0b-a91b4a281687.png)
1 parent 8a94b66 commit 2708b6c

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

python/tvm/meta_schedule/runner/config.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,14 @@ class EvaluatorConfig(NamedTuple):
2828
Parameters
2929
----------
3030
number: int
31-
The number of runs.
31+
The number of times to run this function for taking average.
32+
We call these runs as one `repeat` of measurement.
3233
repeat: int
33-
The number of times to repeat in each run.
34+
The number of times to repeat the measurement.
35+
In total, the function will be invoked (1 + number x repeat) times,
36+
where the first one is warm up and will be discarded.
37+
The returned result contains `repeat` costs,
38+
each of which is an average of `number` costs.
3439
min_repeat_ms: int
3540
Minimum repeat time in ms. if the execution latency is too short,
3641
increase the number of runs to the given time (in ms) to reduce the measurement error.

0 commit comments

Comments
 (0)