diff options
| author | MLIR Team <no-reply@google.com> | 2019-01-23 11:11:43 -0800 |
|---|---|---|
| committer | jpienaar <jpienaar@google.com> | 2019-03-29 15:32:20 -0700 |
| commit | b28009b681a45a0e5b925ed90f33f8285c87e6af (patch) | |
| tree | 26cbc76068bdd27865d3ece9b3c471c2ac4b0fbf /mlir/lib/Transforms/LoopFusion.cpp | |
| parent | 864d9e02a17f02fb396e5ea047623ed4207153f1 (diff) | |
| download | bcm5719-llvm-b28009b681a45a0e5b925ed90f33f8285c87e6af.tar.gz bcm5719-llvm-b28009b681a45a0e5b925ed90f33f8285c87e6af.zip | |
Fix single producer check in loop fusion pass.
PiperOrigin-RevId: 230565482
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 |
