caio/clang-p2996
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[mlir][sparse] minor code changes

Browse Source 
Submitting for Wren

Reviewed By: K-Wu

Differential Revision: https://reviews.llvm.org/D153804
This commit is contained in:
Aart Bik
2023-06-26 12:09:33 -07:00
parent 19e9b9b589
commit 11a4f5bdfb
3 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp
View File

@@ -209,14 +209,18 @@ std::string SparseTensorDimSliceAttr::getStaticString(int64_t v) {
return isDynamic(v) ? "?" : std::to_string(v);
}
void SparseTensorDimSliceAttr::print(llvm::raw_ostream &os) const {
os << '(';
os << getStaticString(getOffset());
os << ", ";
os << getStaticString(getSize());
os << ", ";
os << getStaticString(getStride());
os << ')';
}
void SparseTensorDimSliceAttr::print(AsmPrinter &printer) const {
printer << "(";
printer << getStaticString(getOffset());
printer << ", ";
printer << getStaticString(getSize());
printer << ", ";
printer << getStaticString(getStride());
printer << ")";
print(printer.getStream());
}
static ParseResult parseOptionalStaticSlice(int64_t &result,