# 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') |