3f6648422c
Previously, we were using the static `typeid` of `DynamicType` for checks, which is incorrect. We should instead check against the `typeid` of `DynamicTypeDefinition` (which is a subclass of `SelfOwningTypeID`), and register it via `register_type_caster` so that Python-defined types can use `maybe_downcast`. (The attribute part is same.)
253 lines
8.4 KiB
Python
253 lines
8.4 KiB
Python
# RUN: %PYTHON %s 2>&1 | FileCheck %s
|
|
|
|
from mlir.ir import *
|
|
from mlir.dialects.irdl import *
|
|
import sys
|
|
|
|
|
|
def run(f):
|
|
print("\nTEST:", f.__name__, file=sys.stderr)
|
|
f()
|
|
|
|
|
|
# CHECK: TEST: testIRDL
|
|
@run
|
|
def testIRDL():
|
|
with Context() as ctx, Location.unknown():
|
|
module = Module.create()
|
|
with InsertionPoint(module.body):
|
|
irdl_test = dialect("irdl_test")
|
|
with InsertionPoint(irdl_test.body):
|
|
op = operation_("test_op")
|
|
with InsertionPoint(op.body):
|
|
f32 = is_(TypeAttr.get(F32Type.get()))
|
|
operands_([f32], ["input"], [Variadicity.single])
|
|
type1 = type_("type1")
|
|
with InsertionPoint(type1.body):
|
|
f32 = is_(TypeAttr.get(F32Type.get()))
|
|
parameters([f32], ["val"])
|
|
attr1 = attribute("attr1")
|
|
with InsertionPoint(attr1.body):
|
|
test = is_(StringAttr.get("test"))
|
|
parameters([test], ["val"])
|
|
|
|
# CHECK: module {
|
|
# CHECK: irdl.dialect @irdl_test {
|
|
# CHECK: irdl.operation @test_op {
|
|
# CHECK: %0 = irdl.is f32
|
|
# CHECK: irdl.operands(input: %0)
|
|
# CHECK: }
|
|
# CHECK: irdl.type @type1 {
|
|
# CHECK: %0 = irdl.is f32
|
|
# CHECK: irdl.parameters(val: %0)
|
|
# CHECK: }
|
|
# CHECK: irdl.attribute @attr1 {
|
|
# CHECK: %0 = irdl.is "test"
|
|
# CHECK: irdl.parameters(val: %0)
|
|
# CHECK: }
|
|
# CHECK: }
|
|
# CHECK: }
|
|
module.operation.verify()
|
|
module.dump()
|
|
|
|
load_dialects(module)
|
|
|
|
m = Module.parse(
|
|
"""
|
|
module {
|
|
%a = arith.constant 1.0 : f32
|
|
"irdl_test.test_op"(%a) : (f32) -> ()
|
|
}
|
|
"""
|
|
)
|
|
# CHECK: module {
|
|
# CHECK: "irdl_test.test_op"(%cst) : (f32) -> ()
|
|
# CHECK: }
|
|
m.dump()
|
|
|
|
|
|
# CHECK: TEST: testIRDLTypes
|
|
@run
|
|
def testIRDLTypes():
|
|
with Context() as ctx, Location.unknown():
|
|
module = Module.create()
|
|
with InsertionPoint(module.body):
|
|
irdl_test = dialect("irdl_type_test")
|
|
with InsertionPoint(irdl_test.body):
|
|
type1 = type_("type1")
|
|
with InsertionPoint(type1.body):
|
|
iattr = base(base_name="#builtin.integer")
|
|
parameters([iattr], ["val"])
|
|
type2 = type_("type2")
|
|
with InsertionPoint(type2.body):
|
|
iattr = base(base_name="#builtin.integer")
|
|
unit = is_(UnitAttr.get())
|
|
parameters([iattr, unit], ["val1", "val2"])
|
|
op1 = operation_("op1")
|
|
with InsertionPoint(op1.body):
|
|
t1 = base(base_ref=["irdl_type_test", "type1"])
|
|
results_([t1], ["res"], [Variadicity.single])
|
|
|
|
# CHECK: module {
|
|
# CHECK: irdl.dialect @irdl_type_test {
|
|
# CHECK: irdl.type @type1 {
|
|
# CHECK: %0 = irdl.base "#builtin.integer"
|
|
# CHECK: irdl.parameters(val: %0)
|
|
# CHECK: }
|
|
# CHECK: irdl.type @type2 {
|
|
# CHECK: %0 = irdl.base "#builtin.integer"
|
|
# CHECK: %1 = irdl.is unit
|
|
# CHECK: irdl.parameters(val1: %0, val2: %1)
|
|
# CHECK: }
|
|
# CHECK: irdl.operation @op1 {
|
|
# CHECK: %0 = irdl.base @irdl_type_test::@type1
|
|
# CHECK: irdl.results(res: %0)
|
|
# CHECK: }
|
|
# CHECK: }
|
|
# CHECK: }
|
|
module.operation.verify()
|
|
module.dump()
|
|
|
|
load_dialects(module)
|
|
|
|
i32 = IntegerType.get(32)
|
|
t1 = DynamicType.get("irdl_type_test.type1", [IntegerAttr.get(i32, 42)])
|
|
# CHECK: !irdl_type_test.type1<42 : i32>
|
|
t1.dump()
|
|
# CHECK: irdl_type_test.type1
|
|
print(t1.type_name, file=sys.stderr)
|
|
# CHECK: 1
|
|
print(len(t1.params), file=sys.stderr)
|
|
# CHECK: 42 : i32
|
|
t1.params[0].dump()
|
|
t2 = DynamicType.get(
|
|
"irdl_type_test.type2", [IntegerAttr.get(i32, 33), UnitAttr.get()]
|
|
)
|
|
# CHECK: !irdl_type_test.type2<33 : i32, unit>
|
|
t2.dump()
|
|
# CHECK: irdl_type_test.type2
|
|
print(t2.type_name, file=sys.stderr)
|
|
# CHECK: 2
|
|
print(len(t2.params), file=sys.stderr)
|
|
# CHECK: 33 : i32
|
|
t2.params[0].dump()
|
|
# CHECK: unit
|
|
t2.params[1].dump()
|
|
|
|
# CHECK: True
|
|
print(
|
|
t2.typeid == DynamicType.lookup_typeid("irdl_type_test.type2"),
|
|
file=sys.stderr,
|
|
)
|
|
# CHECK: False
|
|
print(
|
|
t1.typeid == DynamicType.lookup_typeid("irdl_type_test.type2"),
|
|
file=sys.stderr,
|
|
)
|
|
# CHECK: True
|
|
print(
|
|
t1.typeid == DynamicType.lookup_typeid("irdl_type_test.type1"),
|
|
file=sys.stderr,
|
|
)
|
|
|
|
m = Module.create()
|
|
with InsertionPoint(m.body):
|
|
Operation.create("irdl_type_test.op1", results=[t1])
|
|
|
|
assert m.operation.verify()
|
|
# CHECK: %0 = "irdl_type_test.op1"() : () -> !irdl_type_test.type1<42 : i32>
|
|
m.dump()
|
|
|
|
|
|
# CHECK: TEST: testIRDLAttrs
|
|
@run
|
|
def testIRDLAttrs():
|
|
with Context() as ctx, Location.unknown():
|
|
module = Module.create()
|
|
with InsertionPoint(module.body):
|
|
irdl_test = dialect("irdl_attr_test")
|
|
with InsertionPoint(irdl_test.body):
|
|
attr1 = attribute("attr1")
|
|
with InsertionPoint(attr1.body):
|
|
iattr = base(base_name="#builtin.integer")
|
|
parameters([iattr], ["val"])
|
|
attr2 = attribute("attr2")
|
|
with InsertionPoint(attr2.body):
|
|
iattr = base(base_name="#builtin.integer")
|
|
unit = is_(UnitAttr.get())
|
|
parameters([iattr, unit], ["val1", "val2"])
|
|
op1 = operation_("op1")
|
|
with InsertionPoint(op1.body):
|
|
a1 = base(base_ref=["irdl_attr_test", "attr1"])
|
|
attributes_([a1], ["attr"])
|
|
|
|
# CHECK: module {
|
|
# CHECK: irdl.dialect @irdl_attr_test {
|
|
# CHECK: irdl.attribute @attr1 {
|
|
# CHECK: %0 = irdl.base "#builtin.integer"
|
|
# CHECK: irdl.parameters(val: %0)
|
|
# CHECK: }
|
|
# CHECK: irdl.attribute @attr2 {
|
|
# CHECK: %0 = irdl.base "#builtin.integer"
|
|
# CHECK: %1 = irdl.is unit
|
|
# CHECK: irdl.parameters(val1: %0, val2: %1)
|
|
# CHECK: }
|
|
# CHECK: irdl.operation @op1 {
|
|
# CHECK: %0 = irdl.base @irdl_attr_test::@attr1
|
|
# CHECK: irdl.attributes {"attr" = %0}
|
|
# CHECK: }
|
|
# CHECK: }
|
|
# CHECK: }
|
|
module.operation.verify()
|
|
module.dump()
|
|
|
|
load_dialects(module)
|
|
|
|
i32 = IntegerType.get(32)
|
|
a1 = DynamicAttr.get("irdl_attr_test.attr1", [IntegerAttr.get(i32, 42)])
|
|
# CHECK: #irdl_attr_test.attr1<42 : i32>
|
|
a1.dump()
|
|
# CHECK: irdl_attr_test.attr1
|
|
print(a1.attr_name, file=sys.stderr)
|
|
# CHECK: 1
|
|
print(len(a1.params), file=sys.stderr)
|
|
# CHECK: 42 : i32
|
|
a1.params[0].dump()
|
|
a2 = DynamicAttr.get(
|
|
"irdl_attr_test.attr2", [IntegerAttr.get(i32, 33), UnitAttr.get()]
|
|
)
|
|
# CHECK: #irdl_attr_test.attr2<33 : i32, unit>
|
|
a2.dump()
|
|
# CHECK: irdl_attr_test.attr2
|
|
print(a2.attr_name, file=sys.stderr)
|
|
# CHECK: 2
|
|
print(len(a2.params), file=sys.stderr)
|
|
# CHECK: 33 : i32
|
|
a2.params[0].dump()
|
|
# CHECK: unit
|
|
a2.params[1].dump()
|
|
|
|
# CHECK: True
|
|
print(
|
|
a2.typeid == DynamicAttr.lookup_typeid("irdl_attr_test.attr2"),
|
|
file=sys.stderr,
|
|
)
|
|
# CHECK: False
|
|
print(
|
|
a1.typeid == DynamicAttr.lookup_typeid("irdl_attr_test.attr2"),
|
|
file=sys.stderr,
|
|
)
|
|
# CHECK: True
|
|
print(
|
|
a1.typeid == DynamicAttr.lookup_typeid("irdl_attr_test.attr1"),
|
|
file=sys.stderr,
|
|
)
|
|
|
|
m = Module.create()
|
|
with InsertionPoint(m.body):
|
|
Operation.create("irdl_attr_test.op1", attributes={"attr": a1})
|
|
|
|
assert m.operation.verify()
|
|
# CHECK: "irdl_attr_test.op1"() {attr = #irdl_attr_test.attr1<42 : i32>} : () -> ()
|
|
m.dump()
|