[Mlir-commits] [mlir] 575568d - [mlir][sparse] adjust compression scheme for example (#79212)

[llvmlistbot at llvm.org]

Author: Aart Bik
Date: 2024-01-23T14:51:46-08:00
New Revision: 575568de4166bf69e0a5bc68978580afbe936878

URL: https://github.com/llvm/llvm-project/commit/575568de4166bf69e0a5bc68978580afbe936878
DIFF: https://github.com/llvm/llvm-project/commit/575568de4166bf69e0a5bc68978580afbe936878.diff

LOG: [mlir][sparse] adjust compression scheme for example (#79212)

Added: 
    

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

Removed: 
    


################################################################################
diff  --git a/mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_block3d.mlir b/mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_block3d.mlir
index df12a6e042dde0..024e86b4f165bb 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