Skip to content

Commit bd94217

Browse files
committed
[TVMC] Tune each AutoTVM tasks at least once
1 parent 8b60b7e commit bd94217

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/tvm/driver/tvmc/autotuner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -591,7 +591,7 @@ def tune_model(
591591

592592
# In autotvm, trials is specified per task. We can convert the per-model input
593593
# provided to per-task trials by dividing by the number of tasks.
594-
trials = int(trials / max(len(tasks), 1))
594+
trials = int(max(1, trials / max(len(tasks), 1)))
595595
logger.info("Autotuning with %d trials per task.", trials)
596596

597597
tuning_options = {

0 commit comments

Comments
 (0)