[all-commits] [llvm/llvm-project] ce25b1: [mlir][xegpu] Fix DPAS subgroup layout propagation...

[Jianhui Li via All-commits]

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

  Changed paths:
    M mlir/lib/Dialect/XeGPU/Transforms/XeGPULayoutImpl.cpp
    M mlir/test/Dialect/XeGPU/propagate-layout-subgroup.mlir

  Log Message:
  -----------
  [mlir][xegpu] Fix DPAS subgroup layout propagation for broadcast K dim (#211695)

This PR fixes an issue in getDpasSubgroupLayouts(): It failed to find
valid subgroup layouts for otherwise-legal DPAS ops. It treated the K
(contraction) dimension as distributed across subgroups — requiring
wgShape % sgLayout == 0 on every dim and gating on a
checkAlignedSgDataAB() equality — when K is actually broadcast: its full
extent stays in every subgroup. This rejected the only valid candidate,
causing layout propagation to bail out.
 
Update the two affected tests, which now propagate successfully instead
of failing.

assisted-by-claude

Co-authored-by: Claude Opus 4.8 <noreply at anthropic.com>



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