diff options
Diffstat (limited to 'mlir/lib/Transforms/LoopFusion.cpp')
| -rw-r--r-- | mlir/lib/Transforms/LoopFusion.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/mlir/lib/Transforms/LoopFusion.cpp b/mlir/lib/Transforms/LoopFusion.cpp index 239915b1d4b..94d763fcbd1 100644 --- a/mlir/lib/Transforms/LoopFusion.cpp +++ b/mlir/lib/Transforms/LoopFusion.cpp @@ -1177,9 +1177,9 @@ public: // Skip if 'srcNode' is not a loop nest. if (!isa<ForInst>(srcNode->inst)) continue; - - // Skip if 'srcNode' has more than one store to 'memref'. - if (srcNode->getStoreOpCount(memref) != 1) + // Skip if 'srcNode' has more than one store to any memref. + // TODO(andydavis) Support fusing multi-output src loop nests. + if (srcNode->stores.size() != 1) continue; // Skip 'srcNode' if it has in dependence edges. NOTE: This is overly |
