[mlir][sparse] extend unpack operation to support unpacking a batched COO type
authorPeiming Liu <peiming@google.com>
Fri, 21 Apr 2023 20:06:36 +0000 (20:06 +0000)
committerPeiming Liu <peiming@google.com>
Mon, 1 May 2023 18:17:29 +0000 (18:17 +0000)
commitd4db52893857a836940e0951daa205de1bb1d201
treeb4a2a9ffcc6bcce4655de067a595f8ed91d39e49
parentf9fbda7102fb6da79dedf9b15694d7a61e8c49b7
[mlir][sparse] extend unpack operation to support unpacking a batched COO type

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D149103
13 files changed:
mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td
mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp
mlir/lib/Dialect/SparseTensor/Transforms/BufferizableOpInterfaceImpl.cpp
mlir/lib/Dialect/SparseTensor/Transforms/CodegenUtils.cpp
mlir/lib/Dialect/SparseTensor/Transforms/CodegenUtils.h
mlir/lib/Dialect/SparseTensor/Transforms/LoopEmitter.cpp
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorCodegen.cpp
mlir/test/Dialect/SparseTensor/invalid.mlir
mlir/test/Dialect/SparseTensor/roundtrip.mlir
mlir/test/Dialect/SparseTensor/sparse_2d.mlir
mlir/test/Dialect/SparseTensor/sparse_foreach.mlir
mlir/test/Dialect/SparseTensor/sparse_pack.mlir
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_pack.mlir