bixia1
101a0c84f7
[mlir][sparse] Improve concatenate operator rewrite for annotated all dense dimensions results.
...
Reviewed By: aartbik
Differential Revision: https://reviews.llvm.org/D138823
2022-11-29 12:25:52 -08:00
bixia1
aedf5d5831
[mlir][sparse] Improve concatenate operator rewriting for dense tensor results.
...
Reviewed By: Peiming
Differential Revision: https://reviews.llvm.org/D138465
2022-11-28 07:56:01 -08:00
bixia1
6c01b5cdad
[mlir][sparse] Fix a bug in concatenate operator rewriting.
...
When calculating the dynamic dimensions for the concatenate result, we
shouldn't accumulate the sizes for the non-concatenating dimensions.
Reviewed By: aartbik, Peiming
Differential Revision: https://reviews.llvm.org/D138436
2022-11-22 08:17:35 -08:00
bixia1
f81f0cb75a
[mlir][sparse] Split SparseTensorRewrite into PreSparsificationRewrite and PostSparsificationRewrite.
...
Reviewed By: aartbik, wrengr
Differential Revision: https://reviews.llvm.org/D138153
2022-11-17 07:13:55 -08:00
Peiming Liu
4fa00ce15c
[mlir][sparse] extend foreach operation to accept reduction arguments; fix sparse tensor rewriting patterns that do not propagate sparse tensor SSA properly.
...
This patch re-commit D137468 and D137463, which were reverted by mistakes.
Reviewed By: aartbik
Differential Revision: https://reviews.llvm.org/D137579
2022-11-07 21:40:30 +00:00
Stella Stamenova
ec224e3b68
Revert "[mlir][sparse] fix sparse tensor rewriting patterns that do not propagate sparse tensor SSA properly."
...
This reverts commit 70508b614e6478ba2c3fc79e935e2c68e2d79b71.
This change depends on a reverted change that broke the windows mlir buildbot; reverting to bring remaining mlir bots to green
2022-11-07 09:00:08 -08:00
Peiming Liu
70508b614e
[mlir][sparse] fix sparse tensor rewriting patterns that do not propagate sparse tensor SSA properly.
...
Reviewed By: aartbik
Differential Revision: https://reviews.llvm.org/D137468
2022-11-04 23:43:54 +00:00
bixia1
eb877006a6
[mlir][sparse] Add rewriting rule for the convert operator.
...
Reviewed By: aartbik
Differential Revision: https://reviews.llvm.org/D136301
2022-11-01 15:57:34 -07:00
Peiming Liu
b0f8057e4c
[mlir][sparse] use loop emitter to generate loop in sparsification
...
Reviewed By: aartbik
Differential Revision: https://reviews.llvm.org/D136185
2022-10-26 00:27:56 +00:00
Aart Bik
779dcd2ecc
[mlir][sparse] move sparse tensor rewriting into its own pass
...
Makes individual testing and debugging easier.
Reviewed By: bixia
Differential Revision: https://reviews.llvm.org/D135319
2022-10-05 14:52:55 -07:00
Peiming Liu
00ad065548
[mlir][sparse] Add rewriting rules for concatente using foreach operator.
...
Reviewed By: aartbik
Differential Revision: https://reviews.llvm.org/D134895
2022-09-30 21:56:55 +00:00
