[tvm] branch main updated: [MetaSchedule] bug fix ApplyHistoryBest. Previously, ApplyHistoryBest returned the incoming module without applying the tuning history. (#10183)

[ju...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

junrushao 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 afa0870  [MetaSchedule] bug fix ApplyHistoryBest. Previously, ApplyHistoryBest returned the incoming module without applying the tuning history. (#10183)
afa0870 is described below

commit afa0870e11c1bec2fff773c0f05e1a65d0ca08d6
Author: Sunghyun Park <49...@users.noreply.github.com>
AuthorDate: Mon Feb 7 23:23:23 2022 -0800

    [MetaSchedule] bug fix ApplyHistoryBest. Previously, ApplyHistoryBest returned the incoming module without applying the tuning history. (#10183)
---
 src/meta_schedule/integration.cc | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/meta_schedule/integration.cc b/src/meta_schedule/integration.cc
index cc26b56..1ecb537 100644
--- a/src/meta_schedule/integration.cc
+++ b/src/meta_schedule/integration.cc
@@ -130,7 +130,11 @@ Optional<ObjectRef> ApplyHistoryBestNode::Query(runtime::String task_name, IRMod
     Array<TuningRecord> records = database->GetTopK(database->CommitWorkload(prim_mod), 1);
     if (records.size() == 1) {
       LOG(INFO) << "Applied history best for " << task_name << ".";
-      return records[0]->workload->mod;
+      tir::Schedule sch =
+          tir::Schedule::Traced(records[0]->workload->mod, /*seed=*/-1, /*debug_mask=*/0,
+                                /*error_render_level=*/tir::ScheduleErrorRenderLevel::kNone);
+      records[0]->trace->ApplyToSchedule(sch, false);
+      return sch->mod();
     }
   }
   return NullOpt;