Skip to content

Commit 6c6749c

Browse files
[mlir][sparse] Update examples in Ops.td (llvm#69499)
Example: dimLevelType = [ "compressed", "compressed" ] to map = (d0, d1) -> (d0 : compressed, d1 : compressed)
1 parent 81d651e commit 6c6749c

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -446,10 +446,13 @@ def SparseTensor_StorageSpecifierInitOp : SparseTensor_Op<"storage_specifier.ini
446446
Example:
447447

448448
```mlir
449-
#CSR = #sparse_tensor.encoding<{ dimLevelType = [ "dense", "compressed" ]}>
449+
#CSR = #sparse_tensor.encoding<{
450+
map = (i, j) -> (i : dense, j : compressed)
451+
}>
450452
#CSR_SLICE = #sparse_tensor.encoding<{
451-
dimLevelType = [ "dense", "compressed" ],
452-
slice = [ (1, 4, 1), (1, 4, 2) ]
453+
map = (d0 : #sparse_tensor<slice(1, 4, 1)>,
454+
d1 : #sparse_tensor<slice(1, 4, 2)>) ->
455+
(d0 : dense, d1 : compressed)
453456
}>
454457

455458
%0 = sparse_tensor.storage_specifier.init : !sparse_tensor.storage_specifier<#CSR>
@@ -1260,8 +1263,7 @@ def SparseTensor_ForeachOp : SparseTensor_Op<"foreach",
12601263
For example:
12611264
```mlir
12621265
#COL_MAJOR = #sparse_tensor.encoding<{
1263-
dimLevelType = [ "compressed", "compressed" ],
1264-
dimOrdering = affine_map<(i,j) -> (j,i)>
1266+
map = (d0, d1) -> (d1 : compressed, d0 : compressed)
12651267
}>
12661268

12671269
// foreach on a column-major sparse tensor
@@ -1271,7 +1273,7 @@ def SparseTensor_ForeachOp : SparseTensor_Op<"foreach",
12711273
}
12721274

12731275
#ROW_MAJOR = #sparse_tensor.encoding<{
1274-
dimLevelType = [ "compressed", "compressed" ],
1276+
map = (d0, d1) -> (d0 : compressed, d1 : compressed)
12751277
}>
12761278

12771279
// foreach on a row-major sparse tensor

0 commit comments

Comments
 (0)