1*9f3f6d7bSStella Laurenzo# RUN: %PYTHON %s | FileCheck %s 2*9f3f6d7bSStella Laurenzo 3*9f3f6d7bSStella Laurenzofrom mlir.ir import * 4*9f3f6d7bSStella Laurenzoimport mlir.dialects.gpu 5*9f3f6d7bSStella Laurenzoimport mlir.dialects.gpu.passes 6*9f3f6d7bSStella Laurenzofrom mlir.passmanager import * 7*9f3f6d7bSStella Laurenzo 8*9f3f6d7bSStella Laurenzodef run(f): 9*9f3f6d7bSStella Laurenzo print("\nTEST:", f.__name__) 10*9f3f6d7bSStella Laurenzo f() 11*9f3f6d7bSStella Laurenzo 12*9f3f6d7bSStella Laurenzodef testGPUPass(): 13*9f3f6d7bSStella Laurenzo with Context() as context: 14*9f3f6d7bSStella Laurenzo PassManager.parse('gpu-kernel-outlining') 15*9f3f6d7bSStella Laurenzo print('SUCCESS') 16*9f3f6d7bSStella Laurenzo 17*9f3f6d7bSStella Laurenzo# CHECK-LABEL: testGPUPass 18*9f3f6d7bSStella Laurenzo# CHECK: SUCCESS 19*9f3f6d7bSStella Laurenzorun(testGPUPass) 20