Skip to content

Commit 575568d

Browse files
authored
[mlir][sparse] adjust compression scheme for example (#79212)
1 parent c28ab62 commit 575568d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_block3d.mlir

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -32,9 +32,9 @@
3232

3333
#Sparse1 = #sparse_tensor.encoding<{
3434
map = (i, j, k) -> (
35-
i : compressed,
3635
j : compressed,
37-
k : compressed
36+
k : compressed,
37+
i : dense
3838
)
3939
}>
4040

@@ -91,7 +91,7 @@ module {
9191
// dense “fibers” in the i-dim, we end up with 8 stored entries.
9292
//
9393
// CHECK: 8
94-
// CHECK-NEXT: ( 1, 5, 2, 6, 3, 7, 4, 8 )
94+
// CHECK-NEXT: ( 1, 2, 3, 4, 5, 6, 7, 8 )
9595
//
9696
%na = sparse_tensor.number_of_entries %a : tensor<4x4x4xi32, #Sparse1>
9797
vector.print %na : index

0 commit comments

Comments
 (0)