[Mlir-commits] [mlir] [MLIR][XeGPU] Add support for cross-subgroup reduction from wg to sg (PR #170936)

Nishant Patel llvmlistbot at llvm.org
Thu Dec 11 08:35:16 PST 2025


nbpatel wrote:

> * if size(red_dims) > 1, rewrite the reduction into multiple reductions where each reduction is over single dims and return.


why do we need to rewrite it as multiple reduction instructions doing one reduce dim at at time? Wg to sg can do multi-dim reduce locally first  and then do cross-subgroup reduction using SLM? 


https://github.com/llvm/llvm-project/pull/170936


More information about the Mlir-commits mailing list