Boian Petkantchin 5e29112719

[mlir][mesh] Add verification and canonicalization for some collectives (#74905) ...

Add verification and canonicalization for
broadcast, gather, recv, reduce, scatter, send and shift.

The canonicalizations only remove trivial collectives with empty
mesh_axes attrubutes.

2023-12-15 06:41:10 -08:00

..

canonicalization.mlir

[mlir][mesh] Add verification and canonicalization for some collectives (#74905)

2023-12-15 06:41:10 -08:00

invalid.mlir

[mlir][mesh] Add verification and canonicalization for some collectives (#74905)

2023-12-15 06:41:10 -08:00

ops.mlir

[mlir][mesh] Add verification and canonicalization for some collectives (#74905)

2023-12-15 06:41:10 -08:00

sharding-propagation.mlir

[mlir][mesh] Use tensor shape notation for the shape of a cluster (#73826)

2023-12-08 11:34:44 -08:00

simplifications.mlir

[mlir][mesh] Add endomorphism simplification for all-reduce (#73150)

2023-12-12 10:21:52 -08:00