abbbe4a6cd
This fixes an issue with `TransferWriteOp`'s implementation of the `MemoryEffectOpInterface` where the write effect was attached to the stored value rather than the base. This had the effect that when asking for the memory effects for the input memref buffer using `getEffectsOnValue(...)`, the function would return no-effects (as the effect would have been attached to the stored value rather than the input buffer).
88 lines
3.0 KiB
C++
88 lines
3.0 KiB
C++
//===- TestSidEffects.cpp - Pass to test side effects ---------------------===//
|
|
//
|
|
// 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 "TestOps.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
|
|
using namespace mlir;
|
|
|
|
namespace {
|
|
struct SideEffectsPass
|
|
: public PassWrapper<SideEffectsPass, OperationPass<ModuleOp>> {
|
|
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(SideEffectsPass)
|
|
|
|
StringRef getArgument() const final { return "test-side-effects"; }
|
|
StringRef getDescription() const final {
|
|
return "Test side effects interfaces";
|
|
}
|
|
void runOnOperation() override {
|
|
auto module = getOperation();
|
|
|
|
// Walk operations detecting side effects.
|
|
SmallVector<MemoryEffects::EffectInstance, 8> effects;
|
|
module.walk([&](MemoryEffectOpInterface op) {
|
|
effects.clear();
|
|
op.getEffects(effects);
|
|
|
|
if (op->hasTrait<OpTrait::IsTerminator>()) {
|
|
return;
|
|
}
|
|
|
|
// Check to see if this operation has any memory effects.
|
|
if (effects.empty()) {
|
|
op.emitRemark() << "operation has no memory effects";
|
|
return;
|
|
}
|
|
|
|
for (MemoryEffects::EffectInstance instance : effects) {
|
|
auto diag = op.emitRemark() << "found an instance of ";
|
|
|
|
if (isa<MemoryEffects::Allocate>(instance.getEffect()))
|
|
diag << "'allocate'";
|
|
else if (isa<MemoryEffects::Free>(instance.getEffect()))
|
|
diag << "'free'";
|
|
else if (isa<MemoryEffects::Read>(instance.getEffect()))
|
|
diag << "'read'";
|
|
else if (isa<MemoryEffects::Write>(instance.getEffect()))
|
|
diag << "'write'";
|
|
|
|
if (instance.getValue()) {
|
|
if (auto *opOpd = instance.getEffectValue<OpOperand *>())
|
|
diag << " on op operand " << opOpd->getOperandNumber() << ",";
|
|
else if (auto opRes = instance.getEffectValue<OpResult>())
|
|
diag << " on op result " << opRes.getResultNumber() << ",";
|
|
else if (auto opBlk = instance.getEffectValue<BlockArgument>())
|
|
diag << " on block argument " << opBlk.getArgNumber() << ",";
|
|
} else if (SymbolRefAttr symbolRef = instance.getSymbolRef())
|
|
diag << " on a symbol '" << symbolRef << "',";
|
|
|
|
diag << " on resource '" << instance.getResource()->getName() << "'";
|
|
}
|
|
});
|
|
|
|
SmallVector<TestEffects::EffectInstance, 1> testEffects;
|
|
module.walk([&](TestEffectOpInterface op) {
|
|
testEffects.clear();
|
|
op.getEffects(testEffects);
|
|
|
|
if (testEffects.empty())
|
|
return;
|
|
|
|
for (const TestEffects::EffectInstance &instance : testEffects) {
|
|
op.emitRemark() << "found a parametric effect with "
|
|
<< instance.getParameters();
|
|
}
|
|
});
|
|
}
|
|
};
|
|
} // namespace
|
|
|
|
namespace mlir {
|
|
void registerSideEffectTestPasses() { PassRegistration<SideEffectsPass>(); }
|
|
} // namespace mlir
|