[mlir][linalg] Add pattern to push reshape after elementwise operation
authorthomasraoux <thomasraoux@google.com>
Fri, 16 Apr 2021 20:38:15 +0000 (13:38 -0700)
committerthomasraoux <thomasraoux@google.com>
Thu, 22 Apr 2021 04:22:39 +0000 (21:22 -0700)
commitd40a19c3a8b3da1a6be40f3b0b56075ed1e31e3a
treeb01fa1f0810769e17c09d2d0e0529eb3dc162b6c
parentf6d8cf7798440f303d5a273999e6647cbe795ac6
[mlir][linalg] Add pattern to push reshape after elementwise operation

This help expose more fusion opportunities.

Differential Revision: https://reviews.llvm.org/D100685
mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
mlir/lib/Dialect/Linalg/Transforms/FusionOnTensors.cpp
mlir/test/Dialect/Linalg/fusion-push-reshape.mlir [new file with mode: 0644]
mlir/test/lib/Transforms/TestLinalgElementwiseFusion.cpp
mlir/tools/mlir-opt/mlir-opt.cpp