Fix for i386.

apache · Mar 3, 2022 · 01c22c3 · 01c22c3
1 parent 166834c
commit 01c22c3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/meta_schedule/task_scheduler/gradient_based.cc b/src/meta_schedule/task_scheduler/gradient_based.cc
@@ -299,7 +299,8 @@ TaskScheduler TaskScheduler::GradientBased(Array<TuneContext> tasks,
     std::string tag = tag_generation_func(mod);
 
     if (tag_to_group.find(tag) == tag_to_group.end()) {
-      tag_to_group[tag] = tag_to_group.size();
+      int group_cnt = tag_to_group.size();
+      tag_to_group[tag] = group_cnt;
       n->task_groups.push_back(std::set<int>());
     }
     int group_id = tag_to_group[tag];