[mlir] Vectorize linalg.pad_tensor consumed by subtensor_insert
authorMatthias Springer <springerm@google.com>
Mon, 14 Jun 2021 00:58:26 +0000 (09:58 +0900)
committerMatthias Springer <springerm@google.com>
Mon, 14 Jun 2021 00:59:38 +0000 (09:59 +0900)
commitb1fd8a13cc1ab15b53d42996f9cc46aa487d8dbf
tree0fc42179a7b0e9e76a085044df9363255be64400
parentb1b822714db8ea15f811ab03084ee60ff32def21
[mlir] Vectorize linalg.pad_tensor consumed by subtensor_insert

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

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