================
@@ -586,30 +586,56 @@ Attribute SparseTensorEncodingAttr::parse(AsmParser
&parser, Type type) {
}
void SparseTensorEncodingAttr::print(AsmPrinter &printer) const {
- // Print the struct-like storage in dictionary fashion.
- printer << "<{ lvlTypes = [ ";
- llvm::interleaveComma(getLvlTypes(), printer, [&](DimLevelType dlt) {
- printer << "\"" << toMLIRString(dlt) << "\"";
- });
- printer << " ]";
+ auto map = static_cast<AffineMap>(getDimToLvl());
+ auto lvlTypes = getLvlTypes();
+ // Empty affine map indicates identity map
+ if (!map) {
+ map = AffineMap::getMultiDimIdentityMap(getLvlTypes().size(),
getContext());
+ }
+ // Modified version of AsmPrinter::Impl::printAffineMap.
----------------
aartbik wrote:
I would remove this. This is diverged sufficiently to no longer refer to
printAffineMap
https://github.com/llvm/llvm-project/pull/68130
_______________________________________________
cfe-commits mailing list
cfe-commits@lists.llvm.org
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
