diff options
Diffstat (limited to 'mlir/lib/Transforms/LoopFusion.cpp')
| -rw-r--r-- | mlir/lib/Transforms/LoopFusion.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/mlir/lib/Transforms/LoopFusion.cpp b/mlir/lib/Transforms/LoopFusion.cpp index e13616e..d6d26a8 100644 --- a/mlir/lib/Transforms/LoopFusion.cpp +++ b/mlir/lib/Transforms/LoopFusion.cpp @@ -768,7 +768,8 @@ bool MemRefDependenceGraph::init(FuncOp f) { SmallVector<MemoryEffects::EffectInstance, 1> effects; effectInterface.getEffects(effects); if (llvm::any_of(effects, [](const MemoryEffects::EffectInstance &it) { - return isa<MemoryEffects::Write>(it.getEffect()); + return isa<MemoryEffects::Write, MemoryEffects::Free>( + it.getEffect()); })) { Node node(nextNodeId++, &op); nodes.insert({node.id, node}); |