[mlir][sparse] Add layout to the memref for the indices buffers to prepare for the...
authorbixia1 <bixia@google.com>
Tue, 3 Jan 2023 23:16:12 +0000 (15:16 -0800)
committerbixia1 <bixia@google.com>
Wed, 4 Jan 2023 15:36:11 +0000 (07:36 -0800)
commit90aa4362913d4afbf24b930c261a3578cb793e96
tree184c323c2335394a436f129d53058fb53c213ab8
parent8383da1583aa6888fbb304b49760f045bbb1522c
[mlir][sparse] Add layout to the memref for the indices buffers to prepare for the AOS storage optimization for COO regions.

Fix relevant FileCheck tests.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D140742
mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensor.h
mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.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/lib/Dialect/SparseTensor/Transforms/SparseTensorConversion.cpp
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp
mlir/test/Dialect/SparseTensor/convert_dense2sparse.mlir
mlir/test/Dialect/SparseTensor/sorted_coo.mlir