llvm-project
1e8834ea - [mlir][vector][xegpu] Accept uniform values in `getDistributedType` (#163887)

Commit
47 days ago
[mlir][vector][xegpu] Accept uniform values in `getDistributedType` (#163887) Uniform values should not be distributed during vector distribution. Example would be a reduction result where reduction happens across lanes. However, current `getDistributedType` does not accept a zero result affine map (i.e. no distributed dims) when describing the distributed dimensions. This result in null type being returned and crashing the vector distribution in some cases. An example case would be a `scf.for` op (about to be distributed) in which one of the for result is a uniform value and it does not have a user outside the warp op. This necessitates querying the `getDistributedType` to figure our the distributed type of this value.
Author
Parents
Loading