# RUN: %PYTHON %s | FileCheck %s | |
# This is just a smoke test that the dialect is functional. | |
from mlir.ir import * | |
from mlir.dialects import rocdl | |
def constructAndPrintInModule(f): | |
print("\nTEST:", f.__name__) | |
with Context(), Location.unknown(): | |
module = Module.create() | |
with InsertionPoint(module.body): | |
f() | |
print(module) | |
return f | |
# CHECK-LABEL: testSmoke | |
@constructAndPrintInModule | |
def testSmoke(): | |
# CHECK: rocdl.barrier | |
rocdl.BarrierOp() |