158d12332SAart Bik# RUN: %PYTHON %s | FileCheck %s 258d12332SAart Bik 358d12332SAart Bikfrom mlir.ir import * 458d12332SAart Bikfrom mlir.passmanager import * 558d12332SAart Bik 658d12332SAart Bikfrom mlir.dialects import sparse_tensor as st 758d12332SAart Bik 858d12332SAart Bik 958d12332SAart Bikdef run(f): 10*f9008e63STobias Hieta print("\nTEST:", f.__name__) 1158d12332SAart Bik f() 1258d12332SAart Bik return f 1358d12332SAart Bik 1458d12332SAart Bik 1558d12332SAart Bik# CHECK-LABEL: TEST: testSparseTensorPass 1658d12332SAart Bik@run 1758d12332SAart Bikdef testSparseTensorPass(): 1858d12332SAart Bik with Context() as context: 19*f9008e63STobias Hieta PassManager.parse("any(sparsification)") 20*f9008e63STobias Hieta PassManager.parse("any(sparse-tensor-conversion)") 2158d12332SAart Bik # CHECK: SUCCESS 22*f9008e63STobias Hieta print("SUCCESS") 23