2c1ae801e1
This patch adds more precise side effects to the current ops with memory effects, allowing us to determine which OpOperand/OpResult/BlockArgument the operation reads or writes, rather than just recording the reading and writing of values. This allows for convenient use of precise side effects to achieve analysis and optimization. Related discussions: https://discourse.llvm.org/t/rfc-add-operandindex-to-sideeffect-instance/79243
88 lines
2.9 KiB
C++
88 lines
2.9 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 (instance.getEffectValue<OpOperand *>())
|
|
diag << " on a op operand,";
|
|
else if (instance.getEffectValue<OpResult>())
|
|
diag << " on a op result,";
|
|
else if (instance.getEffectValue<BlockArgument>())
|
|
diag << " on a block argument,";
|
|
} 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
|