@@ -47,12 +47,10 @@ def affine_store_test(arg0):
4747# CHECK-LABEL: TEST: testAffineDelinearizeInfer
4848@constructAndPrintInModule
4949def testAffineDelinearizeInfer ():
50- # CHECK: %[[C0:.*]] = arith.constant 0 : index
51- c0 = arith .ConstantOp (T .index (), 0 )
5250 # CHECK: %[[C1:.*]] = arith.constant 1 : index
5351 c1 = arith .ConstantOp (T .index (), 1 )
54- # CHECK: %{{.*}}:2 = affine.delinearize_index %[[C1:.*]] into (%[[C1:.*]], %[[C0:.*]] ) : index, index
55- two_indices = affine .AffineDelinearizeIndexOp (c1 , [c1 , c0 ])
52+ # CHECK: %{{.*}}:2 = affine.delinearize_index %[[C1:.*]] into (2, 3 ) : index, index
53+ two_indices = affine .AffineDelinearizeIndexOp (c1 , [], [ 2 , 3 ])
5654
5755
5856# CHECK-LABEL: TEST: testAffineLoadOp
@@ -159,7 +157,7 @@ def testAffineForOpErrors():
159157 )
160158
161159 try :
162- two_indices = affine .AffineDelinearizeIndexOp (c1 , [c1 , c1 ])
160+ two_indices = affine .AffineDelinearizeIndexOp (c1 , [], [ 1 , 1 ])
163161 affine .AffineForOp (
164162 two_indices ,
165163 c2 ,
@@ -265,3 +263,4 @@ def range_loop_8(lb, ub, memref_v):
265263 add = arith .addi (i , i )
266264 memref .store (add , it , [i ])
267265 affine .yield_ ([it ])
266+
0 commit comments