[mlir][sparse] Introduce new sparse_tensor.storage_get/set to access memory that...
authorPeiming Liu <peiming@google.com>
Wed, 31 Aug 2022 20:44:41 +0000 (20:44 +0000)
committerPeiming Liu <peiming@google.com>
Wed, 31 Aug 2022 22:15:15 +0000 (22:15 +0000)
commit7ea643c06d8977045d0cf79507f36d828773378c
tree6fa0566b16568501b54ecbdd3a73e00b6ba0ceb2
parentf767f0925268981272e0b48cbc3709d761b5600e
[mlir][sparse] Introduce new sparse_tensor.storage_get/set to access memory that stores the handle of a sparse tensor

Introduce new sparse_tensor.storage_get/set to access memory that stores the handle of a sparse tensor. The sparse tensor storage are represented as a tuple, these operation will later be eliminated and the tuple will be flattened after sparse tensor codegen

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D133049
mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td
mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp
mlir/test/Dialect/SparseTensor/invalid.mlir
mlir/test/Dialect/SparseTensor/roundtrip.mlir