[Mlir-commits] [mlir] dad10a9 - NFC: Adapt operation.py to builtin operation print format changes.

[Stella Laurenzo]

Author: Stella Laurenzo
Date: 2021-07-28T23:57:08Z
New Revision: dad10a9afbc383076bfa273f151de9e7bdb94eda

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

LOG: NFC: Adapt operation.py to builtin operation print format changes.

Added: 
    

Modified: 
    mlir/test/python/ir/operation.py

Removed: 
    


################################################################################
diff  --git a/mlir/test/python/ir/operation.py b/mlir/test/python/ir/operation.py
index fa107164fec6..d4fa09131807 100644
--- a/mlir/test/python/ir/operation.py
+++ b/mlir/test/python/ir/operation.py
@@ -59,7 +59,7 @@ def walk_operations(indent, op):
 
   # CHECK: REGION 0:
   # CHECK:   BLOCK 0:
-  # CHECK:     OP 0: func
+  # CHECK:     OP 0: builtin.func
   # CHECK:       REGION 0:
   # CHECK:         BLOCK 0:
   # CHECK:           OP 0: %0 = "custom.addi"
@@ -96,14 +96,14 @@ def walk_operations(indent, op):
 
   # CHECK: REGION 0:
   # CHECK:   BLOCK 0:
-  # CHECK:     OP 0: func
-  # CHECK:     OP 0: parent module
+  # CHECK:     OP 0: builtin.func
+  # CHECK:     OP 0: parent builtin.module
   # CHECK:       REGION 0:
   # CHECK:         BLOCK 0:
   # CHECK:           OP 0: %0 = "custom.addi"
-  # CHECK:           OP 0: parent func
+  # CHECK:           OP 0: parent builtin.func
   # CHECK:           OP 1: return
-  # CHECK:           OP 1: parent func
+  # CHECK:           OP 1: parent builtin.func
   walk_operations("", module.operation)
 
 run(testTraverseOpRegionBlockIndices)
@@ -585,7 +585,7 @@ def testPrintInvalidOperation():
     # to the generic printer for safety.
     block = module.regions[0].blocks.append()
     # CHECK: // Verification failed, printing generic form
-    # CHECK: "module"() ( {
+    # CHECK: "builtin.module"() ( {
     # CHECK: }) : () -> ()
     print(module)
     # CHECK: .verify = False
@@ -600,22 +600,22 @@ def testCreateWithInvalidAttributes():
     try:
       Operation.create("builtin.module", attributes={None:StringAttr.get("name")})
     except Exception as e:
-      # CHECK: Invalid attribute key (not a string) when attempting to create the operation "module"
+      # CHECK: Invalid attribute key (not a string) when attempting to create the operation "builtin.module"
       print(e)
     try:
       Operation.create("builtin.module", attributes={42:StringAttr.get("name")})
     except Exception as e:
-      # CHECK: Invalid attribute key (not a string) when attempting to create the operation "module"
+      # CHECK: Invalid attribute key (not a string) when attempting to create the operation "builtin.module"
       print(e)
     try:
       Operation.create("builtin.module", attributes={"some_key":ctx})
     except Exception as e:
-      # CHECK: Invalid attribute value for the key "some_key" when attempting to create the operation "module"
+      # CHECK: Invalid attribute value for the key "some_key" when attempting to create the operation "builtin.module"
       print(e)
     try:
       Operation.create("builtin.module", attributes={"some_key":None})
     except Exception as e:
-      # CHECK: Found an invalid (`None`?) attribute value for the key "some_key" when attempting to create the operation "module"
+      # CHECK: Found an invalid (`None`?) attribute value for the key "some_key" when attempting to create the operation "builtin.module"
       print(e)
 run(testCreateWithInvalidAttributes)