diff --git a/src/meta_schedule/integration.cc b/src/meta_schedule/integration.cc index d2cb4b307bbf..2a870ca61157 100644 --- a/src/meta_schedule/integration.cc +++ b/src/meta_schedule/integration.cc @@ -140,7 +140,8 @@ IRModule ApplyHistoryBestNode::Query(runtime::String task_name, IRModule mod, Ta return IRModule({{gv, func}}); } } - LOG(WARNING) << "Cannot find workload: " << task_name << "\n" << tir::AsTVMScript(prim_mod); + LOG(WARNING) << "Cannot find workload: " << task_name; + DLOG(INFO) << tir::AsTVMScript(prim_mod); return IRModule{nullptr}; } diff --git a/src/relay/backend/task_extraction.cc b/src/relay/backend/task_extraction.cc index 62f103e22a46..d7757c31bed8 100644 --- a/src/relay/backend/task_extraction.cc +++ b/src/relay/backend/task_extraction.cc @@ -46,14 +46,14 @@ Array ExtractTask(IRModule mod, Target target, Map tasks; - std::unordered_set cache_; + std::unordered_set cache; std::unordered_map name_map; - PostOrderVisit(opt_mod->Lookup("main"), [target, &tasks, &cache_, &name_map](const Expr& exp) { + PostOrderVisit(opt_mod->Lookup("main"), [target, &tasks, &cache, &name_map](const Expr& exp) { if (exp->IsInstance()) { Function relay_func = Downcast(exp); tec::CCacheKey cache_key(relay_func, target); - if (relay_func->HasNonzeroAttr(attr::kPrimitive) && cache_.find(cache_key) == cache_.end()) { + if (relay_func->HasNonzeroAttr(attr::kPrimitive) && cache.find(cache_key) == cache.end()) { Array outputs; std::string fused_name; std::tie(outputs, fused_name) = @@ -64,7 +64,7 @@ Array ExtractTask(IRModule mod, Target target, Map