llvm-project/mlir/test/python/dialects/async_dialect.py
Stella Laurenzo 9f3f6d7bd8 Move MLIR python sources to mlir/python.
* NFC but has some fixes for CMake glitches discovered along the way (things not cleaning properly, co-mingled depends).
* Includes previously unsubmitted fix in D98681 and a TODO to fix it more appropriately in a smaller followup.

Differential Revision: https://reviews.llvm.org/D101493
2021-05-03 18:36:48 +00:00

20 lines
404 B
Python

# RUN: %PYTHON %s | FileCheck %s
from mlir.ir import *
import mlir.dialects.async_dialect
import mlir.dialects.async_dialect.passes
from mlir.passmanager import *
def run(f):
print("\nTEST:", f.__name__)
f()
def testAsyncPass():
with Context() as context:
PassManager.parse('async-to-async-runtime')
print('SUCCESS')
# CHECK-LABEL: testAsyncPass
# CHECK: SUCCESS
run(testAsyncPass)