[PATCH] D73149: [mlir][Linalg] Add tensor support to Linalg EDSC Builders

[Nicolas Vasilache via Phabricator via llvm-commits]

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

https://reviews.llvm.org/D72863 landed in an improper state and was reverted, I must have mixed something when updating the diff and later patching again before pushing...
This is the proper update.


This diff extends the Linalg EDSC builders so we can easily create mixed
tensor/buffer linalg.generic ops. This is expected to be useful for
HLO -> Linalg lowering.

The StructuredIndexed struct is made to derive from ValueHandle and can
now capture a type + indexing expressions. This is used to represent return
tensors.

Pointwise unary and binary builders are extended to allow both output buffers
and return tensors. This has implications on the number of region arguments.


Repository:
  rG LLVM Github Monorepo

https://reviews.llvm.org/D73149

Files:
  mlir/include/mlir/Dialect/Linalg/EDSC/Builders.h
  mlir/include/mlir/EDSC/Builders.h
  mlir/lib/Dialect/Linalg/EDSC/Builders.cpp
  mlir/test/EDSC/builder-api-test.cpp

-------------- next part --------------
A non-text attachment was scrubbed...
Name: D73149.239460.patch
Type: text/x-patch
Size: 11354 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20200122/21b7afc9/attachment.bin>