[all-commits] [llvm/llvm-project] 2b8a4d: [mlir][sparse] Introduce new reduce op

Jim Kitchen via All-commits all-commits at lists.llvm.org
Fri Jul 15 13:31:10 PDT 2022


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 2b8a4d9ce1fc72fc97ef55170a5851b9fbda0889
      https://github.com/llvm/llvm-project/commit/2b8a4d9ce1fc72fc97ef55170a5851b9fbda0889
  Author: Jim Kitchen <jim22k at gmail.com>
  Date:   2022-07-15 (Fri, 15 Jul 2022)

  Changed paths:
    M mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td
    M mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp
    M mlir/test/Dialect/SparseTensor/invalid.mlir
    M mlir/test/Dialect/SparseTensor/roundtrip.mlir

  Log Message:
  -----------
  [mlir][sparse] Introduce new reduce op

A new sparse_tensor operation allows for
custom reduction code to be injected during
linalg.generic lowering for sparse tensors.
An identity value is provided to indicate
the starting value of the reduction. A single
block region is required to contain the
custom reduce computation.

Reviewed by: aartbik

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




More information about the All-commits mailing list