[Mlir-commits] [mlir] [mlir][linalg] Add more precise memory effects to linalg op (PR #92079)
donald chen
llvmlistbot at llvm.org
Tue May 14 05:49:50 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()) {
----------------
cxy-1993 wrote:
Because only linalg.softmax does not have the LinalgOp Interface...
https://github.com/llvm/llvm-project/pull/92079
More information about the Mlir-commits
mailing list