850b2c6b3c
The current implementation of takeBody first clears the Region, before then taking ownership of the blocks of the other regions. The issue here however, is that when clearing the region, it does not take into account references of operations to each other. In particular, blocks are deleted from front to back, and operations within a block are very likely to be deleted despite still having uses, causing an assertion to trigger [0].
This patch fixes that issue by simply calling dropAllReferences()before clearing the blocks.
[0] 9a8bb4bc63/mlir/lib/IR/Operation.cpp (L154)
Differential Revision: https://reviews.llvm.org/D123913
46 lines
1.4 KiB
C++
46 lines
1.4 KiB
C++
//===- TestRegions.cpp - Pass to test Region's methods --------------------===//
|
|
//
|
|
// 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/Func/IR/FuncOps.h"
|
|
#include "mlir/IR/BuiltinOps.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
|
|
using namespace mlir;
|
|
|
|
namespace {
|
|
/// This is a test pass that tests Region's takeBody method by making the first
|
|
/// function take the body of the second.
|
|
struct TakeBodyPass
|
|
: public PassWrapper<TakeBodyPass, OperationPass<ModuleOp>> {
|
|
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(TakeBodyPass)
|
|
|
|
StringRef getArgument() const final { return "test-take-body"; }
|
|
StringRef getDescription() const final { return "Test Region's takeBody"; }
|
|
|
|
void runOnOperation() override {
|
|
auto module = getOperation();
|
|
|
|
SmallVector<func::FuncOp> functions =
|
|
llvm::to_vector(module.getOps<func::FuncOp>());
|
|
if (functions.size() != 2) {
|
|
module.emitError("Expected only two functions in test");
|
|
signalPassFailure();
|
|
return;
|
|
}
|
|
|
|
functions[0].getBody().takeBody(functions[1].getBody());
|
|
}
|
|
};
|
|
|
|
} // namespace
|
|
|
|
namespace mlir {
|
|
void registerRegionTestPasses() { PassRegistration<TakeBodyPass>(); }
|
|
} // namespace mlir
|