[all-commits] [llvm/llvm-project] 9b800b: [mlir][sparse] Improve the non-stable sort impleme...

[Bixia Zheng via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 9b800bf79d9d2fa18ed5be891346155238015515
      https://github.com/llvm/llvm-project/commit/9b800bf79d9d2fa18ed5be891346155238015515
  Author: bixia1 <bixia at google.com>
  Date:   2022-11-07 (Mon, 07 Nov 2022)

  Changed paths:
    M mlir/lib/Dialect/SparseTensor/Transforms/SparseBufferRewriting.cpp
    M mlir/test/Dialect/SparseTensor/buffer_rewriting.mlir
    M mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_rewrite_sort.mlir

  Log Message:
  -----------
  [mlir][sparse] Improve the non-stable sort implementation.

Replace the quick sort partition method with one that is more similar to the
method used by C++ std quick sort. This improves the runtime for sorting
sk_2005.mtx by more than 10x.

Reviewed By: aartbik

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