Magic-Mirror/llvm-capstone
1
0
Fork 0
mirror of https://github.com/capstone-engine/llvm-capstone.git synced 2024-10-07 19:03:57 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[mlir][sparse] (NFC) Reordering extraClassDeclaration for STEA

Browse Source 
Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D151171
This commit is contained in:
wren romano 2023-05-22 16:07:48 -07:00
parent fa2827f079
commit 54c87f216c
1 changed files with 41 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorAttrDefs.td
View File

@ -293,15 +293,9 @@ def SparseTensorEncodingAttr : SparseTensor_Attr<"SparseTensorEncoding",
];
let extraClassDeclaration = [{
/// Returns the type for position storage based on posWidth.
/// Asserts that the encoding is non-null (since there's nowhere
/// to get the `MLIRContext` from).
Type getPosType() const;
/// Returns the type for coordinate storage based on crdWidth.
/// Asserts that the encoding is non-null (since there's nowhere
/// to get the `MLIRContext` from).
Type getCrdType() const;
//
// Factory methods.
//
/// Constructs a new encoding with the dimOrdering and higherOrdering
/// reset to the default/identity.
@ -311,18 +305,9 @@ def SparseTensorEncodingAttr : SparseTensor_Attr<"SparseTensorEncoding",
/// reset to the default.
SparseTensorEncodingAttr withoutBitWidths() const;
/// Returns true if every level is dense. Also returns true for
/// the null encoding (since dense-tensors are always all-dense).
bool isAllDense() const;
/// Returns true if every level is ordered. Also returns true for
/// the null encoding (since dense-tensors are always all-ordered).
bool isAllOrdered() const;
/// Returns true if the encoding has an identity dimension ordering.
/// Also returns true for the null encoding (since dense-tensors
/// always have the identity ordering).
bool hasIdDimOrdering() const;
//
// lvlTypes methods.
//
/// Returns the number of storage levels. Asserts that the encoding
/// is non-null (since there is no fixed result that's valid for
@ -341,6 +326,41 @@ def SparseTensorEncodingAttr : SparseTensor_Attr<"SparseTensorEncoding",
bool isOrderedLvl(::mlir::sparse_tensor::Level l) const { return isOrderedDLT(getLvlType(l)); }
bool isUniqueLvl(::mlir::sparse_tensor::Level l) const { return isUniqueDLT(getLvlType(l)); }
/// Returns true if every level is dense. Also returns true for
/// the null encoding (since dense-tensors are always all-dense).
bool isAllDense() const;
/// Returns true if every level is ordered. Also returns true for
/// the null encoding (since dense-tensors are always all-ordered).
bool isAllOrdered() const;
//
// dimOrdering/higherOrdering methods.
//
/// Returns true if the encoding has an identity dimension ordering.
/// Also returns true for the null encoding (since dense-tensors
/// always have the identity ordering).
bool hasIdDimOrdering() const;
//
// posWidth/crdWidth methods.
//
/// Returns the type for position storage based on posWidth.
/// Asserts that the encoding is non-null (since there's nowhere
/// to get the `MLIRContext` from).
Type getPosType() const;
/// Returns the type for coordinate storage based on crdWidth.
/// Asserts that the encoding is non-null (since there's nowhere
/// to get the `MLIRContext` from).
Type getCrdType() const;
//
// dimSlices methods.
//
bool isSlice() const;
::mlir::sparse_tensor::SparseTensorDimSliceAttr getDimSlice(::mlir::sparse_tensor::Dimension dim) const;