commit 32a44bd58cc5f533ca7c327cd43b115dfe9113ad
author Han-Chung Wang <hanhan0912@gmail.com> Thu Sep 26 15:09:31 2024 -0700
committer GitHub <noreply@github.com> Thu Sep 26 22:09:31 2024 +0000
tree 05e6dc64a41f13f88d3f6c9ab71579e376d29c8a
parent 7db91cef123a068960b4edb54424d8dbfa54d613

[DispatchCreation][NFC] Simplify the logic of dumping CollapseInfo. (#18610)

It uses llvm::interleaveComma to simplify the code. Also, it replaces
the label with `CollapseDims` for consistency.

Signed-off-by: hanhanW <hanhan0912@gmail.com>

compiler/src/iree/compiler/DispatchCreation/CollapseDimensions.cpp

diff --git a/compiler/src/iree/compiler/DispatchCreation/CollapseDimensions.cpp b/compiler/src/iree/compiler/DispatchCreation/CollapseDimensions.cpp
index f0b0322..1fd5a7e 100644
--- a/compiler/src/iree/compiler/DispatchCreation/CollapseDimensions.cpp
+++ b/compiler/src/iree/compiler/DispatchCreation/CollapseDimensions.cpp
@@ -431,35 +431,20 @@
 }
 
 void CollapseInfo::print(raw_ostream &os) const {
-  os << "[CollapseDimensions] CollapseInfo:\n";
+  os << "[CollapseDims] CollapseInfo:\n";
 
   os << "Reassociation: ";
   os << "[";
   for (auto &vec : reassociation) {
     os << "[";
-    bool first = true;
-    for (auto elem : vec) {
-      if (!first) {
-        os << ", ";
-      }
-      first = false;
-      os << elem;
-    }
+    llvm::interleaveComma(vec, os);
     os << "]";
   }
   os << "]";
   os << "\n";
 
-  os << "Collapsable: ";
-  os << "{";
-  bool first = true;
-  for (auto elem : collapsableLoops) {
-    if (!first) {
-      os << ", ";
-    }
-    first = false;
-    os << elem;
-  }
+  os << "Collapsable: {";
+  llvm::interleaveComma(collapsableLoops, os);
   os << "}";
 }