[all-commits] [llvm/llvm-project] 6c01b5: [mlir][sparse] Fix a bug in concatenate operator r...

[Bixia Zheng via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 6c01b5cdaddce8df325020659d73cd7728778392
      https://github.com/llvm/llvm-project/commit/6c01b5cdaddce8df325020659d73cd7728778392
  Author: bixia1 <bixia at google.com>
  Date:   2022-11-22 (Tue, 22 Nov 2022)

  Changed paths:
    M mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp
    M mlir/test/Dialect/SparseTensor/sparse_concat_codegen.mlir

  Log Message:
  -----------
  [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