commit 2eaadfc4fe662e1ba312929cd650ed349fe7f2b4
author Rahul Joshi <jurahul@google.com> Wed Jun 17 20:20:36 2020
committer Rahul Joshi <jurahul@google.com> Wed Jun 17 20:26:10 2020
tree b6ccdc68be1a08c8cf242f5036eaaf3c9bc0d638
parent 6c45532908bacc0cf569bcad74f6e04e1e2a6c61

[NFC] Use llvm::hasSingleElement() in place of .size() == 1

- Also use functions in Region instead of Region::getBlocks() where possible.

Differential Revision: https://reviews.llvm.org/D82032

mlir/lib/Transforms/LoopFusion.cpp

diff --git a/mlir/lib/Transforms/LoopFusion.cpp b/mlir/lib/Transforms/LoopFusion.cpp
index c3b8912..0ae6657 100644
--- a/mlir/lib/Transforms/LoopFusion.cpp
+++ b/mlir/lib/Transforms/LoopFusion.cpp
@@ -636,7 +636,7 @@
   DenseMap<Value, SetVector<unsigned>> memrefAccesses;
 
   // TODO: support multi-block functions.
-  if (f.getBlocks().size() != 1)
+  if (!llvm::hasSingleElement(f))
     return false;
 
   DenseMap<Operation *, unsigned> forToNodeMap;