[all-commits] [llvm/llvm-project] fbbb8a: [mlir][gpu] Add passes to attach (NVVM|ROCDL) targ...

Fabian Mora via All-commits all-commits at lists.llvm.org
Fri Aug 11 17:45:49 PDT 2023


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: fbbb8adef15cc0f598facc7c6ecffaa247f60b48
      https://github.com/llvm/llvm-project/commit/fbbb8adef15cc0f598facc7c6ecffaa247f60b48
  Author: Fabian Mora <fmora.dev at gmail.com>
  Date:   2023-08-12 (Sat, 12 Aug 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/GPU/Transforms/Passes.td
    M mlir/lib/Dialect/GPU/CMakeLists.txt
    A mlir/lib/Dialect/GPU/Transforms/NVVMAttachTarget.cpp
    A mlir/lib/Dialect/GPU/Transforms/ROCDLAttachTarget.cpp
    A mlir/test/Dialect/LLVMIR/attach-targets.mlir

  Log Message:
  -----------
  [mlir][gpu] Add passes to attach (NVVM|ROCDL) target attributes to GPU Modules

Adds the passes `nvvm-attach-target` & `rocdl-attach-target for attaching `nvvm.target` & `rocdl.target` attributes to GPU Modules.

These passes search GPU Modules in the immediate region of the Op being acted on, attaching the target attribute to the module.
Modules can be selected using a regex string, allowing fine grain attachment of targets, see the test `attach-target.mlir` for an example.

Depends on D154153

Reviewed By: mehdi_amini

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




More information about the All-commits mailing list