diff options
author | Spenser Bauman <sbauman@mathworks.com> | 2023-09-05 20:21:52 +0200 |
---|---|---|
committer | Lorenzo Chelini <l.chelini@icloud.com> | 2023-09-05 21:08:13 +0200 |
commit | f32427e0447c787755fb29415f4deeb4c00adacd (patch) | |
tree | ce7a787d9b8fbe17410010679b066eb75d339909 /clang/unittests/Frontend/CompilerInvocationTest.cpp | |
parent | 4146b30785fb5db6fa4da47e6198f7bd961559c5 (diff) | |
download | llvm-f32427e0447c787755fb29415f4deeb4c00adacd.zip llvm-f32427e0447c787755fb29415f4deeb4c00adacd.tar.gz llvm-f32427e0447c787755fb29415f4deeb4c00adacd.tar.bz2 |
[mlir][linalg] Fix lowering of tensor.pack operations
Tensor pack operations are optimistically lowered to pad + insert_slice
when the pack operation only pads the input tensor. The existing
lowering emits insert_slice operations which do not meet the
rank-reducibility requirements of insert_slice.
This change updates the logic in linalg::lowerPack to first check the
rank-reducibility requirement. When the requirement is not met, the
lowering will emit the full sequence of pad + expand + transpose.
Reviewed By: chelini
Differential Revision: https://reviews.llvm.org/D159382
Diffstat (limited to 'clang/unittests/Frontend/CompilerInvocationTest.cpp')
0 files changed, 0 insertions, 0 deletions