[Mlir-commits] [mlir] [mlir][linalg] Add more precise memory effects to linalg op (PR #92079)
Oleksandr Alex Zinenko
llvmlistbot at llvm.org
Fri May 17 04:52:27 PDT 2024
================
@@ -2494,8 +2496,20 @@ SoftmaxOp::reifyResultShapes(OpBuilder &b,
void SoftmaxOp::getEffects(
SmallVectorImpl<SideEffects::EffectInstance<MemoryEffects::Effect>>
&effects) {
- getGenericEffectsImpl(effects, getOperation()->getResults(), getDpsInputs(),
- getDpsInits());
+ for (auto operand : getDpsInputs()) {
+ if (!llvm::isa<MemRefType>(operand.getType()))
+ continue;
+ effects.emplace_back(MemoryEffects::Read::get(), operand,
+ SideEffects::DefaultResource::get());
+ }
+ for (auto operand : getDpsInits()) {
----------------
ftynse wrote:
Nit: please expand auto unless the type is obvious from the line context.
https://github.com/llvm/llvm-project/pull/92079
More information about the Mlir-commits
mailing list