diff --git a/mlir/python/mlir/dialects/_structured_transform_ops_ext.py b/mlir/python/mlir/dialects/_structured_transform_ops_ext.py --- a/mlir/python/mlir/dialects/_structured_transform_ops_ext.py +++ b/mlir/python/mlir/dialects/_structured_transform_ops_ext.py @@ -462,7 +462,7 @@ class MultiTileSizesOp: - """Specialization for MultitileSizesOp class.""" + """Specialization for MultiTileSizesOp class.""" def __init__( self, @@ -475,8 +475,6 @@ loc=None, ip=None, ): - if divisor is None: - divisor = 1 super().__init__( result_type, result_type, diff --git a/mlir/test/python/dialects/transform_structured_ext.py b/mlir/test/python/dialects/transform_structured_ext.py --- a/mlir/test/python/dialects/transform_structured_ext.py +++ b/mlir/test/python/dialects/transform_structured_ext.py @@ -297,7 +297,7 @@ @run -def testMultitileSizes(): +def testMultitileSizesCompact(): sequence = transform.SequenceOp( transform.FailurePropagationMode.Propagate, [], pdl.OperationType.get() ) @@ -308,8 +308,34 @@ transform.YieldOp() # CHECK-LABEL: TEST: testMultitileSizes # CHECK: transform.sequence + # CHECK-NOT: divisor # CHECK: transform.structured.multitile_sizes + # CHECK-NOT: divisor # CHECK-DAG: dimension = 1 + # CHECK-NOT: divisor + # CHECK-DAG: target_size = 42 + # CHECK-NOT: divisor + + +@run +def testMultitileSizesAllArgs(): + sequence = transform.SequenceOp( + transform.FailurePropagationMode.Propagate, [], pdl.OperationType.get() + ) + with InsertionPoint(sequence.body): + structured.MultiTileSizesOp( + pdl.OperationType.get(), + sequence.bodyTarget, + dimension=1, + target_size=42, + divisor=2, + ) + transform.YieldOp() + # CHECK-LABEL: TEST: testMultitileSizes + # CHECK: transform.sequence + # CHECK: transform.structured.multitile_sizes + # CHECK-DAG: dimension = 1 + # CHECK-DAG: divisor = 2 # CHECK-DAG: target_size = 42