blob: 9319e16e054deb0cbc8bfcd126a0f24b12f02498 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
# 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('any(sparsification)')
PassManager.parse('any(sparse-tensor-conversion)')
# CHECK: SUCCESS
print('SUCCESS')
|
