[all-commits] [llvm/llvm-project] c194b4: [mlir][sparse] add full dimension ordering support

Aart Bik via All-commits all-commits at lists.llvm.org
Fri May 21 12:35:40 PDT 2021


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: c194b49c9c8dfe01804ecd0b90814d1e98382fc1
      https://github.com/llvm/llvm-project/commit/c194b49c9c8dfe01804ecd0b90814d1e98382fc1
  Author: Aart Bik <ajcbik at google.com>
  Date:   2021-05-21 (Fri, 21 May 2021)

  Changed paths:
    M mlir/include/mlir/ExecutionEngine/CRunnerUtils.h
    M mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorConversion.cpp
    M mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
    M mlir/lib/ExecutionEngine/SparseUtils.cpp
    M mlir/test/Dialect/SparseTensor/conversion.mlir
    M mlir/test/Dialect/SparseTensor/sparse_lower.mlir
    A mlir/test/Dialect/SparseTensor/sparse_lower_col.mlir
    M mlir/test/Dialect/SparseTensor/sparse_lower_inplace.mlir
    A mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_flatten.mlir

  Log Message:
  -----------
  [mlir][sparse] add full dimension ordering support

This revision completes the "dimension ordering" feature
of sparse tensor types that enables the programmer to
define a preferred order on dimension access (other than
the default left-to-right order). This enables e.g. selection
of column-major over row-major storage for sparse matrices,
but generalized to any rank, as in:

dimOrdering = affine_map<(i,j,k,l,m,n,o,p) -> (p,o,j,k,i,l,m,n)>

Reviewed By: bixia

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




More information about the All-commits mailing list