maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

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

This commit is contained in:
Stella Laurenzo 2021-07-28 23:56:38 +00:00
parent ca7c66ccb8
commit dad10a9afb
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
mlir/test/python/ir/operation.py
View File

@ -59,7 +59,7 @@ def testTraverseOpRegionBlockIterators():
# 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 testTraverseOpRegionBlockIndices():
# 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)