[mlir][linalg] Lower PadTensorOps with non-constant pad value
authorMatthias Springer <springerm@google.com>
Mon, 14 Jun 2021 06:00:30 +0000 (15:00 +0900)
committerMatthias Springer <springerm@google.com>
Mon, 14 Jun 2021 06:11:13 +0000 (15:11 +0900)
commitddda52ce3cf2936d9ee05e06ed70e7d270cfcd73
tree6bf58ced7c921fac205d55e854f2f2b7a0c5e925
parent73cbc91c93dd5a7ee1b8b1a9d507e194e835b446
[mlir][linalg] Lower PadTensorOps with non-constant pad value

The padding of such ops is not generated in a vectorized way. Instead, emit a tensor::GenerateOp.

We may vectorize GenerateOps in the future.

Differential Revision: https://reviews.llvm.org/D103879
mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp
mlir/test/Dialect/Linalg/vectorization.mlir