[mlir][sparse] properly record dimension level type and properties
authorAart Bik <ajcbik@google.com>
Fri, 9 Sep 2022 22:42:46 +0000 (15:42 -0700)
committerAart Bik <ajcbik@google.com>
Mon, 12 Sep 2022 16:59:53 +0000 (09:59 -0700)
commitb22397fee45c35fcdfb679bf27f976038a3542a0
treeab763895ffbfdb1191153443dbf9762992892d5b
parentde3445e0ef15c420955ad720fccf08473f460443
[mlir][sparse] properly record dimension level type and properties

A next step towards supporting the new dimension level types and
properties. This changes properly records the properties in the
Merger, so that subsequent computations (lattice optimizations)
and code generation (during sparsification) can do the right thing.

https://github.com/llvm/llvm-project/issues/51658

Reviewed By: Peiming

Differential Revision: https://reviews.llvm.org/D133620
mlir/include/mlir/Dialect/SparseTensor/Utils/Merger.h
mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
mlir/lib/Dialect/SparseTensor/Utils/Merger.cpp
mlir/unittests/Dialect/SparseTensor/MergerTest.cpp