[mlir] Vectorize linalg.pad_tensor consumed by transfer_write
authorMatthias Springer <springerm@google.com>
Mon, 14 Jun 2021 01:16:22 +0000 (10:16 +0900)
committerMatthias Springer <springerm@google.com>
Mon, 14 Jun 2021 01:17:23 +0000 (10:17 +0900)
commit562f9e995dfa9e0aaecdfacc880eaa3b357844c2
tree29f5211c3b6621bdc8712ab07713978d980eede3
parentb1fd8a13cc1ab15b53d42996f9cc46aa487d8dbf
[mlir] Vectorize linalg.pad_tensor consumed by transfer_write

Vectorize linalg.pad_tensor without generating a linalg.init_tensor when consumed by a transfer_write.

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