diff options
author | Aart Bik <39774503+aartbik@users.noreply.github.com> | 2024-01-23 14:51:46 -0800 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-01-23 14:51:46 -0800 |
commit | 575568de4166bf69e0a5bc68978580afbe936878 (patch) | |
tree | aedbbdf1ee37859b8cc50ac21d21dc2183394fac | |
parent | c28ab6274322da693ea41a49e301c15b551c974a (diff) |
[mlir][sparse] adjust compression scheme for example (#79212)
-rwxr-xr-x | mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_block3d.mlir | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_block3d.mlir b/mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_block3d.mlir index df12a6e042dd..024e86b4f165 100755 --- a/mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_block3d.mlir +++ b/mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_block3d.mlir @@ -32,9 +32,9 @@ #Sparse1 = #sparse_tensor.encoding<{ map = (i, j, k) -> ( - i : compressed, j : compressed, - k : compressed + k : compressed, + i : dense ) }> @@ -91,7 +91,7 @@ module { // dense “fibers” in the i-dim, we end up with 8 stored entries. // // CHECK: 8 - // CHECK-NEXT: ( 1, 5, 2, 6, 3, 7, 4, 8 ) + // CHECK-NEXT: ( 1, 2, 3, 4, 5, 6, 7, 8 ) // %na = sparse_tensor.number_of_entries %a : tensor<4x4x4xi32, #Sparse1> vector.print %na : index |