[Ansor][FLAKY] Bug fix for compute at mutation error (#6557)

apache · Sep 25, 2020 · 8889c7a · 8889c7a
1 parent 1a9dcf1
commit 8889c7a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/auto_scheduler/search_policy/sketch_policy_rules.cc b/src/auto_scheduler/search_policy/sketch_policy_rules.cc
@@ -514,7 +514,11 @@ PopulationGenerationRule::ResultKind InitChangeComputeLocation::Apply(
     }
   }
 
-  *state = policy->search_task->compute_dag.InferBound(*state);
+  try {
+    *state = policy->search_task->compute_dag.InferBound(*state);
+  } catch (std::exception& e) {
+    return PopulationGenerationRule::ResultKind::kInvalid;
+  }
   return PopulationGenerationRule::ResultKind::kValid;
 }