intel/llvm
1
0
Fork 0
mirror of https://github.com/intel/llvm.git synced 2026-01-18 16:50:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

[mlir][sparse] support sparse tensor element type conversion in codegen path

Browse Source 
Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D144578
This commit is contained in:
Peiming Liu
2023-02-22 19:04:02 +00:00
parent 230e61658b
commit 85dbb3fc4b
8 changed files with 265 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -135,6 +135,12 @@ SparseTensorEncodingAttr SparseTensorEncodingAttr::withoutOrdering() const {
getPointerBitWidth(), getIndexBitWidth());
}
SparseTensorEncodingAttr SparseTensorEncodingAttr::withoutBitWidths() const {
return SparseTensorEncodingAttr::get(getContext(), getDimLevelType(),
getDimOrdering(), getHigherOrdering(), 0,
0);
}
bool SparseTensorEncodingAttr::isAllDense() const {
return !getImpl() || llvm::all_of(getDimLevelType(), isDenseDLT);
}