diff --git a/include/llvm/ADT/STLExtras.h b/include/llvm/ADT/STLExtras.h index de82f2bce7f..4ce2e9fb72a 100644 --- a/include/llvm/ADT/STLExtras.h +++ b/include/llvm/ADT/STLExtras.h @@ -1322,44 +1322,13 @@ void stable_sort(R &&Range, Compare C) { std::stable_sort(adl_begin(Range), adl_end(Range), C); } -/// Binary search for the first index where a predicate is true. -/// Returns the first I in [Lo, Hi) where C(I) is true, or Hi if it never is. -/// Requires that C is always false below some limit, and always true above it. -/// -/// Example: -/// size_t DawnModernEra = bsearch(1776, 2050, [](size_t Year){ -/// return Presidents.for(Year).twitterHandle() != None; -/// }); -/// -/// Note the return value differs from std::binary_search! -template -size_t bsearch(size_t Lo, size_t Hi, Predicate P) { - while (Lo != Hi) { - assert(Hi > Lo); - size_t Mid = Lo + (Hi - Lo) / 2; - if (P(Mid)) - Hi = Mid; - else - Lo = Mid + 1; - } - return Hi; -} - -/// Binary search for the first iterator where a predicate is true. -/// Returns the first I in [Lo, Hi) where C(*I) is true, or Hi if it never is. -/// Requires that C is always false below some limit, and always true above it. -template ())> -It bsearch(It Lo, It Hi, Predicate P) { - return std::lower_bound(Lo, Hi, 0u, - [&](const Val &V, unsigned) { return !P(V); }); -} - /// Binary search for the first iterator in a range where a predicate is true. /// Requires that C is always false below some limit, and always true above it. -template +template ()))> auto bsearch(R &&Range, Predicate P) -> decltype(adl_begin(Range)) { - return bsearch(adl_begin(Range), adl_end(Range), P); + return std::partition_point(adl_begin(Range), adl_end(Range), + [&](const Val &V) { return !P(V); }); } /// Wrapper function around std::equal to detect if all elements diff --git a/include/llvm/CodeGen/SlotIndexes.h b/include/llvm/CodeGen/SlotIndexes.h index 2c86930405b..10ab4cca831 100644 --- a/include/llvm/CodeGen/SlotIndexes.h +++ b/include/llvm/CodeGen/SlotIndexes.h @@ -492,8 +492,9 @@ class raw_ostream; /// Move iterator to the next IdxMBBPair where the SlotIndex is greater or /// equal to \p To. MBBIndexIterator advanceMBBIndex(MBBIndexIterator I, SlotIndex To) const { - return llvm::bsearch(I, idx2MBBMap.end(), - [=](const IdxMBBPair &IM) { return To <= IM.first; }); + return std::partition_point( + I, idx2MBBMap.end(), + [=](const IdxMBBPair &IM) { return IM.first < To; }); } /// Get an iterator pointing to the IdxMBBPair with the biggest SlotIndex diff --git a/unittests/ADT/STLExtrasTest.cpp b/unittests/ADT/STLExtrasTest.cpp index 456ad282447..ebb8c9f6f5d 100644 --- a/unittests/ADT/STLExtrasTest.cpp +++ b/unittests/ADT/STLExtrasTest.cpp @@ -470,19 +470,7 @@ TEST(STLExtrasTest, to_address) { } TEST(STLExtrasTest, bsearch) { - // Integer version. - EXPECT_EQ(7u, bsearch(5, 10, [](unsigned X) { return X >= 7; })); - EXPECT_EQ(5u, bsearch(5, 10, [](unsigned X) { return X >= 1; })); - EXPECT_EQ(10u, bsearch(5, 10, [](unsigned X) { return X >= 50; })); - - // Iterator version. std::vector V = {1, 3, 5, 7, 9}; - EXPECT_EQ(V.begin() + 3, - bsearch(V.begin(), V.end(), [](unsigned X) { return X >= 7; })); - EXPECT_EQ(V.begin(), - bsearch(V.begin(), V.end(), [](unsigned X) { return X >= 1; })); - EXPECT_EQ(V.end(), - bsearch(V.begin(), V.end(), [](unsigned X) { return X >= 50; })); // Range version. EXPECT_EQ(V.begin() + 3, bsearch(V, [](unsigned X) { return X >= 7; }));