1*58d12332SAart Bik# RUN: %PYTHON %s | FileCheck %s 2*58d12332SAart Bik 3*58d12332SAart Bikfrom mlir.ir import * 4*58d12332SAart Bikfrom mlir.passmanager import * 5*58d12332SAart Bik 6*58d12332SAart Bikfrom mlir.dialects import sparse_tensor as st 7*58d12332SAart Bik 8*58d12332SAart Bik 9*58d12332SAart Bikdef run(f): 10*58d12332SAart Bik print('\nTEST:', f.__name__) 11*58d12332SAart Bik f() 12*58d12332SAart Bik return f 13*58d12332SAart Bik 14*58d12332SAart Bik 15*58d12332SAart Bik# CHECK-LABEL: TEST: testSparseTensorPass 16*58d12332SAart Bik@run 17*58d12332SAart Bikdef testSparseTensorPass(): 18*58d12332SAart Bik with Context() as context: 19*58d12332SAart Bik PassManager.parse('sparsification') 20*58d12332SAart Bik PassManager.parse('sparse-tensor-conversion') 21*58d12332SAart Bik # CHECK: SUCCESS 22*58d12332SAart Bik print('SUCCESS') 23