[all-commits] [llvm/llvm-project] c6395b: [MLIR][XeGPU] Enhance Layout Propagation for broad...

[Jianhui Li via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: c6395bb287417be01c982085a3bb363d7c34290e
      https://github.com/llvm/llvm-project/commit/c6395bb287417be01c982085a3bb363d7c34290e
  Author: Jianhui Li <jian.hui.li at intel.com>
  Date:   2026-03-12 (Thu, 12 Mar 2026)

  Changed paths:
    M mlir/lib/Dialect/XeGPU/Transforms/XeGPULayoutImpl.cpp
    M mlir/lib/Dialect/XeGPU/Transforms/XeGPUPropagateLayout.cpp
    M mlir/lib/Dialect/XeGPU/Transforms/XeGPUSubgroupDistribute.cpp
    M mlir/lib/Dialect/XeGPU/Transforms/XeGPUWgToSgDistribute.cpp
    M mlir/test/Dialect/XeGPU/propagate-layout-inst-data.mlir
    M mlir/test/Dialect/XeGPU/propagate-layout-subgroup.mlir
    M mlir/test/Dialect/XeGPU/propagate-layout.mlir
    M mlir/test/Dialect/XeGPU/xegpu-wg-to-sg-unify-ops-rr.mlir
    M mlir/test/Dialect/XeGPU/xegpu-wg-to-sg-unify-ops.mlir

  Log Message:
  -----------
  [MLIR][XeGPU] Enhance Layout Propagation for broadcasting both leading dimensions and  inner unit dimensions (#185583)

This PR enhances the layout propagation rules for broadcast operations.

The source layout is derived from the result layout based on the
broadcast pattern:
1. Broadcast on leading dimensions
  The source layout is the slice layout of the result layout.
2. Broadcast on inner unit dimensions
The source layout matches the result layout, with sg_data and lane_data
set to 1.
3. Broadcast on both leading dimensions and inner unit dimensions
  The source layout is derived by combining the above two rules.



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