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

[Charitha Saumya]

charithaintc wrote:

I thought about this again. I think its not complicated to support reduction over multiple dims. We can use progressive lowering inside the pattern itself. 

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

all of this can be done in same pattern in my view. 

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