Petr Kurapov fa6f88af10 [MLIR][XeGPU] Allow some nd ops to have argument shapes mismatch for … (#120566) ...

…the distributed IR case.

This patch allows `nd_load` and `nd_store` to preserve the tensor
descriptor shape during distribution to SIMT. The validation now expects
the distributed instruction to retain the `sg_map` attribute and uses it
to verify the consistency.

2025-01-22 18:03:36 +01:00

..

invalid.mlir

[MLIR][XeGPU] Allow some nd ops to have argument shapes mismatch for … (#120566)

2025-01-22 18:03:36 +01:00

xegpu-fold-alias-ops.mlir

…

XeGPUOps.mlir

[MLIR][XeGPU] Allow some nd ops to have argument shapes mismatch for … (#120566)

2025-01-22 18:03:36 +01:00