search

aqjune / mlir-tv

A translation validation framework for MLIR
Apache License 2.0
73 stars 10 forks source link

Support SparseTensor #95

Open aqjune opened 3 years ago

aqjune commented 3 years ago

https://mlir.llvm.org/docs/Dialects/SparseTensorOps/

Key ops: indices, values, pointer; use forall quantifiers or uninterpreted functions to encode them.

aqjune commented 3 years ago

NOTE: Many linalg operations work "out of the box" by annotating tensor types as sparse; even without adding support for the key ops, things will work in practice

aqjune commented 3 years ago

https://github.com/aqjune/mlir-tv/commit/46d02903a6bbb38a1d25eb3f03a202b56a5faecf