[mlir][sparse] Refactor the code that reshapes the values buffer for annotated all...
authorbixia1 <bixia@google.com>
Wed, 11 Jan 2023 16:59:00 +0000 (08:59 -0800)
committerbixia1 <bixia@google.com>
Thu, 12 Jan 2023 00:02:46 +0000 (16:02 -0800)
commitd114dfba2dc4187bd9b98291963e64706acab6b5
tree90569493f94c4922173acce781ba8dd191cef29b
parentfd5e2627066075f3d15ef774ef368e08735a9ac9
[mlir][sparse] Refactor the code that reshapes the values buffer for annotated all dense tensors.

Move the functionality to codegen utils for sharing with the codegen path.

Reviewed By: Peiming

Differential Revision: https://reviews.llvm.org/D141514
mlir/lib/Dialect/SparseTensor/Transforms/CodegenUtils.cpp
mlir/lib/Dialect/SparseTensor/Transforms/CodegenUtils.h
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorConversion.cpp