Fix different time counting methods.

Author: zxybazh

src/meta_schedule/task_scheduler/gradient_based.cc

@@ -131,7 +131,7 @@ class GradientBasedNode final : public TaskSchedulerNode {
   }
 
   int NextTaskId() final {
-    time_counter.start_timer("TaskScheduler::NextTaskId");
+    time_counter.start_timer("GradientBasedTaskScheduler::NextTaskId");
     int n_tasks = task_records_.size();
     // Round robin
     if (num_rounds_already_ == 0) {
@@ -182,12 +182,12 @@ class GradientBasedNode final : public TaskSchedulerNode {
     if (tasks[task_id]->runner_futures.defined()) {
       JoinRunningTask(task_id);
     }
-    time_counter.end_timer("TaskScheduler::NextTaskId");
+    time_counter.end_timer("GradientBasedTaskScheduler::NextTaskId");
     return task_id;
   }
 
   Array<RunnerResult> JoinRunningTask(int task_id) final {
-    time_counter.start_timer("TaskScheduler::JoinRunningTask");
+    time_counter.start_timer("GradientBasedTaskScheduler::JoinRunningTask");
     TaskRecord& record = task_records_[task_id];
     Array<RunnerResult> results = TaskSchedulerNode::JoinRunningTask(task_id);
     double& best_time_cost = this->best_time_cost_per_task_[task_id];
@@ -201,7 +201,7 @@ class GradientBasedNode final : public TaskSchedulerNode {
     TVM_PY_LOG(INFO, this->logging_func)
         << "[Updated] Task #" << task_id << ": " << record.task->task_name << "\n"
         << this->TuningStatistics();
-    time_counter.end_timer("TaskScheduler::JoinRunningTask");
+    time_counter.end_timer("GradientBasedTaskScheduler::JoinRunningTask");
     return results;
   }
 };

src/meta_schedule/task_scheduler/task_scheduler.cc

@@ -123,7 +123,6 @@ void TaskSchedulerNode::InitializeTask(int task_id) {
 }
 
 void TaskSchedulerNode::Tune() {
-  time_counter.start_timer("TaskScheduler::Tune");
   int n_tasks = this->tasks.size();
   for (int task_id = 0; task_id < n_tasks; ++task_id) {
     InitializeTask(task_id);
@@ -166,7 +165,6 @@ void TaskSchedulerNode::Tune() {
     }
     task->search_strategy.value()->PostTuning();
   }
-  time_counter.end_timer("TaskScheduler::Tune");
 }
 
 void TaskSchedulerNode::TouchTask(int task_id) {
@@ -191,9 +189,11 @@ Array<RunnerResult> TaskSchedulerNode::JoinRunningTask(int task_id) {
   int n = futures.size();
   Array<RunnerResult> results;
   results.reserve(n);
+  time_counter.start_timer("TaskScheduler::JoinRunningTask::future->Result");
   for (RunnerFuture future : futures) {
     results.push_back(future->Result());
   }
+  time_counter.end_timer("TaskScheduler::JoinRunningTask::future->Result");
   task->search_strategy.value()->NotifyRunnerResults(task, task->measure_candidates.value(),
                                                      results);
   // Invoke the callbacks