xref: /llvm-project/mlir/test/python/dialects/spirv_dialect.py (revision 2292fd0129362865d07777329fa38850d7a642a3)
1*2292fd01SJungwook Park# RUN: %PYTHON %s | FileCheck %s
2*2292fd01SJungwook Park
3*2292fd01SJungwook Parkfrom mlir.ir import *
4*2292fd01SJungwook Parkimport mlir.dialects.spirv as spirv
5*2292fd01SJungwook Park
6*2292fd01SJungwook Park
7*2292fd01SJungwook Parkdef run(f):
8*2292fd01SJungwook Park    print("\nTEST:", f.__name__)
9*2292fd01SJungwook Park    f()
10*2292fd01SJungwook Park
11*2292fd01SJungwook Park
12*2292fd01SJungwook Park# CHECK-LABEL: TEST: testConstantOp
13*2292fd01SJungwook Park@run
14*2292fd01SJungwook Parkdef testConstantOps():
15*2292fd01SJungwook Park    with Context() as ctx, Location.unknown():
16*2292fd01SJungwook Park        module = Module.create()
17*2292fd01SJungwook Park        with InsertionPoint(module.body):
18*2292fd01SJungwook Park            i32 = IntegerType.get_signless(32)
19*2292fd01SJungwook Park            spirv.ConstantOp(value=IntegerAttr.get(i32, 42), constant=i32)
20*2292fd01SJungwook Park        # CHECK: spirv.Constant 42 : i32
21*2292fd01SJungwook Park        print(module)
22