[all-commits] [llvm/llvm-project] f2b88a: [mlir][SCF][GPU] Add DeviceMaskingAttrInterface su...

[Nicolas Vasilache via All-commits]

  Branch: refs/heads/users/nico/map-to-lanes-3
  Home:   https://github.com/llvm/llvm-project
  Commit: f2b88a1439c84aaf00cdf31670b531c49317cba9
      https://github.com/llvm/llvm-project/commit/f2b88a1439c84aaf00cdf31670b531c49317cba9
  Author: Nicolas Vasilache <nico.vasilache at amd.com>
  Date:   2025-07-07 (Mon, 07 Jul 2025)

  Changed paths:
    M mlir/include/mlir/Dialect/GPU/IR/GPUDeviceMappingAttr.td
    M mlir/include/mlir/Dialect/GPU/TransformOps/Utils.h
    M mlir/include/mlir/Dialect/SCF/IR/DeviceMappingInterface.td
    M mlir/include/mlir/Dialect/SCF/IR/SCFOps.td
    M mlir/lib/Dialect/GPU/CMakeLists.txt
    M mlir/lib/Dialect/GPU/IR/GPUDialect.cpp
    M mlir/lib/Dialect/GPU/TransformOps/GPUTransformOps.cpp
    M mlir/lib/Dialect/GPU/TransformOps/Utils.cpp
    M mlir/lib/Dialect/SCF/IR/SCF.cpp
    M mlir/test/Dialect/GPU/transform-gpu-failing.mlir
    M mlir/test/Dialect/GPU/transform-gpu.mlir
    M mlir/test/Dialect/SCF/invalid.mlir

  Log Message:
  -----------
  [mlir][SCF][GPU] Add DeviceMaskingAttrInterface support to scf::ForallOp and use it to implement warp specialization.

This revision adds DeviceMaskingAttrInterface and extends
DeviceMappingArrayAttr to accept a union of DeviceMappingAttrInterface
and DeviceMaskingAttrInterface.

The first implementation is if the form of a GPUMappingMaskAttr, which
can be additionally passed to the scf.forall.mapping attribute to
specify a mask on compute resources that should be active.

Support is added to GPUTransformOps to take advantage of this
information and lower to block/warpgroup/warp/thread specialization when
mapped to linear ids.

Co-authored-by: Oleksandr "Alex" Zinenko <git at ozinenko.com>


  Commit: 8a0fe9c9d56d2c042e245c7b2a5a231dcf1dec7c
      https://github.com/llvm/llvm-project/commit/8a0fe9c9d56d2c042e245c7b2a5a231dcf1dec7c
  Author: Nicolas Vasilache <nico.vasilache at amd.com>
  Date:   2025-07-07 (Mon, 07 Jul 2025)

  Changed paths:
    M mlir/include/mlir/Dialect/GPU/IR/GPUDeviceMappingAttr.td
    M mlir/include/mlir/Dialect/GPU/TransformOps/GPUTransformOps.h
    M mlir/include/mlir/Dialect/GPU/TransformOps/Utils.h
    M mlir/include/mlir/Dialect/SCF/IR/DeviceMappingInterface.td
    M mlir/lib/Dialect/GPU/IR/GPUDialect.cpp
    M mlir/lib/Dialect/GPU/TransformOps/GPUTransformOps.cpp
    M mlir/lib/Dialect/GPU/TransformOps/Utils.cpp

  Log Message:
  -----------
  Address review


Compare: https://github.com/llvm/llvm-project/compare/0cb1bc9465fb...8a0fe9c9d56d

To unsubscribe from these emails, change your notification settings at https://github.com/llvm/llvm-project/settings/notifications