1//===- TestDecomposeAffineOps.cpp - Test affine ops decomposition utility -===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// This file implements a pass to test affine data copy utility functions and
10// options.
11//
12//===----------------------------------------------------------------------===//
13
14#include "mlir/Dialect/Affine/IR/AffineOps.h"
15#include "mlir/Dialect/Affine/Transforms/Transforms.h"
16#include "mlir/Dialect/Func/IR/FuncOps.h"
17#include "mlir/IR/PatternMatch.h"
18#include "mlir/Pass/Pass.h"
19#include "mlir/Transforms/Passes.h"
20
21#define PASS_NAME "test-decompose-affine-ops"
22
23using namespace mlir;
24using namespace mlir::affine;
25
26namespace {
27
28struct TestDecomposeAffineOps
29 : public PassWrapper<TestDecomposeAffineOps, OperationPass<func::FuncOp>> {
30 MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(TestDecomposeAffineOps)
31
32 StringRef getArgument() const final { return PASS_NAME; }
33 StringRef getDescription() const final {
34 return "Tests affine ops decomposition utility functions.";
35 }
36 TestDecomposeAffineOps() = default;
37 TestDecomposeAffineOps(const TestDecomposeAffineOps &pass) = default;
38
39 void runOnOperation() override;
40};
41
42} // namespace
43
44void TestDecomposeAffineOps::runOnOperation() {
45 IRRewriter rewriter(&getContext());
46 this->getOperation().walk([&](AffineApplyOp op) {
47 rewriter.setInsertionPoint(op);
48 reorderOperandsByHoistability(rewriter, op);
49 (void)decompose(rewriter, op);
50 });
51}
52
53namespace mlir {
54void registerTestDecomposeAffineOpPass() {
55 PassRegistration<TestDecomposeAffineOps>();
56}
57} // namespace mlir
58

source code of mlir/test/lib/Dialect/Affine/TestDecomposeAffineOps.cpp