maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[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
This commit is contained in:
wren romano 2022-10-05 17:59:34 -07:00
parent 25ee36c6b1
commit 794d347988
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mlir/test/python/dialects/sparse_tensor/dialect.py
View File

@ -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}")