shylie/llvm-project
1
0
Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
447,783 Commits 1,713 Branches 298 Tags
Commit Graph

12 Commits

Author SHA1 Message Date
Peiming Liu
a3672add76 [mlir][sparse] avoid unnecessary tmp COO buffer and convert when lowering ConcatentateOp. 
When concat along dim 0, and all inputs/outputs are ordered with identity dimension ordering,
the concatenated coordinates will be yield in lexOrder, thus no need to sort.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D140228
 2022-12-16 18:26:39 +00:00
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