From 78f57e34a25367ef3192cd35da36b01c763f4ecf Mon Sep 17 00:00:00 2001
From: Gunhyun Park <gunhyun@google.com>
Date: Mon, 9 Oct 2023 15:48:13 -0700
Subject: [PATCH] Integrate remainder of llvm/llvm-project@2a1f1b5fde0a (#1798)

---
 stablehlo/tests/print_stablehlo.mlir | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/stablehlo/tests/print_stablehlo.mlir b/stablehlo/tests/print_stablehlo.mlir
index b2f54b0443b..a49ce86b16b 100644
--- a/stablehlo/tests/print_stablehlo.mlir
+++ b/stablehlo/tests/print_stablehlo.mlir
@@ -292,11 +292,11 @@ func.func @extensions(%arg0 : tensor<?x?xf32, #stablehlo.bounds<3, ?>>,
 }
 
 #CSR = #sparse_tensor.encoding<{
-  lvlTypes = ["dense", "compressed"]
+  map = (d0, d1) -> (d0 : dense, d1 : compressed)
 }>
 
 #DCSR = #sparse_tensor.encoding<{
-  lvlTypes = ["compressed", "compressed"]
+  map = (d0, d1) -> (d0 : compressed, d1 : compressed)
 }>
 
 // CHECK-LABEL: func @encodings