[Mlir-commits] [mlir] a48d8c2 - [mlir][transform][linalg][python] Fix test for TileToForallOp mixin.

[Ingo Müller]

Author: Ingo Müller
Date: 2023-07-19T14:30:00Z
New Revision: a48d8c238f461eadfcb28b065341ae4b55920cf6

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

LOG: [mlir][transform][linalg][python] Fix test for TileToForallOp mixin.

The test was introduced recently by https://reviews.llvm.org/D155090,
but https://reviews.llvm.org/D155567, which I worked on concurrently
affected the output of that test and the two patches were never tested
together.

Differential Revision: https://reviews.llvm.org/D155709

Added: 
    

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

Removed: 
    


################################################################################
diff  --git a/mlir/test/python/dialects/transform_structured_ext.py b/mlir/test/python/dialects/transform_structured_ext.py
index 1663ea3b7ae733..ceeb62c4dfa9bd 100644
--- a/mlir/test/python/dialects/transform_structured_ext.py
+++ b/mlir/test/python/dialects/transform_structured_ext.py
@@ -314,11 +314,11 @@ def testTileToForallMixedDynamic():
         transform.YieldOp()
     # CHECK-LABEL: TEST: testTileToForallMixedDynamic
     # CHECK: = transform.structured.tile_to_forall_op
-    # CHECK-SAME: num_threads [%{{.*}} : !pdl.operation, 3, 4]
+    # CHECK-SAME: num_threads [%{{.*}} : !transform.any_op, 3, 4]
 
 
 @run
-def testTileToForallMPackedDynamic():
+def testTileToForallPackedDynamic():
     sequence = transform.SequenceOp(
         transform.FailurePropagationMode.PROPAGATE, [], transform.AnyOpType.get()
     )
@@ -326,9 +326,9 @@ def testTileToForallMPackedDynamic():
         n = structured.MatchOp.match_op_names(sequence.bodyTarget, ["test.dummy"])
         structured.TileToForallOp(sequence.bodyTarget, num_threads=n)
         transform.YieldOp()
-    # CHECK-LABEL: TEST: testTileToForallMPackedDynamic
+    # CHECK-LABEL: TEST: testTileToForallPackedDynamic
     # CHECK: = transform.structured.tile_to_forall_op
-    # CHECK-SAME: num_threads *(%0 : !pdl.operation)
+    # CHECK-SAME: num_threads *(%0 : !transform.any_op)
 
 
 @run