| # 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('gpu-kernel-outlining') | |
| print('SUCCESS') | |
| # CHECK-LABEL: testGPUPass | |
| # CHECK: SUCCESS | |
| run(testGPUPass) |