[all-commits] [llvm/llvm-project] 2140a9: [mlir][Linalg] Extend generic ops to allow ten...

[Nicolas Vasilache via All-commits]

  Branch: refs/heads/master
  Home:   https://github.com/llvm/llvm-project
  Commit: 2140a973f272969b5391532f3c0b7c90aafd803b
      https://github.com/llvm/llvm-project/commit/2140a973f272969b5391532f3c0b7c90aafd803b
  Author: Nicolas Vasilache <ntv at google.com>
  Date:   2020-01-02 (Thu, 02 Jan 2020)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/IR/LinalgBase.td
    M mlir/include/mlir/Dialect/Linalg/IR/LinalgOps.td
    M mlir/include/mlir/Dialect/Linalg/IR/LinalgStructuredOps.td
    M mlir/include/mlir/Dialect/Linalg/IR/LinalgTraits.h
    M mlir/include/mlir/Dialect/Linalg/Transforms/LinalgTransformPatterns.td
    M mlir/include/mlir/Dialect/Linalg/Transforms/LinalgTransforms.h
    M mlir/include/mlir/Dialect/VectorOps/VectorTransforms.h
    M mlir/lib/Conversion/LinalgToLLVM/LinalgToLLVM.cpp
    M mlir/lib/Dialect/Linalg/EDSC/Builders.cpp
    M mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp
    M mlir/lib/Dialect/Linalg/Transforms/LinalgTransforms.cpp
    M mlir/lib/Dialect/VectorOps/VectorTransforms.cpp
    M mlir/test/Dialect/Linalg/invalid.mlir
    M mlir/test/Dialect/Linalg/roundtrip.mlir
    M mlir/test/lib/DeclarativeTransforms/TestLinalgTransformPatterns.td
    M mlir/tools/mlir-tblgen/RewriterGen.cpp

  Log Message:
  -----------
      [mlir][Linalg] Extend generic ops to allow tensors

    Summary:
    This diff adds support to allow `linalg.generic` and
    `linalg.indexed_generic` to take tensor input and output
    arguments.

    The subset of output tensor operand types must appear
    verbatim in the result types after an arrow. The parser,
    printer and verifier are extended to accomodate this
    behavior.

    The Linalg operations now support variadic ranked tensor
    return values. This extension exhibited issues with the
    current handling of NativeCall in RewriterGen.cpp. As a
    consequence, an explicit cast to `SmallVector<Value, 4>`
    is added in the proper place to support the new behavior
    (better suggestions are welcome).

    Relevant cleanups and name uniformization are applied.

    Relevant invalid and roundtrip test are added.

    Reviewers: mehdi_amini, rriddle, jpienaar, antiagainst, ftynse

    Subscribers: burmako, shauheen, llvm-commits

    Tags: #llvm

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