[all-commits] [llvm/llvm-project] 0804a8: [mlir][linalg] Transform PadTensorOp into InitOp, ...

[Nicolas Agostini via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 0804a88e48ac23bcf73d6b985ef755559419ee11
      https://github.com/llvm/llvm-project/commit/0804a88e48ac23bcf73d6b985ef755559419ee11
  Author: Nicolas Agostini <n.b.agostini at gmail.com>
  Date:   2021-06-03 (Thu, 03 Jun 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
    M mlir/lib/Dialect/Linalg/Transforms/Transforms.cpp
    A mlir/test/Dialect/Linalg/lower-pad-tensor.mlir
    M mlir/test/lib/Dialect/Linalg/TestLinalgTransforms.cpp

  Log Message:
  -----------
  [mlir][linalg] Transform PadTensorOp into InitOp, FillOp, GenericOp

Introduces a test pass that rewrites PadTensorOps with static shapes as a sequence of:

```
linalg.init_tensor // to create output
linalg.fill        // to initialize with padding value
linalg.generic     // to copy the original contents to the padded tensor
```

The pass can be triggered with:

- `--test-linalg-transform-patterns="test-transform-pad-tensor"`

Differential Revision: https://reviews.llvm.org/D102804