[all-commits] [llvm/llvm-project] 6d2fd3: [mlir][linalg] Replace monomorphic contration ops ...

Stella Laurenzo via All-commits all-commits at lists.llvm.org
Mon Mar 1 21:22:46 PST 2021


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 6d2fd3d9cdd6ed24784ec47741e7e70c236a140e
      https://github.com/llvm/llvm-project/commit/6d2fd3d9cdd6ed24784ec47741e7e70c236a140e
  Author: Stella Laurenzo <stellaraccident at gmail.com>
  Date:   2021-03-01 (Mon, 01 Mar 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOps.yaml
    M mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc
    M mlir/test/Dialect/Linalg/generalize-named-polymorphic-ops.mlir

  Log Message:
  -----------
  [mlir][linalg] Replace monomorphic contration ops with polymorphic variants.

* Moves `batch_matmul`, `matmul`, `matvec`, `vectmat`, `dot` to the new mechanism.
* This is not just an NFC change, in addition to using a new code generation mechanism, it also activates symbolic casting, allowing mixed precision operands and results.
* These definitions were generated from DSL by the tool: https://github.com/stellaraccident/mlir-linalgpy/blob/main/mlir_linalg/oplib/core.py (will be upstreamed in a subsequent set of changes).

Reviewed By: nicolasvasilache, ThomasRaoux

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




More information about the All-commits mailing list