f8479d9de5
Historically the builtin dialect has had an empty namespace. This has unfortunately created a very awkward situation, where many utilities either have to special case the empty namespace, or just don't work at all right now. This revision adds a namespace to the builtin dialect, and starts to cleanup some of the utilities to no longer handle empty namespaces. For now, the assembly form of builtin operations does not require the `builtin.` prefix. (This should likely be re-evaluated though) Differential Revision: https://reviews.llvm.org/D105149
107 lines
2.6 KiB
Python
107 lines
2.6 KiB
Python
# RUN: %PYTHON %s | FileCheck %s
|
|
|
|
import gc
|
|
from mlir.ir import *
|
|
|
|
|
|
def run(f):
|
|
print("\nTEST:", f.__name__)
|
|
f()
|
|
gc.collect()
|
|
assert Context._get_live_count() == 0
|
|
return f
|
|
|
|
|
|
# CHECK-LABEL: TEST: testDialectDescriptor
|
|
@run
|
|
def testDialectDescriptor():
|
|
ctx = Context()
|
|
d = ctx.get_dialect_descriptor("std")
|
|
# CHECK: <DialectDescriptor std>
|
|
print(d)
|
|
# CHECK: std
|
|
print(d.namespace)
|
|
try:
|
|
_ = ctx.get_dialect_descriptor("not_existing")
|
|
except ValueError:
|
|
pass
|
|
else:
|
|
assert False, "Expected exception"
|
|
|
|
|
|
# CHECK-LABEL: TEST: testUserDialectClass
|
|
@run
|
|
def testUserDialectClass():
|
|
ctx = Context()
|
|
# Access using attribute.
|
|
d = ctx.dialects.std
|
|
# CHECK: <Dialect std (class mlir.dialects._std_ops_gen._Dialect)>
|
|
print(d)
|
|
try:
|
|
_ = ctx.dialects.not_existing
|
|
except AttributeError:
|
|
pass
|
|
else:
|
|
assert False, "Expected exception"
|
|
|
|
# Access using index.
|
|
d = ctx.dialects["std"]
|
|
# CHECK: <Dialect std (class mlir.dialects._std_ops_gen._Dialect)>
|
|
print(d)
|
|
try:
|
|
_ = ctx.dialects["not_existing"]
|
|
except IndexError:
|
|
pass
|
|
else:
|
|
assert False, "Expected exception"
|
|
|
|
# Using the 'd' alias.
|
|
d = ctx.d["std"]
|
|
# CHECK: <Dialect std (class mlir.dialects._std_ops_gen._Dialect)>
|
|
print(d)
|
|
|
|
|
|
# CHECK-LABEL: TEST: testCustomOpView
|
|
# This test uses the standard dialect AddFOp as an example of a user op.
|
|
# TODO: Op creation and access is still quite verbose: simplify this test as
|
|
# additional capabilities come online.
|
|
@run
|
|
def testCustomOpView():
|
|
|
|
def createInput():
|
|
op = Operation.create("pytest_dummy.intinput", results=[f32])
|
|
# TODO: Auto result cast from operation
|
|
return op.results[0]
|
|
|
|
with Context() as ctx, Location.unknown():
|
|
ctx.allow_unregistered_dialects = True
|
|
m = Module.create()
|
|
|
|
with InsertionPoint(m.body):
|
|
f32 = F32Type.get()
|
|
# Create via dialects context collection.
|
|
input1 = createInput()
|
|
input2 = createInput()
|
|
op1 = ctx.dialects.std.AddFOp(input1.type, input1, input2)
|
|
|
|
# Create via an import
|
|
from mlir.dialects.std import AddFOp
|
|
AddFOp(input1.type, input1, op1.result)
|
|
|
|
# CHECK: %[[INPUT0:.*]] = "pytest_dummy.intinput"
|
|
# CHECK: %[[INPUT1:.*]] = "pytest_dummy.intinput"
|
|
# CHECK: %[[R0:.*]] = addf %[[INPUT0]], %[[INPUT1]] : f32
|
|
# CHECK: %[[R1:.*]] = addf %[[INPUT0]], %[[R0]] : f32
|
|
m.operation.print()
|
|
|
|
|
|
# CHECK-LABEL: TEST: testIsRegisteredOperation
|
|
@run
|
|
def testIsRegisteredOperation():
|
|
ctx = Context()
|
|
|
|
# CHECK: std.cond_br: True
|
|
print(f"std.cond_br: {ctx.is_registered_operation('std.cond_br')}")
|
|
# CHECK: std.not_existing: False
|
|
print(f"std.not_existing: {ctx.is_registered_operation('std.not_existing')}")
|