72c65b698e
TestDialect has many operations and they all live in ::mlir namespace. Sometimes it is not clear whether the ops used in the code for the test passes belong to Standard or to Test dialects. Also, with this change it is easier to understand what test passes registered in mlir-opt are actually passes in mlir/test. Differential Revision: https://reviews.llvm.org/D90794
71 lines
2.3 KiB
C++
71 lines
2.3 KiB
C++
//===- TestInlining.cpp - Pass to inline calls in the test dialect --------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// TODO: This pass is only necessary because the main inlining pass
|
|
// has no abstracted away the call+callee relationship. When the inlining
|
|
// interface has this support, this pass should be removed.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "TestDialect.h"
|
|
#include "mlir/Dialect/StandardOps/IR/Ops.h"
|
|
#include "mlir/IR/BlockAndValueMapping.h"
|
|
#include "mlir/IR/Function.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
#include "mlir/Transforms/InliningUtils.h"
|
|
#include "mlir/Transforms/Passes.h"
|
|
#include "llvm/ADT/StringSet.h"
|
|
|
|
using namespace mlir;
|
|
using namespace mlir::test;
|
|
|
|
namespace {
|
|
struct Inliner : public PassWrapper<Inliner, FunctionPass> {
|
|
void runOnFunction() override {
|
|
auto function = getFunction();
|
|
|
|
// Collect each of the direct function calls within the module.
|
|
SmallVector<CallIndirectOp, 16> callers;
|
|
function.walk([&](CallIndirectOp caller) { callers.push_back(caller); });
|
|
|
|
// Build the inliner interface.
|
|
InlinerInterface interface(&getContext());
|
|
|
|
// Try to inline each of the call operations.
|
|
for (auto caller : callers) {
|
|
auto callee = dyn_cast_or_null<FunctionalRegionOp>(
|
|
caller.getCallee().getDefiningOp());
|
|
if (!callee)
|
|
continue;
|
|
|
|
// Inline the functional region operation, but only clone the internal
|
|
// region if there is more than one use.
|
|
if (failed(inlineRegion(
|
|
interface, &callee.body(), caller, caller.getArgOperands(),
|
|
caller.getResults(), caller.getLoc(),
|
|
/*shouldCloneInlinedRegion=*/!callee.getResult().hasOneUse())))
|
|
continue;
|
|
|
|
// If the inlining was successful then erase the call and callee if
|
|
// possible.
|
|
caller.erase();
|
|
if (callee.use_empty())
|
|
callee.erase();
|
|
}
|
|
}
|
|
};
|
|
} // end anonymous namespace
|
|
|
|
namespace mlir {
|
|
namespace test {
|
|
void registerInliner() {
|
|
PassRegistration<Inliner>("test-inline", "Test inlining region calls");
|
|
}
|
|
} // namespace test
|
|
} // namespace mlir
|