summaryrefslogtreecommitdiff
path: root/mlir/test/python/dialects/async_dialect.py
blob: da3103cecddf2809f1fbb4659d46517754aa6d17 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
# 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('any(async-to-async-runtime)')
  print('SUCCESS')

# CHECK-LABEL: testAsyncPass
#       CHECK: SUCCESS
run(testAsyncPass)