projects / platform / upstream / llvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 2edafe8) | patch
[mlir][sparse] Fix a bug in concatenate operator rewriting.
authorbixia1 <bixia@google.com>
Mon, 21 Nov 2022 05:57:30 +0000 (21:57 -0800)
committerbixia1 <bixia@google.com>
Tue, 22 Nov 2022 16:17:35 +0000 (08:17 -0800)
commit6c01b5cdaddce8df325020659d73cd7728778392
tree7543c4c6eb2184a0f43e029802a286595bee8ae6tree | snapshot
parent2edafe8393fa7c9ad2e95d691191469adb7bf4b0commit | diff
[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
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp diff | blob | history
mlir/test/Dialect/SparseTensor/sparse_concat_codegen.mlir diff | blob | history
Domain: System / Toolchain;