[all-commits] [llvm/llvm-project] 8a992b: [mlir][gpu] Add basic support to do elementwise op...

[Thomas via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 8a992b20dba54a061717a14eab86ccbe097da4c0
      https://github.com/llvm/llvm-project/commit/8a992b20dba54a061717a14eab86ccbe097da4c0
  Author: thomasraoux <thomasraoux at google.com>
  Date:   2021-11-01 (Mon, 01 Nov 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/GPU/CMakeLists.txt
    M mlir/include/mlir/Dialect/GPU/GPUBase.td
    M mlir/include/mlir/Dialect/GPU/GPUDialect.h
    M mlir/include/mlir/Dialect/GPU/GPUOps.td
    M mlir/include/mlir/IR/OpBase.td
    M mlir/lib/Conversion/GPUToNVVM/WmmaOpsToNvvm.cpp
    M mlir/lib/Dialect/GPU/CMakeLists.txt
    M mlir/lib/Dialect/GPU/IR/GPUDialect.cpp
    M mlir/test/Conversion/GPUToNVVM/wmma-ops-to-nvvm.mlir
    M mlir/test/Dialect/GPU/ops.mlir
    M utils/bazel/llvm-project-overlay/mlir/BUILD.bazel

  Log Message:
  -----------
  [mlir][gpu] Add basic support to do elementwise ops on mma matrix type

In order to support fusion with mma matrix type we need to be able to
execute elementwise operations on them. This add an op to be able to
support some basic elementwise operations. This is a is not a full
solution as it only supports a limited scope or operations. Ideally we would
want to be able to fuse with more kind of operations.

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