Skip to content

Commit

Permalink
[MetaSchedule][Minor] Fix EvaluatorConfig Argument Description (apach…
Browse files Browse the repository at this point in the history
…e#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)
  • Loading branch information
zxybazh authored Jun 17, 2022
1 parent 8a94b66 commit 2708b6c
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions python/tvm/meta_schedule/runner/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,9 +28,14 @@ class EvaluatorConfig(NamedTuple):
Parameters
----------
number: int
The number of runs.
The number of times to run this function for taking average.
We call these runs as one `repeat` of measurement.
repeat: int
The number of times to repeat in each run.
The number of times to repeat the measurement.
In total, the function will be invoked (1 + number x repeat) times,
where the first one is warm up and will be discarded.
The returned result contains `repeat` costs,
each of which is an average of `number` costs.
min_repeat_ms: int
Minimum repeat time in ms. if the execution latency is too short,
increase the number of runs to the given time (in ms) to reduce the measurement error.
Expand Down

0 comments on commit 2708b6c

Please sign in to comment.