[PATCH] D77678: [mlir][Linalg] Add loop.parallel lowering for all Linalg Ops.

[Mahesh Ravishankar via Phabricator via llvm-commits]

mravishankar created this revision.
Herald added subscribers: llvm-commits, grosul1, Joonsoo, liufengdb, aartbik, lucyrfox, mgester, arpith-jacob, nicolasvasilache, antiagainst, shauheen, burmako, jpienaar, rriddle, mehdi_amini.
Herald added a reviewer: nicolasvasilache.
Herald added a project: LLVM.
mravishankar added reviewers: antiagainst, hanchung, asaadaldien, ftynse, rriddle.
mravishankar removed subscribers: rriddle, antiagainst, nicolasvasilache.
mravishankar added a reviewer: bondhugula.

The outer parallel loops of a linalg operation is lowered to
loop.parallel, with the other loops lowered to loop.for. This gets the
lowering to loop.parallel on par with the loop.for lowering. In future
the reduction loop could also be lowered to loop.parallel.

Also add a utility function that returns the loops that are
created. This requires change to the EDSC builders to return the
created ops.


Repository:
  rG LLVM Github Monorepo

https://reviews.llvm.org/D77678

Files:
  mlir/include/mlir/Dialect/Affine/EDSC/Builders.h
  mlir/include/mlir/Dialect/Linalg/EDSC/Builders.h
  mlir/include/mlir/Dialect/Linalg/Transforms/LinalgTransforms.h
  mlir/include/mlir/Dialect/LoopOps/EDSC/Builders.h
  mlir/include/mlir/EDSC/Builders.h
  mlir/lib/Dialect/Affine/EDSC/Builders.cpp
  mlir/lib/Dialect/Linalg/EDSC/Builders.cpp
  mlir/lib/Dialect/Linalg/Transforms/LinalgToLoops.cpp
  mlir/lib/Dialect/LoopOps/EDSC/Builders.cpp
  mlir/test/Dialect/Linalg/loops.mlir
  mlir/test/Dialect/Linalg/parallel_loops.mlir

-------------- next part --------------
A non-text attachment was scrubbed...
Name: D77678.255795.patch
Type: text/x-patch
Size: 129500 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20200407/12cf91a5/attachment-0001.bin>