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

4.2 KiB

Raw Blame History

View Raw