Boian Petkantchin 1a8fb88719

[mlir][mesh] Add resharding spmdization on a 1D device mesh (#76179) ...

The current implementation supports only sharding of tensor axes that
have size divisible by the mesh axis size.

2024-01-02 15:50:07 -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 resharding spmdization on a 1D device mesh (#76179)

2024-01-02 15:50:07 -08:00

ops.mlir

[mlir][mesh] Add resharding spmdization on a 1D device mesh (#76179)

2024-01-02 15:50:07 -08:00

resharding-spmdization.mlir

[mlir][mesh] Add resharding spmdization on a 1D device mesh (#76179)

2024-01-02 15:50:07 -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