[all-commits] [llvm/llvm-project] 2ce6a4: [MLIR] Add Linalg support for integer (generalized...

[Geoffrey Martin-Noble via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 2ce6a42cc94dbe1b0456ebc34c0db238bf5530d6
      https://github.com/llvm/llvm-project/commit/2ce6a42cc94dbe1b0456ebc34c0db238bf5530d6
  Author: Geoffrey Martin-Noble <gcmn at google.com>
  Date:   2021-02-22 (Mon, 22 Feb 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/IR/LinalgInterfaces.td
    M mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc

  Log Message:
  -----------
  [MLIR] Add Linalg support for integer (generalized) matmuls

This patch adds Linalg named ops for various types of integer matmuls.
Due to limitations in the tc spec/linalg-ods-gen ops cannot be type
polymorphic, so this instead creates new ops (improvements to the
methods for defining Linalg named ops are underway with a prototype at
https://github.com/stellaraccident/mlir-linalgpy).

To avoid the necessity of directly referencing these many new ops, this
adds additional methods to ContractionOpInterface to allow classifying
types of operations based on their indexing maps.

Reviewed By: nicolasvasilache, mravishankar

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