blob: 38bf038a5eeed86de50f065b1591a1edaf706127 (
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.gpu
import mlir.dialects.gpu.passes
from mlir.passmanager import *
def run(f):
print("\nTEST:", f.__name__)
f()
def testGPUPass():
with Context() as context:
PassManager.parse('any(gpu-kernel-outlining)')
print('SUCCESS')
# CHECK-LABEL: testGPUPass
# CHECK: SUCCESS
run(testGPUPass)
|
