mlir/test/python/dialects/sparse_tensor/passes.py

# RUN: %PYTHON %s | FileCheck %s

from mlir.ir import *
from mlir.passmanager import *

from mlir.dialects import sparse_tensor as st


def run(f):
  print('\nTEST:', f.__name__)
  f()
  return f


# CHECK-LABEL: TEST: testSparseTensorPass
@run
def testSparseTensorPass():
  with Context() as context:
    PassManager.parse('sparsification')
    PassManager.parse('sparse-tensor-conversion')
  # CHECK: SUCCESS
  print('SUCCESS')