[Mlir-commits] [mlir] 794d347 - [mlir][sparse] Fixing bug in python test

[wren romano]

Author: wren romano
Date: 2022-10-05T18:06:22-07:00
New Revision: 794d347988b9c6e4e44c59b7045bd1c6225cddc0

URL: https://github.com/llvm/llvm-project/commit/794d347988b9c6e4e44c59b7045bd1c6225cddc0
DIFF: https://github.com/llvm/llvm-project/commit/794d347988b9c6e4e44c59b7045bd1c6225cddc0.diff

LOG: [mlir][sparse] Fixing bug in python test

This is a followup to D135004, to correct one of the tests that didn't get caught by the buildbot.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D135336

Added: 
    

Modified: 
    mlir/test/python/dialects/sparse_tensor/dialect.py

Removed: 
    


################################################################################
diff  --git a/mlir/test/python/dialects/sparse_tensor/dialect.py b/mlir/test/python/dialects/sparse_tensor/dialect.py
index a4f3359b667f0..e8e753df61878 100644
--- a/mlir/test/python/dialects/sparse_tensor/dialect.py
+++ b/mlir/test/python/dialects/sparse_tensor/dialect.py
@@ -25,7 +25,7 @@ def testEncodingAttr1D():
     # CHECK: equal: True
     print(f"equal: {casted == parsed}")
 
-    # CHECK: dim_level_types: [<DimLevelType.compressed: 1>]
+    # CHECK: dim_level_types: [<DimLevelType.compressed: 8>]
     print(f"dim_level_types: {casted.dim_level_types}")
     # CHECK: dim_ordering: None
     print(f"dim_ordering: {casted.dim_ordering}")
@@ -64,7 +64,7 @@ def testEncodingAttr2D():
     # CHECK: equal: True
     print(f"equal: {casted == parsed}")
 
-    # CHECK: dim_level_types: [<DimLevelType.dense: 0>, <DimLevelType.compressed: 1>]
+    # CHECK: dim_level_types: [<DimLevelType.dense: 4>, <DimLevelType.compressed: 8>]
     print(f"dim_level_types: {casted.dim_level_types}")
     # CHECK: dim_ordering: (d0, d1) -> (d1, d0)
     print(f"dim_ordering: {casted.dim_ordering}")