projects / platform / upstream / llvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 08d90f7) | patch
[mlir][sparse] refine optimization, add few more test cases
authorAart Bik <ajcbik@google.com>
Thu, 19 Nov 2020 20:56:30 +0000 (12:56 -0800)
committerAart Bik <ajcbik@google.com>
Sat, 21 Nov 2020 01:01:59 +0000 (17:01 -0800)
commitaf42550523d9feef72e6838d5b4e553ef4af79f9
treeeb7e27adadd56c66b921e4ab273d837c31358a79tree | snapshot
parent08d90f72cebd72dd0a972565ffcc445e57f50d8acommit | diff
[mlir][sparse] refine optimization, add few more test cases

Adds tests for full sum reduction (tensors summed up into scalars)
and the well-known sampled-dense-dense-matrix-product. Refines
the optimizations rules slightly to handle the summation better.

Reviewed By: penpornk

Differential Revision: https://reviews.llvm.org/D91818
mlir/lib/Dialect/Linalg/Transforms/Sparsification.cpp diff | blob | history
mlir/test/Dialect/Linalg/sparse_1d.mlir diff | blob | history
mlir/test/Dialect/Linalg/sparse_2d.mlir diff | blob | history
mlir/test/Dialect/Linalg/sparse_3d.mlir diff | blob | history
Domain: System / Toolchain;