[PATCH] D73391: [mlir] Add primitive transform pattern to rewrite linalg.fill into vector.broadcast form.

[Ahmed S. Taei via Phabricator via llvm-commits]

asaadaldien created this revision.
asaadaldien added a reviewer: nicolasvasilache.
Herald added subscribers: llvm-commits, liufengdb, aartbik, lucyrfox, mgester, arpith-jacob, antiagainst, shauheen, burmako, jpienaar, rriddle, mehdi_amini.
Herald added a project: LLVM.

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).


Repository:
  rG LLVM Github Monorepo

https://reviews.llvm.org/D73391

Files:
  mlir/lib/Dialect/Linalg/Transforms/LinalgTransforms.cpp
  mlir/test/Dialect/Linalg/transform-patterns.mlir
  mlir/test/lib/DeclarativeTransforms/TestLinalgTransformPatterns.td

-------------- next part --------------
A non-text attachment was scrubbed...
Name: D73391.240324.patch
Type: text/x-patch
Size: 6081 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20200125/0009eaf8/attachment.bin>