[Mlir-commits] [mlir] [MLIR][XeGPU] Scattered ops sg-to-wi distribution (PR #154949)
Charitha Saumya
llvmlistbot at llvm.org
Mon Aug 25 10:01:54 PDT 2025
charithaintc wrote:
> > even receive it as an argument from the runtime." what does this mean? a func argument?
>
> Cases where sg distribution has no access to `%offsets = arith.constant dense<0> : vector<16xindex>`. At the SG level of a scattered op, we only see a vector of `n` offsets. We cannot assume (unless otherwise stated in the pass restrictions) whether its producer is distributable or not. Without this assumption, how can we rely on <1xindex>?
>
> I agree that extracting at idx 0 would work if the input is already distributed, but we go bottom up, and we cannot assume that it will be distributed.
I get your point. But I think gather scatter logic should not care about this. It should simply assume this is always distributable. Maybe we should wait for @Jianhui-Li's input also :-)
https://github.com/llvm/llvm-project/pull/154949
More information about the Mlir-commits
mailing list