Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
#0: Fix test_all_gather_multiple_submeshes (#18319)
### Ticket N/A ### Problem description `test_all_gather_multiple_submeshes` is broken, as all gather OP assumes ring connected topology. Physical 2D mesh of devices is as follows (the first 2x2 submesh of T3K): ``` 4 0 5 1 ``` ... which givers `4 0 5 1` row-major ordering, while all gather expects `4 0 1 5`. ### What's changed Reshape submesh to `1x4` to force the the correct ordering. ### Checklist - [X] Ran the test locally and confirmed it fixes the issue.
- Loading branch information