[all-commits] [llvm/llvm-project] 7a6674: [mlir][xegpu] Handle scalar uniform ops in SIMT di...

[Charitha Saumya via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 7a6674622603c76274959a9797c0fafb4bc84c44
      https://github.com/llvm/llvm-project/commit/7a6674622603c76274959a9797c0fafb4bc84c44
  Author: Charitha Saumya <136391709+charithaintc at users.noreply.github.com>
  Date:   2025-05-08 (Thu, 08 May 2025)

  Changed paths:
    M mlir/lib/Dialect/XeGPU/Transforms/XeGPUSubgroupDistribute.cpp
    M mlir/test/Dialect/XeGPU/subgroup-distribution.mlir

  Log Message:
  -----------
  [mlir][xegpu] Handle scalar uniform ops in SIMT distribution.  (#138593)

This PR adds support for moving scalar uniform (gpu index ops, constants
etc) outside the `gpu.warp_execute_on_lane0` op. These kinds of ops do
not require distribution and are safe to move out of the warp op. This
also avoid adding separate distribution patterns for these ops.

Example:
```
   %1 = gpu.warp_execute_on_lane_0(%laneid) -> (index) {
     ...
     %block_id_x = gpu.block_id x
     gpu.yield %block_id_x
   }
  // use %1
```
To:
```
   %block_id_x = gpu.block_id x
   %1 = gpu.warp_execute_on_lane_0(%laneid) -> (index) {
     ...
     
     gpu.yield %block_id_x
   }
  // use %1

```



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