TheRock/compiler/amd-llvm/mlir/test/mlir-pdll/CodeGen/CPP/general.pdll

// RUN: mlir-pdll %s -I %S -I %S/../../../../include -split-input-file -x cpp | FileCheck %s

// Check that we generate a wrapper pattern for each PDL pattern. Also
// add in a pattern awkwardly named the same as our generated patterns to
// check that we handle overlap.

// CHECK: struct GeneratedPDLLPattern0 : ::mlir::PDLPatternModule {
// CHECK:  template <typename... ConfigsT>
// CHECK:  : ::mlir::PDLPatternModule(::mlir::parseSourceString<::mlir::ModuleOp>(
// CHECK:  R"mlir(
// CHECK:    pdl.pattern
// CHECK:      operation "test.op"
// CHECK:  )mlir", context), std::forward<ConfigsT>(configs)...)

// CHECK: struct NamedPattern : ::mlir::PDLPatternModule {
// CHECK:  : ::mlir::PDLPatternModule(::mlir::parseSourceString<::mlir::ModuleOp>(
// CHECK:  R"mlir(
// CHECK:    pdl.pattern
// CHECK:      operation "test.op2"
// CHECK:  )mlir", context), std::forward<ConfigsT>(configs)...)

// CHECK: struct GeneratedPDLLPattern1 : ::mlir::PDLPatternModule {

// CHECK: struct GeneratedPDLLPattern2 : ::mlir::PDLPatternModule {
// CHECK:  : ::mlir::PDLPatternModule(::mlir::parseSourceString<::mlir::ModuleOp>(
// CHECK:  R"mlir(
// CHECK:    pdl.pattern
// CHECK:      operation "test.op3"
// CHECK:  )mlir", context), std::forward<ConfigsT>(configs)...)

// CHECK{LITERAL}: [[maybe_unused]] static void populateGeneratedPDLLPatterns(::mlir::RewritePatternSet &patterns, ConfigsT &&...configs) {
// CHECK-NEXT:   patterns.add<GeneratedPDLLPattern0>(patterns.getContext(), configs...);
// CHECK-NEXT:   patterns.add<NamedPattern>(patterns.getContext(), configs...);
// CHECK-NEXT:   patterns.add<GeneratedPDLLPattern1>(patterns.getContext(), configs...);
// CHECK-NEXT:   patterns.add<GeneratedPDLLPattern2>(patterns.getContext(), configs...);
// CHECK-NEXT: }

Pattern => erase op<test.op>;
Pattern NamedPattern => erase op<test.op2>;
Pattern GeneratedPDLLPattern1 => erase op<>;
Pattern => erase op<test.op3>;

// -----

// Check the generation of native constraints and rewrites.

#include "include/ods.td"

// CHECK:      static ::llvm::LogicalResult TestCstPDLFn(::mlir::PatternRewriter &rewriter,
// CHECK-SAME:     ::mlir::Attribute attr, ::mlir::Operation * op, ::mlir::Type type,
// CHECK-SAME:     ::mlir::Value value, ::mlir::TypeRange typeRange, ::mlir::ValueRange valueRange) {
// CHECK-NEXT:   return success();
// CHECK: }

// CHECK-NOT: TestUnusedCst

// CHECK: static void TestRewritePDLFn(::mlir::PatternRewriter &rewriter,
// CHECK-SAME:     ::mlir::Attribute attr, ::mlir::Operation * op, ::mlir::Type type,
// CHECK-SAME:     ::mlir::Value value, ::mlir::TypeRange typeRange, ::mlir::ValueRange valueRange) {
// CHECK: foo;
// CHECK: }

// CHECK: TestAttrInterface TestRewriteODSPDLFn(::mlir::PatternRewriter &rewriter, TestAttrInterface attr) {
// CHECK: static ::mlir::Attribute TestRewriteSinglePDLFn(::mlir::PatternRewriter &rewriter) {
// CHECK: std::tuple<::mlir::Attribute, ::mlir::Type> TestRewriteTuplePDLFn(::mlir::PatternRewriter &rewriter) {

// CHECK-NOT: TestUnusedRewrite

// CHECK: struct TestCstAndRewrite : ::mlir::PDLPatternModule {
// CHECK:   registerConstraintFunction("TestCst", TestCstPDLFn);
// CHECK:   registerRewriteFunction("TestRewrite", TestRewritePDLFn);

Constraint TestCst(attr: Attr, op: Op, type: Type, value: Value, typeRange: TypeRange, valueRange: ValueRange) [{
  return success();
}];
Constraint TestUnusedCst() [{ return success(); }];

Rewrite TestRewrite(attr: Attr, op: Op, type: Type, value: Value, typeRange: TypeRange, valueRange: ValueRange) [{ foo; }];
Rewrite TestRewriteODS(attr: TestAttrInterface) -> TestAttrInterface [{}];
Rewrite TestRewriteSingle() -> Attr [{}];
Rewrite TestRewriteTuple() -> (Attr, Type) [{}];
Rewrite TestUnusedRewrite(op: Op) [{}];

Pattern TestCstAndRewrite {
  let root = op<>(operand: Value, operands: ValueRange) -> (type: Type, types: TypeRange);
  TestCst(attr<"true">, root, type, operand, types, operands);
  rewrite root with {
    TestRewrite(attr<"true">, root, type, operand, types, operands);
    TestRewriteODS(attr<"true">);
    TestRewriteSingle();
    TestRewriteTuple();
    erase root;
  };
}