diff --git a/src/meta_schedule/search_strategy/evolutionary_search.cc b/src/meta_schedule/search_strategy/evolutionary_search.cc index e43e15c9997a..4872f3aa5f6e 100644 --- a/src/meta_schedule/search_strategy/evolutionary_search.cc +++ b/src/meta_schedule/search_strategy/evolutionary_search.cc @@ -592,6 +592,7 @@ std::vector EvolutionarySearchNode::State::EvolveWithCostModel( // Loop until success for (int fail_count = 0; fail_count <= self->genetic_max_fail_count; ++fail_count) { sampled_trace_id = trace_sampler(); + sampled_trace_id = sampled_trace_id % self->population_size; tir::Trace trace = population.at(sampled_trace_id)->trace().value(); if (Optional opt_mutator = mutator_sampler()) { // Decision: mutate