[Mlir-commits] [mlir] 6c8a884 - [mlir] fix python test

[Alex Zinenko]

Author: Alex Zinenko
Date: 2023-01-19T08:58:34Z
New Revision: 6c8a884707e573eec66bb8e512aeaec5aeb8ab8e

URL: https://github.com/llvm/llvm-project/commit/6c8a884707e573eec66bb8e512aeaec5aeb8ab8e
DIFF: https://github.com/llvm/llvm-project/commit/6c8a884707e573eec66bb8e512aeaec5aeb8ab8e.diff

LOG: [mlir] fix python test

It was using an incorrect attribute type, but the test was still passing
because of the value being present in the output.

Added: 
    

Modified: 
    mlir/python/mlir/dialects/_structured_transform_ops_ext.py
    mlir/test/python/dialects/transform_structured_ext.py

Removed: 
    


################################################################################
diff  --git a/mlir/python/mlir/dialects/_structured_transform_ops_ext.py b/mlir/python/mlir/dialects/_structured_transform_ops_ext.py
index 5fd5cfe1073ad..2525ea34c375c 100644
--- a/mlir/python/mlir/dialects/_structured_transform_ops_ext.py
+++ b/mlir/python/mlir/dialects/_structured_transform_ops_ext.py
@@ -110,7 +110,7 @@ def __init__(self,
                loc=None,
                ip=None):
     pdl_operation_type = pdl.OperationType.get()
-    interchange_attr = _get_int_array_attr(iterator_interchange)
+    interchange_attr = _get_dense_int64_array_attr(iterator_interchange)
     super().__init__(
         pdl_operation_type,
         _get_op_result_or_value(target),

diff  --git a/mlir/test/python/dialects/transform_structured_ext.py b/mlir/test/python/dialects/transform_structured_ext.py
index 34c86a317920b..b88f7465e08b8 100644
--- a/mlir/test/python/dialects/transform_structured_ext.py
+++ b/mlir/test/python/dialects/transform_structured_ext.py
@@ -44,9 +44,7 @@ def testInterchange():
   with InsertionPoint(sequence.body):
     structured.InterchangeOp(
         sequence.bodyTarget,
-        iterator_interchange=[
-            IntegerAttr.get(IntegerType.get_signless(64), 1), 0
-        ])
+        iterator_interchange=[1, 0])
     transform.YieldOp()
   # CHECK-LABEL: TEST: testInterchange
   # CHECK: transform.sequence