xref: /llvm-project/mlir/test/python/dialects/emitc_dialect.py (revision 3c464d23682b0f9e6f70965e8f8f3861c9ba5417)
1*3c464d23SEliud de León# RUN: %PYTHON %s | FileCheck %s
2*3c464d23SEliud de León
3*3c464d23SEliud de Leónfrom mlir.ir import *
4*3c464d23SEliud de Leónimport mlir.dialects.emitc as emitc
5*3c464d23SEliud de León
6*3c464d23SEliud de León
7*3c464d23SEliud de Leóndef run(f):
8*3c464d23SEliud de León    print("\nTEST:", f.__name__)
9*3c464d23SEliud de León    with Context() as ctx, Location.unknown():
10*3c464d23SEliud de León        module = Module.create()
11*3c464d23SEliud de León        with InsertionPoint(module.body):
12*3c464d23SEliud de León            f(ctx)
13*3c464d23SEliud de León        print(module)
14*3c464d23SEliud de León
15*3c464d23SEliud de León
16*3c464d23SEliud de León# CHECK-LABEL: TEST: testConstantOp
17*3c464d23SEliud de León@run
18*3c464d23SEliud de Leóndef testConstantOp(ctx):
19*3c464d23SEliud de León    i32 = IntegerType.get_signless(32)
20*3c464d23SEliud de León    a = emitc.ConstantOp(result=i32, value=IntegerAttr.get(i32, 42))
21*3c464d23SEliud de León    # CHECK: %{{.*}} = "emitc.constant"() <{value = 42 : i32}> : () -> i32
22*3c464d23SEliud de León
23*3c464d23SEliud de León
24*3c464d23SEliud de León# CHECK-LABEL: TEST: testAddOp
25*3c464d23SEliud de León@run
26*3c464d23SEliud de Leóndef testAddOp(ctx):
27*3c464d23SEliud de León    i32 = IntegerType.get_signless(32)
28*3c464d23SEliud de León    lhs = emitc.ConstantOp(result=i32, value=IntegerAttr.get(i32, 0))
29*3c464d23SEliud de León    rhs = emitc.ConstantOp(result=i32, value=IntegerAttr.get(i32, 0))
30*3c464d23SEliud de León    a = emitc.AddOp(i32, lhs, rhs)
31*3c464d23SEliud de León    # CHECK: %{{.*}} = emitc.add %{{.*}}, %{{.*}} : (i32, i32) -> i32
32