Add hasNItemsOrLess and container variants of hasNItems and friends

[jurahul]

Summary

• Fixed a bug in hasNItems()
• Extend STLExtras unit test to add testing for hasSingleElement() and hasNItems() and friends.

[jurahul]

Fix clang-tidy failures

[jurahul]

Another attemp to fix clang-tidy issues

[mehdi_amini]

llvm/include/llvm/ADT/STLExtras.h

1948

Nit: period.

1963

Seems like two patches here, this added function to STLExtras seems to be the only thing relevant for the MLIR part. Can you split this?

[jurahul]

llvm/include/llvm/ADT/STLExtras.h

1963

Sure, I can split the MLIR adoption of this into its own change.

[mehdi_amini]

llvm/include/llvm/ADT/STLExtras.h

1963

I meant that hasNItems and the MLIR change can go in together, but the addition of the others hasNItemsOrLess and hasNItemsOrMore don't seem related here.

[jurahul]

Split the MLIR adoption into its own change, and add unit tests

[jurahul]

Complete sentence for comments

[jurahul]

llvm/include/llvm/ADT/STLExtras.h

1963

I have split the MLIR change and added unit tests for these functions (and fixed a bug in hasNItems)

[mehdi_amini]

Thanks for adding the tests! Did you find the bug while adding these by the way?

llvm/include/llvm/ADT/STLExtras.h

 }
 
 /// Test whether \p RangeOrContainer is empty. Similar to C++17 std::empty.
 template <typename T>
 constexpr bool empty(const T &RangeOrContainer) {
   return adl_begin(RangeOrContainer) == adl_end(RangeOrContainer);
 }
 
-/// Returns true of the given range only contains a single element.
-template <typename ContainerTy> bool hasSingleElement(ContainerTy &&c) {
-  auto it = std::begin(c), e = std::end(c);
+/// Returns true if the given container only contains a single element.
+template <typename ContainerTy> bool hasSingleElement(ContainerTy &&C) {
+  auto it = std::begin(C), e = std::end(C);
   return it != e && std::next(it) == e;
 }
 
 /// Return a range covering \p RangeOrContainer with the first N elements
 /// excluded.
 template <typename T> auto drop_begin(T &&RangeOrContainer, size_t N) {
   return make_range(std::next(adl_begin(RangeOrContainer), N),
                     adl_end(RangeOrContainer));
if (Begin == End)
       return false; // Too few.
     N -= ShouldBeCounted(*Begin);
   }
   return Begin == End;
 }
 
 /// Return true if the sequence [Begin, End) has N or more items. Runs in O(N)
 /// time. Not meant for use with random-access iterators.
-/// Can optionally take a predicate to filter lazily some items.
+/// Can optionally take a predicate to lazily filter some items.
 template<typename IterTy,
          typename Pred = bool (*)(const decltype(*std::declval<IterTy>()) &)>
 bool hasNItemsOrMore(
     IterTy &&Begin, IterTy &&End, unsigned N,
     Pred &&ShouldBeCounted =
         [](const decltype(*std::declval<IterTy>()) &) { return true; },
     std::enable_if_t<
         !std::is_same<typename std::iterator_traits<std::remove_reference_t<
                           decltype(Begin)>>::iterator_category,
                       std::random_access_iterator_tag>::value,
         void> * = nullptr) {
   for (; N; ++Begin) {
     if (Begin == End)
       return false; // Too few.
     N -= ShouldBeCounted(*Begin);
   }
   return true;
 }
 
+/// Returns true if the sequence [Begin, End) has N or less items. Can
+/// optionally take a predicate to lazily filter some items
+template <typename IterTy,
+          typename Pred = bool (*)(const decltype(*std::declval<IterTy>()) &)>
+bool hasNItemsOrLess(
+    IterTy &&Begin, IterTy &&End, unsigned N,
+    Pred &&ShouldBeCounted = [](const decltype(*std::declval<IterTy>()) &) {
+      return true;
+    }) {
+  assert(N != std::numeric_limits<unsigned>::max());
+  return !hasNItemsOrMore(Begin, End, N + 1, ShouldBeCounted);
+}
+
+/// Returns true if the given container has exactly N items
+template <typename ContainerTy> bool hasNItems(ContainerTy &&C, unsigned N) {
+  return hasNItems(std::begin(C), std::end(C), N);
+}
+
+/// Returns true if the given container has N or more items
+template <typename ContainerTy>
+bool hasNItemsOrMore(ContainerTy &&C, unsigned N) {
+  return hasNItemsOrMore(std::begin(C), std::end(C), N);
+}
+
+/// Returns true if the given container has N or less items
+template <typename ContainerTy>
+bool hasNItemsOrLess(ContainerTy &&C, unsigned N) {
+  return hasNItemsOrLess(std::begin(C), std::end(C), N);
+}
+
 /// Returns a raw pointer that represents the same address as the argument.
 ///
 /// This implementation can be removed once we move to C++20 where it's defined
 /// as std::to_address().
 ///
 /// The std::pointer_traits<>::to_address(p) variations of these overloads has
 /// not been implemented.
 template <class Ptr> auto to_address(const Ptr &P) { return P.operator->(); }
 template <class T> constexpr T *to_address(T *P) { return P; }
 
 } // end namespace llvm
 
 #endif // LLVM_ADT_STLEXTRAS_H

mlir/include/mlir/IR/OpBase.td

 class Region<Pred condition, string descr = ""> :
     RegionConstraint<condition, descr>;
 
 // Any region.
 def AnyRegion : Region<CPred<"true">, "any region">;
 
 // A region with the given number of blocks.
 class SizedRegion<int numBlocks> : Region<
-  CPred<"$_self.getBlocks().size() == " # numBlocks>,
+  CPred<"llvm::hasNItems($_self, " # numBlocks # ")">,
   "region with " # numBlocks # " blocks">;
 
 // A variadic region constraint. It expands to zero or more of the base region.
 class VariadicRegion<Region region>
   : Region<region.predicate, region.description>;
 
 //===----------------------------------------------------------------------===//
 // Successor definitions