[mlir][sparse] Add AOS optimization.
authorbixia1 <bixia@google.com>
Wed, 4 Jan 2023 21:08:06 +0000 (13:08 -0800)
committerbixia1 <bixia@google.com>
Thu, 5 Jan 2023 02:16:04 +0000 (18:16 -0800)
commit3fdd85da067edb3d8f96095f344e7b07bcce5241
treed429ea4889da9e35d965bb8703be365cef0e0088
parentff66d410fd87608db16a4ffb5caae1fd62374d5a
[mlir][sparse] Add AOS optimization.

Use an array of structures to represent the indices for the tailing COO region
of a sparse tensor.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D140870
mlir/lib/Dialect/SparseTensor/Transforms/SparseStorageSpecifierToLLVM.cpp
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorCodegen.cpp
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorStorageLayout.cpp
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorStorageLayout.h
mlir/test/Dialect/SparseTensor/codegen.mlir
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_insert_2d.mlir
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_insert_3d.mlir
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_sorted_coo.mlir