xla

8531d1c5 - [MoE] Test sorting lhs for gmm (#7121)

[MoE] Test sorting lhs for gmm (#7121) Summary: This pull request adds a test case that sort the lhs and produce group_sizes for gmm. Test Plan: python test/test_gmm.py -v -k test_sorting_input