This is an automated email from the ASF dual-hosted git repository.
lmzheng pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new dfc231c [Bugfix][AutoScheduler] Correctly resume status (#7614)
dfc231c is described below
commit dfc231c97b6fa211256719894f23fa9a80d7326f
Author: Cody Yu <[email protected]>
AuthorDate: Tue Mar 9 05:38:13 2021 -0800
[Bugfix][AutoScheduler] Correctly resume status (#7614)
---
python/tvm/auto_scheduler/task_scheduler.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/python/tvm/auto_scheduler/task_scheduler.py
b/python/tvm/auto_scheduler/task_scheduler.py
index b6b0529..0221870 100644
--- a/python/tvm/auto_scheduler/task_scheduler.py
+++ b/python/tvm/auto_scheduler/task_scheduler.py
@@ -516,9 +516,9 @@ class TaskScheduler:
if res.error_no == 0:
cost = array_mean(res.costs)
- if self.best_costs[task_idx] < cost:
+ if cost < self.best_costs[task_idx]:
self.best_costs[task_idx] = cost
- self.task_best_cts = self.task_cts[task_idx]
+ self.task_best_cts[task_idx] = self.task_cts[task_idx]
for idx in range(len(self.tasks)):
if self.task_cts[idx] - self.task_best_cts[idx] >
self.early_stopping_task:
