[mlir][ir] OpRewritePattern should accept generatedNames

mlir/include/mlir/IR/PatternMatch.h

 
 /// OpRewritePattern is a wrapper around RewritePattern that allows for
 /// matching and rewriting against an instance of a derived operation class as
 /// opposed to a raw Operation.
 template <typename SourceOp>
 struct OpRewritePattern
     : public detail::OpOrInterfaceRewritePatternBase<SourceOp> {
   /// Patterns must specify the root operation name they match against, and can
-  /// also specify the benefit of the pattern matching.
-  OpRewritePattern(MLIRContext *context, PatternBenefit benefit = 1)
+  /// also specify the benefit of the pattern matching and a list of generated
+  /// ops.
+  OpRewritePattern(MLIRContext *context, PatternBenefit benefit = 1,
+                   ArrayRef<StringRef> generatedNames = {})
       : detail::OpOrInterfaceRewritePatternBase<SourceOp>(
-            SourceOp::getOperationName(), benefit, context) {}
+            SourceOp::getOperationName(), benefit, context, generatedNames) {}
 };
 
 /// OpInterfaceRewritePattern is a wrapper around RewritePattern that allows for
 /// matching and rewriting against an instance of an operation interface instead
 /// of a raw Operation.
 template <typename SourceOp>
 struct OpInterfaceRewritePattern
     : public detail::OpOrInterfaceRewritePatternBase<SourceOp> {

mlir/unittests/IR/CMakeLists.txt

 add_mlir_unittest(MLIRIRTests
   AttributeTest.cpp
   DialectTest.cpp
   InterfaceAttachmentTest.cpp
   OperationSupportTest.cpp
+  PatternMatchTest.cpp
   ShapedTypeTest.cpp
   SubElementInterfaceTest.cpp
 
   DEPENDS
   MLIRTestInterfaceIncGen
 )
 target_include_directories(MLIRIRTests PRIVATE "${MLIR_BINARY_DIR}/test/lib/Dialect/Test")
 target_link_libraries(MLIRIRTests
   PRIVATE
   MLIRIR
   MLIRTestDialect)

mlir/unittests/IR/PatternMatchTest.cpp

+//===- PatternMatchTest.cpp - PatternMatch unit tests ---------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#include "mlir/IR/PatternMatch.h"
+#include "gtest/gtest.h"
+
+#include "../../test/lib/Dialect/Test/TestDialect.h"
+
+using namespace mlir;
+
+namespace {
+struct AnOpRewritePattern : OpRewritePattern<test::OpA> {
+  AnOpRewritePattern(MLIRContext *context)
+      : OpRewritePattern(context, /*benefit=*/1,
+                         /*generatedNames=*/{test::OpB::getOperationName()}) {}
+};
+TEST(OpRewritePatternTest, GetGeneratedNames) {
+  MLIRContext context;
+  AnOpRewritePattern pattern(&context);
+  ArrayRef<OperationName> ops = pattern.getGeneratedOps();
+
+  ASSERT_EQ(ops.size(), 1u);
+  ASSERT_EQ(ops.front().getStringRef(), test::OpB::getOperationName());
+}
+} // end anonymous namespace