diff --git a/mlir/lib/IR/AsmPrinter.cpp b/mlir/lib/IR/AsmPrinter.cpp --- a/mlir/lib/IR/AsmPrinter.cpp +++ b/mlir/lib/IR/AsmPrinter.cpp @@ -158,7 +158,8 @@ /// Return if the given ElementsAttr should be elided. bool OpPrintingFlags::shouldElideElementsAttr(ElementsAttr attr) const { return elementsAttrElementLimit.hasValue() && - *elementsAttrElementLimit < int64_t(attr.getNumElements()); + *elementsAttrElementLimit < int64_t(attr.getNumElements()) && + !attr.isa(); } /// Return the size limit for printing large ElementsAttr. diff --git a/mlir/test/IR/pretty-attributes.mlir b/mlir/test/IR/pretty-attributes.mlir --- a/mlir/test/IR/pretty-attributes.mlir +++ b/mlir/test/IR/pretty-attributes.mlir @@ -16,3 +16,6 @@ // CHECK: opaque<"", "0xDEADBEEF"> : tensor<100xf32> "test.opaque_attr"() {foo.opaque_attr = opaque<"", "0xEBFE"> : tensor<100xf32> } : () -> () + +// CHECK: dense<1> : tensor<3xi32> +"test.dense_splat"() {foo.dense_attr = dense<1> : tensor<3xi32>} : () -> ()