[all-commits] [llvm/llvm-project] b74cfc: [mlir][nvgpu] Improve nvgpu->nvvm transformation o...

Guray Ozen via All-commits all-commits at lists.llvm.org
Thu Oct 5 01:17:13 PDT 2023


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

  Changed paths:
    M mlir/lib/Conversion/NVGPUToNVVM/NVGPUToNVVM.cpp

  Log Message:
  -----------
  [mlir][nvgpu] Improve nvgpu->nvvm transformation of `warpgroup.mma` Op (NFC) (#67325)

This PR introduces substantial improvements to the readability and
maintainability of the `nvgpu.warpgroup.mma` Op transformation from
nvgpu->nvvm. This transformation plays a crucial role in GEMM and
manages complex operations such as generating multiple wgmma ops and
iterating their descriptors. The prior code lacked clarity, but this PR
addresses that issue effectively.

**PR does followings:**
**Introduces a helper class:** `WarpgroupGemm` class encapsulates the
necessary functionality, making the code cleaner and more
understandable.

**Detailed Documentation:** Each function within the helper class is
thoroughly documented to provide clear insights into its purpose and
functionality.




More information about the All-commits mailing list