[mlir][sparse] extend pack operation to support packing a batched COO type
authorPeiming Liu <peiming@google.com>
Tue, 18 Apr 2023 22:33:25 +0000 (22:33 +0000)
committerPeiming Liu <peiming@google.com>
Thu, 20 Apr 2023 01:35:30 +0000 (01:35 +0000)
commit7864d736cf8b86b93c064c766396898e6f7b24cc
tree745d984dc2a20dbda52a53a0f465f07506189cb4
parentfc5c1a767658314ff30e1af6bf52a956d96b1b04
[mlir][sparse] extend pack operation to support packing a batched COO type

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D148670
mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensor.h
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/SparseTensorCodegen.cpp
mlir/test/Dialect/SparseTensor/invalid.mlir
mlir/test/Dialect/SparseTensor/roundtrip.mlir