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

Nicolas Vasilache via Phabricator via llvm-commits llvm-commits at lists.llvm.org
Thu Jan 16 12:28:06 PST 2020


nicolasvasilache created this revision.
nicolasvasilache added reviewers: ftynse, herhut, hanchung.
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 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 current assumption is that we only pass the result tensor types and the canonical identity identity map is assumed for returned tensors.
In the future we may want to relax this identity assumption (e.g.
for automatic differentiation purposes). In that case we will want to make
StructuredIndexed work with ValueHandle to encode type or value.


Repository:
  rG LLVM Github Monorepo

https://reviews.llvm.org/D72863

Files:
  mlir/include/mlir/Dialect/Linalg/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: D72863.238575.patch
Type: text/x-patch
Size: 27217 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20200116/44ecb966/attachment.bin>


More information about the llvm-commits mailing list