[Mlir-commits] [mlir] [mlir][sparse] support sparsifying batch levels (PR #83898)
Aart Bik
llvmlistbot at llvm.org
Mon Mar 4 14:11:05 PST 2024
================
@@ -253,7 +253,10 @@ class SparseTensorType {
CrdTransDirectionKind::dim2lvl);
}
- /// Returns the Level-shape.
+ /// Returns the batched level rank.
+ unsigned getBatchLvlRank() const { return getEncoding().getBatchLvlRank(); }
+
+ /// Returns the batched Level-shape.
----------------
aartbik wrote:
for consistency level-shape (lower case L)
also, L250 should probably be changed to level-shape
for consisteny
https://github.com/llvm/llvm-project/pull/83898
More information about the Mlir-commits
mailing list