[all-commits] [llvm/llvm-project] 428a62: [mlir][gpu] Add op to create MMA constant matrix

Thomas via All-commits all-commits at lists.llvm.org
Thu Jun 10 08:34:30 PDT 2021


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 428a62f65f16f1640b1bfe033d20e6a4f545dd3e
      https://github.com/llvm/llvm-project/commit/428a62f65f16f1640b1bfe033d20e6a4f545dd3e
  Author: thomasraoux <thomasraoux at google.com>
  Date:   2021-06-10 (Thu, 10 Jun 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/GPU/GPUOps.td
    M mlir/lib/Conversion/GPUToNVVM/WmmaOpsToNvvm.cpp
    M mlir/test/Conversion/GPUToNVVM/wmma-ops-to-nvvm.mlir
    M mlir/test/Dialect/GPU/ops.mlir

  Log Message:
  -----------
  [mlir][gpu] Add op to create MMA constant matrix

This allow creating a matrix with all elements set to a given value. This is
needed to be able to implement a simple dot op.

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




More information about the All-commits mailing list