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

neo-ai · Oct 19, 2020 · db8296a · db8296a
1 parent 88b5da4
commit db8296a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/auto_scheduler/search_policy/sketch_policy_rules.cc b/src/auto_scheduler/search_policy/sketch_policy_rules.cc
@@ -519,9 +519,15 @@ PopulationGenerationRule::ResultKind InitChangeComputeLocation::Apply(
   try {
     *state = policy->search_task->compute_dag.InferBound(*state);
   } catch (std::exception& e) {
+<<<<<<< HEAD
     return ResultKind::kInvalid;
   }
   return ResultKind::kValid;
+=======
+    return PopulationGenerationRule::ResultKind::kInvalid;
+  }
+  return PopulationGenerationRule::ResultKind::kValid;
+>>>>>>> [Ansor][FLAKY] Bug fix for compute at mutation error (#6557)
 }
 
 PopulationGenerationRule::ResultKind InitParallel::Apply(SketchPolicyNode* policy, State* state,