@@ -446,10 +446,13 @@ def SparseTensor_StorageSpecifierInitOp : SparseTensor_Op<"storage_specifier.ini
446
446
Example:
447
447
448
448
```mlir
449
- #CSR = #sparse_tensor.encoding<{ dimLevelType = [ "dense", "compressed" ]}>
449
+ #CSR = #sparse_tensor.encoding<{
450
+ map = (i, j) -> (i : dense, j : compressed)
451
+ }>
450
452
#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)
453
456
}>
454
457
455
458
%0 = sparse_tensor.storage_specifier.init : !sparse_tensor.storage_specifier<#CSR>
@@ -1260,8 +1263,7 @@ def SparseTensor_ForeachOp : SparseTensor_Op<"foreach",
1260
1263
For example:
1261
1264
```mlir
1262
1265
#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)
1265
1267
}>
1266
1268
1267
1269
// foreach on a column-major sparse tensor
@@ -1271,7 +1273,7 @@ def SparseTensor_ForeachOp : SparseTensor_Op<"foreach",
1271
1273
}
1272
1274
1273
1275
#ROW_MAJOR = #sparse_tensor.encoding<{
1274
- dimLevelType = [ " compressed", "compressed" ],
1276
+ map = (d0, d1) -> (d0 : compressed, d1 : compressed)
1275
1277
}>
1276
1278
1277
1279
// foreach on a row-major sparse tensor
0 commit comments