[mlir][linalg] Vectorize 0-d tensor extract
authorAndrzej Warzynski <andrzej.warzynski@arm.com>
Wed, 5 Jul 2023 15:45:43 +0000 (16:45 +0100)
committerAndrzej Warzynski <andrzej.warzynski@arm.com>
Thu, 6 Jul 2023 07:34:51 +0000 (08:34 +0100)
commit2712b2805b47f10b3864ab19a4016ea306126ad7
treea0bf34715ebb9ff739b172cad7d0a68df3a51ff6
parent51feb1a72c5beb8f46d2f5569f3c3398f8902272
[mlir][linalg] Vectorize 0-d tensor extract

This patch adds the missing logic to vectorise `tensor.extract` for 0-d
tensors.

Fixes #63688

Differential Revision: https://reviews.llvm.org/D154518
mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp
mlir/test/Dialect/Linalg/vectorize-tensor-extract.mlir