[all-commits] [llvm/llvm-project] 5bb8d2: [mlir][Linalg] Add tensor support to Linalg EDSC B...

[Nicolas Vasilache via All-commits]

  Branch: refs/heads/master
  Home:   https://github.com/llvm/llvm-project
  Commit: 5bb8d28e61cc6babfad9c232078a48c046af5a1e
      https://github.com/llvm/llvm-project/commit/5bb8d28e61cc6babfad9c232078a48c046af5a1e
  Author: Nicolas Vasilache <ntv at google.com>
  Date:   2020-01-22 (Wed, 22 Jan 2020)

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

  Log Message:
  -----------
  [mlir][Linalg] Add tensor support to Linalg EDSC Builders

Summary:
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.

Reviewers: ftynse, hanchung, asaadaldien

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

Tags: #llvm

Differential Revision: https://reviews.llvm.org/D73149