diff --git a/src/meta_schedule/database/memory_database.cc b/src/meta_schedule/database/memory_database.cc index 8cbde46f83b7..533a86acacfd 100644 --- a/src/meta_schedule/database/memory_database.cc +++ b/src/meta_schedule/database/memory_database.cc @@ -85,7 +85,7 @@ class MemoryDatabaseNode : public DatabaseNode { } std::stable_sort(results.begin(), results.end(), SortTuningRecordByMeanRunSecs()); if (results.size() > static_cast(top_k)) { - return {results.begin(), results.end() + top_k}; + return {results.begin(), results.begin() + top_k}; } else { if (results.size() < static_cast(top_k)) { LOG(WARNING) << "The size of the GetTopK result is smaller than requested. There are not "