[all-commits] [llvm/llvm-project] 2c81d4: [mlir][sparse] Improve the implementation of spars...

[Bixia Zheng via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 2c81d43241f79168dc3f63691c1f154d33942dc9
      https://github.com/llvm/llvm-project/commit/2c81d43241f79168dc3f63691c1f154d33942dc9
  Author: bixia1 <bixia at google.com>
  Date:   2023-03-01 (Wed, 01 Mar 2023)

  Changed paths:
    M mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorCodegen.cpp
    M mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp
    M mlir/test/Dialect/SparseTensor/codegen.mlir
    M mlir/test/Dialect/SparseTensor/rewriting_for_codegen.mlir

  Log Message:
  -----------
  [mlir][sparse] Improve the implementation of sparse_tensor.new for the codegen path.

Rewrite a NewOp into a NewOp of a sorted COO tensor and a ConvertOp for
converting the sorted COO tensor to the destination tensor type.

Codegen a NewOp of a sorted COO tensor to use the new bulk reader API and sort
the elements only when the input is not sorted.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D144504