[tvm] branch main updated: [Bugfix][AutoScheduler] Correctly resume status (#7614)

[lm...@apache.org]

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 <co...@gmail.com>
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: