diff options
| author | Alexander Belyaev <pifon@google.com> | 2021-03-25 18:08:30 +0100 |
|---|---|---|
| committer | Alexander Belyaev <pifon@google.com> | 2021-03-25 18:11:05 +0100 |
| commit | 7f2236cf581e6d666e4c3eb512a76f1608fe0bf7 (patch) | |
| tree | 4904a45769d98b7c967e8afd83e46467ea49def2 /llvm/lib/Transforms/Utils/BasicBlockUtils.cpp | |
| parent | 4c7ebf79e923072e8d298134e6ca04618fe4eba9 (diff) | |
| download | llvm-7f2236cf581e6d666e4c3eb512a76f1608fe0bf7.zip llvm-7f2236cf581e6d666e4c3eb512a76f1608fe0bf7.tar.gz llvm-7f2236cf581e6d666e4c3eb512a76f1608fe0bf7.tar.bz2 | |
[mlir][linalg] Add output tensor args folding for linalg.tiled_loop.
Folds away TiledLoopOp output tensors when the following conditions are met:
* result of `linalg.tiled_loop` has no uses
* output tensor is the argument of `linalg.yield`
Example:
```
%0 = linalg.tiled_loop ... outs (%out, %out_buf:tensor<...>, memref<...>) {
...
linalg.yield %out : tensor ...
}
```
Becomes
```
linalg.tiled_loop ... outs (%out_buf:memref<...>) {
...
linalg.yield
}
```
Differential Revision: https://reviews.llvm.org/D99333
Diffstat (limited to 'llvm/lib/Transforms/Utils/BasicBlockUtils.cpp')
0 files changed, 0 insertions, 0 deletions