From 3ac970cb1de4640b2ecb7d1de6d9d69b4a07190d Mon Sep 17 00:00:00 2001 From: zincnode Date: Tue, 5 Nov 2024 23:30:43 +0800 Subject: [PATCH] [fix] Python Bindings LIT - Fix Python Bindings LIT based on upstream update - Reference: https://github.com/llvm/llvm-project/pull/113846 --- test/python/dialects/affine.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/test/python/dialects/affine.py b/test/python/dialects/affine.py index 832ec37..344336a 100644 --- 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]) +