diff --git a/src/auto_scheduler/search_policy/sketch_policy_rules.cc b/src/auto_scheduler/search_policy/sketch_policy_rules.cc
index 2eaa1329ff13..089ac774d1d6 100644
--- 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;
 }