Also elide large array attribute in OpGraph Dump

PiperOrigin-RevId: 281114034

diff --git a/mlir/lib/Transforms/ViewOpGraph.cpp b/mlir/lib/Transforms/ViewOpGraph.cpp
index f080ccd..503a82b 100644 (file)
--- a/mlir/lib/Transforms/ViewOpGraph.cpp
+++ b/mlir/lib/Transforms/ViewOpGraph.cpp
@@ -96,6 +96,12 @@ std::string DOTGraphTraits<mlir::Block *>::getNodeLabel(mlir::Operation *op,
       continue;
     }
 
+    auto array = attr.second.dyn_cast<mlir::ArrayAttr>();
+    if (array && static_cast<int64_t>(array.size()) > elideIfLarger) {
+      os << "[...]";
+      continue;
+    }
+
     // Print all other attributes.
     attr.second.print(os);
   }