[all-commits] [llvm/llvm-project] 315ab3: [MLIR][NVGPU] Introduce `warpgroup.init.accumulato...

Guray Ozen via All-commits all-commits at lists.llvm.org
Wed Oct 11 08:28:40 PDT 2023


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 315ab3c44b56b94725d629137c6c402e676ecb04
      https://github.com/llvm/llvm-project/commit/315ab3c44b56b94725d629137c6c402e676ecb04
  Author: Guray Ozen <guray.ozen at gmail.com>
  Date:   2023-10-11 (Wed, 11 Oct 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/NVGPU/IR/NVGPU.td
    M mlir/lib/Conversion/NVGPUToNVVM/NVGPUToNVVM.cpp
    M mlir/lib/Dialect/NVGPU/IR/NVGPUDialect.cpp
    M mlir/test/Conversion/NVGPUToNVVM/nvgpu-to-nvvm.mlir

  Log Message:
  -----------
  [MLIR][NVGPU] Introduce `warpgroup.init.accumulator` Op (#67530)

This Op generates and initilizes the accumulator matrix for
`nvgpu.warpgroup.mma` op to perform matrix-multiply-and-accumulate
(mma).

Its associated transformation generates `!llvm.struct<>` and fill it
with the initial values. The size of struct is number of required inout
registers for `nvgpu.warpgroup.mma` op.




More information about the All-commits mailing list