[mlir] Add primitive transform pattern to rewrite linalg.fill into vector.broadcast...
authorAhmed Taei <ataei@google.com>
Sat, 25 Jan 2020 00:23:16 +0000 (16:23 -0800)
committerAhmed Taei <ataei@google.com>
Tue, 28 Jan 2020 19:21:56 +0000 (11:21 -0800)
commit16e82d855a1bbedee8035a55680b9c9744479071
tree92fdfa7d2cea6cb08fb8c61b023a1e169447075b
parent60b884208427f8cd317e8d40e64f113c6475c609
[mlir] Add primitive transform pattern to rewrite linalg.fill into vector.broadcast form.

Summary:
This diff adds a transformation patter to rewrite linalg.fill as broadcasting a scaler into a vector.
It uses the same preconditioning as matmul (memory is contiguous).

Reviewers: nicolasvasilache

Subscribers: mehdi_amini, rriddle, jpienaar, burmako, shauheen, antiagainst, arpith-jacob, mgester, lucyrfox, aartbik, liufengdb, llvm-commits

Tags: #llvm

Differential Revision: https://reviews.llvm.org/D73391
mlir/lib/Dialect/Linalg/Transforms/LinalgTransforms.cpp
mlir/test/Dialect/Linalg/transform-patterns.mlir
mlir/test/lib/DeclarativeTransforms/TestLinalgTransformPatterns.td