blob: f6b2b2b1c683f62f6296e6335f9c4409da735e96 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
//===- TestPDLByteCode.cpp - Test PDLL functionality ----------------------===//
//
// 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 "TestDialect.h"
#include "mlir/Dialect/PDL/IR/PDL.h"
#include "mlir/Dialect/PDLInterp/IR/PDLInterp.h"
#include "mlir/Interfaces/CastInterfaces.h"
#include "mlir/Parser/Parser.h"
#include "mlir/Pass/Pass.h"
#include "mlir/Pass/PassManager.h"
#include "mlir/Transforms/GreedyPatternRewriteDriver.h"
using namespace mlir;
#include "TestPDLLPatterns.h.inc"
namespace {
struct TestPDLLPass : public PassWrapper<TestPDLLPass, OperationPass<>> {
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(TestPDLLPass)
StringRef getArgument() const final { return "test-pdll-pass"; }
StringRef getDescription() const final { return "Test PDLL functionality"; }
void getDependentDialects(DialectRegistry ®istry) const override {
registry.insert<pdl::PDLDialect, pdl_interp::PDLInterpDialect, test::TestDialect>();
}
LogicalResult initialize(MLIRContext *ctx) override {
// Build the pattern set within the `initialize` to avoid recompiling PDL
// patterns during each `runOnOperation` invocation.
RewritePatternSet patternList(ctx);
populateGeneratedPDLLPatterns(patternList);
patterns = std::move(patternList);
return success();
}
void runOnOperation() final {
// Invoke the pattern driver with the provided patterns.
(void)applyPatternsGreedily(getOperation(), patterns);
}
FrozenRewritePatternSet patterns;
};
} // namespace
namespace mlir {
namespace test {
void registerTestPDLLPasses() { PassRegistration<TestPDLLPass>(); }
} // namespace test
} // namespace mlir
|
