[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