Skip to content

Commit 4b7b7de

Browse files
authored
[MetaScheduler] Improve print info about builder/runner state (#19767)
Adds on-screen info about samples failed to build, that may happen in early stages. ---------- #### Before: ``` 2026-06-14 18:36:30 [INFO] [task_scheduler.cc:196] TaskScheduler picks Task #0: "dense" 2026-06-14 18:36:32 [INFO] [task_scheduler.cc:209] Sending 64 sample(s) to builder 2026-06-14 18:38:47 [INFO] [task_scheduler.cc:211] Sending 64 sample(s) to runner ``` #### After: ``` 2026-06-14 19:53:23 [INFO] [task_scheduler.cc:193] TaskScheduler picks Task #0: "dense" 2026-06-14 19:53:25 [INFO] [task_scheduler.cc:206] Sending 64 sample(s) to builder 2026-06-14 19:55:39 [INFO] [task_scheduler.cc:215] Build errors: 64 sample(s) 2026-06-14 19:55:39 [INFO] [task_scheduler.cc:217] Sending 0 valid sample(s) to runner ```
1 parent 8b79dfd commit 4b7b7de

1 file changed

Lines changed: 12 additions & 1 deletion

File tree

src/s_tir/meta_schedule/task_scheduler/task_scheduler.cc

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -205,7 +205,18 @@ void TaskSchedulerNode::Tune(ffi::Array<TuneContext> ctxs, ffi::Array<FloatImm>
205205
num_trials_already += num_candidates;
206206
TVM_PY_LOG(INFO, this->logger) << "Sending " << num_candidates << " sample(s) to builder";
207207
SendToBuilder(task, builder);
208-
TVM_PY_LOG(INFO, this->logger) << "Sending " << num_candidates << " sample(s) to runner";
208+
int n_build_errs = 0;
209+
const ffi::Array<BuilderResult>& builder_results = task->builder_results.value();
210+
for (int i = 0; i < num_candidates; i++) {
211+
if (builder_results[i]->error_msg.has_value())
212+
++n_build_errs;
213+
}
214+
if (n_build_errs > 0) {
215+
TVM_PY_LOG(INFO, this->logger) << "Build errors: " << n_build_errs << " sample(s)";
216+
}
217+
TVM_PY_LOG(INFO, this->logger) << "Sending "
218+
<< num_candidates - n_build_errs
219+
<< " valid sample(s) to runner";
209220
SendToRunner(task, runner);
210221
} else {
211222
TerminateTask(task_id);

0 commit comments

Comments
 (0)