From 575568de4166bf69e0a5bc68978580afbe936878 Mon Sep 17 00:00:00 2001
From: Aart Bik <39774503+aartbik@users.noreply.github.com>
Date: Tue, 23 Jan 2024 14:51:46 -0800
Subject: [mlir][sparse] adjust compression scheme for example (#79212)

---
 mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_block3d.mlir | 6 +++---
 1 file 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 df12a6e..024e86b 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
-- 
cgit v1.1