Merge pull request #27 from zincnode/fix

[fix] Python Bindings LIT
zincnode · Nov 5, 2024 · ec16b79 · ec16b79
2 parents 4eb8136 + 3ac970c
commit ec16b79
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/test/python/dialects/affine.py b/test/python/dialects/affine.py
@@ -47,12 +47,10 @@ def affine_store_test(arg0):
 # CHECK-LABEL: TEST: testAffineDelinearizeInfer
 @constructAndPrintInModule
 def testAffineDelinearizeInfer():
-    # CHECK: %[[C0:.*]] = arith.constant 0 : index
-    c0 = arith.ConstantOp(T.index(), 0)
     # CHECK: %[[C1:.*]] = arith.constant 1 : index
     c1 = arith.ConstantOp(T.index(), 1)
-    # CHECK: %{{.*}}:2 = affine.delinearize_index %[[C1:.*]] into (%[[C1:.*]], %[[C0:.*]]) : index, index
-    two_indices = affine.AffineDelinearizeIndexOp(c1, [c1, c0])
+    # CHECK: %{{.*}}:2 = affine.delinearize_index %[[C1:.*]] into (2, 3) : index, index
+    two_indices = affine.AffineDelinearizeIndexOp(c1, [], [2, 3])
 
 
 # CHECK-LABEL: TEST: testAffineLoadOp
@@ -159,7 +157,7 @@ def testAffineForOpErrors():
         )
 
     try:
-        two_indices = affine.AffineDelinearizeIndexOp(c1, [c1, c1])
+        two_indices = affine.AffineDelinearizeIndexOp(c1, [], [1, 1])
         affine.AffineForOp(
             two_indices,
             c2,
@@ -265,3 +263,4 @@ def range_loop_8(lb, ub, memref_v):
             add = arith.addi(i, i)
             memref.store(add, it, [i])
             affine.yield_([it])
+