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

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